73 lines
1.4 KiB
Go
73 lines
1.4 KiB
Go
package views
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/derailed/k9s/internal/resource"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestTVSortRows(t *testing.T) {
|
|
uu := []struct {
|
|
rows resource.RowEvents
|
|
col int
|
|
asc bool
|
|
first resource.Row
|
|
e []string
|
|
}{
|
|
{
|
|
resource.RowEvents{
|
|
"row1": {Fields: resource.Row{"x", "y"}},
|
|
"row2": {Fields: resource.Row{"a", "b"}},
|
|
},
|
|
0,
|
|
true,
|
|
resource.Row{"a", "b"},
|
|
[]string{"row2", "row1"},
|
|
},
|
|
{
|
|
resource.RowEvents{
|
|
"row1": {Fields: resource.Row{"x", "y"}},
|
|
"row2": {Fields: resource.Row{"a", "b"}},
|
|
},
|
|
1,
|
|
true,
|
|
resource.Row{"a", "b"},
|
|
[]string{"row2", "row1"},
|
|
},
|
|
{
|
|
resource.RowEvents{
|
|
"row1": {Fields: resource.Row{"x", "y"}},
|
|
"row2": {Fields: resource.Row{"a", "b"}},
|
|
},
|
|
1,
|
|
false,
|
|
resource.Row{"x", "y"},
|
|
[]string{"row1", "row2"},
|
|
},
|
|
}
|
|
|
|
var v *tableView
|
|
for _, u := range uu {
|
|
keys := make([]string, len(u.rows))
|
|
v.sortRows(u.rows, v.defaultSort, sortColumn{u.col, len(u.rows), u.asc}, keys)
|
|
assert.Equal(t, u.e, keys)
|
|
assert.Equal(t, u.first, u.rows[u.e[0]].Fields)
|
|
}
|
|
}
|
|
|
|
func BenchmarkTVSortRows(b *testing.B) {
|
|
evts := resource.RowEvents{
|
|
"row1": {Fields: resource.Row{"x", "y"}},
|
|
"row2": {Fields: resource.Row{"a", "b"}},
|
|
}
|
|
sc := sortColumn{0, 2, true}
|
|
var v *tableView
|
|
keys := make([]string, len(evts))
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
v.sortRows(evts, v.defaultSort, sc, keys)
|
|
}
|
|
}
|