156 lines
3.8 KiB
Go
156 lines
3.8 KiB
Go
package view
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/derailed/k9s/internal"
|
|
"github.com/derailed/k9s/internal/config"
|
|
"github.com/derailed/k9s/internal/model"
|
|
"github.com/derailed/k9s/internal/render"
|
|
"github.com/derailed/k9s/internal/ui"
|
|
"github.com/derailed/tview"
|
|
"github.com/stretchr/testify/assert"
|
|
v1 "k8s.io/api/core/v1"
|
|
)
|
|
|
|
func TestTableSave(t *testing.T) {
|
|
v := NewTable("test")
|
|
v.Init(makeContext())
|
|
v.SetTitle("k9s-test")
|
|
|
|
dir := filepath.Join(config.K9sDumpDir, v.app.Config.K9s.CurrentCluster)
|
|
c1, _ := ioutil.ReadDir(dir)
|
|
v.saveCmd(nil)
|
|
|
|
c2, _ := ioutil.ReadDir(dir)
|
|
assert.Equal(t, len(c2), len(c1)+1)
|
|
}
|
|
|
|
func TestTableNew(t *testing.T) {
|
|
v := NewTable("test")
|
|
v.Init(makeContext())
|
|
|
|
data := render.NewTableData()
|
|
data.Header = render.HeaderRow{
|
|
render.Header{Name: "NAMESPACE"},
|
|
render.Header{Name: "NAME", Align: tview.AlignRight},
|
|
render.Header{Name: "FRED"},
|
|
render.Header{Name: "AGE", Decorator: render.AgeDecorator},
|
|
}
|
|
data.RowEvents = render.RowEvents{
|
|
render.RowEvent{
|
|
Row: render.Row{
|
|
Fields: render.Fields{"ns1", "a", "10", "3m"},
|
|
},
|
|
},
|
|
render.RowEvent{
|
|
Row: render.Row{
|
|
Fields: render.Fields{"ns1", "b", "15", "1m"},
|
|
},
|
|
},
|
|
}
|
|
data.Namespace = ""
|
|
|
|
v.Update(*data)
|
|
assert.Equal(t, 3, v.GetRowCount())
|
|
}
|
|
|
|
func TestTableViewFilter(t *testing.T) {
|
|
v := NewTable("test")
|
|
v.Init(makeContext())
|
|
v.SetModel(&testTableModel{})
|
|
v.SearchBuff().SetActive(true)
|
|
v.SearchBuff().Set("blee")
|
|
v.Refresh()
|
|
assert.Equal(t, 2, v.GetRowCount())
|
|
}
|
|
|
|
func TestTableViewSort(t *testing.T) {
|
|
v := NewTable("test")
|
|
v.Init(makeContext())
|
|
v.SetModel(&testTableModel{})
|
|
v.SortColCmd(1, true)(nil)
|
|
assert.Equal(t, 3, v.GetRowCount())
|
|
assert.Equal(t, "blee", v.GetCell(1, 1).Text)
|
|
|
|
v.SortInvertCmd(nil)
|
|
assert.Equal(t, 3, v.GetRowCount())
|
|
assert.Equal(t, "fred", v.GetCell(1, 1).Text)
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// Helpers...
|
|
|
|
type testTableModel struct{}
|
|
|
|
var _ ui.Tabular = &testTableModel{}
|
|
|
|
func (t *testTableModel) Empty() bool { return false }
|
|
func (t *testTableModel) Peek() render.TableData { return makeTableData() }
|
|
func (t *testTableModel) ClusterWide() bool { return false }
|
|
func (t *testTableModel) GetNamespace() string { return "blee" }
|
|
func (t *testTableModel) SetNamespace(string) {}
|
|
func (t *testTableModel) AddListener(model.TableListener) {}
|
|
func (t *testTableModel) Watch(context.Context) {}
|
|
func (t *testTableModel) InNamespace(string) bool { return true }
|
|
func (t *testTableModel) SetRefreshRate(time.Duration) {}
|
|
|
|
func makeTableData() render.TableData {
|
|
t := render.NewTableData()
|
|
|
|
t.Header = render.HeaderRow{
|
|
render.Header{Name: "NAMESPACE"},
|
|
render.Header{Name: "NAME", Align: tview.AlignRight},
|
|
render.Header{Name: "FRED"},
|
|
render.Header{Name: "AGE", Decorator: render.AgeDecorator},
|
|
}
|
|
t.RowEvents = render.RowEvents{
|
|
render.RowEvent{
|
|
Row: render.Row{
|
|
Fields: render.Fields{"ns1", "blee", "10", "3m"},
|
|
},
|
|
},
|
|
render.RowEvent{
|
|
Row: render.Row{
|
|
Fields: render.Fields{"ns1", "fred", "15", "1m"},
|
|
},
|
|
Deltas: render.DeltaRow{"", "", "20", ""},
|
|
},
|
|
}
|
|
t.Namespace = ""
|
|
|
|
return *t
|
|
}
|
|
|
|
func makeContext() context.Context {
|
|
a := NewApp(config.NewConfig(ks{}))
|
|
ctx := context.WithValue(context.Background(), internal.KeyApp, a)
|
|
return context.WithValue(ctx, internal.KeyStyles, a.Styles)
|
|
}
|
|
|
|
type ks struct{}
|
|
|
|
func (k ks) CurrentContextName() (string, error) {
|
|
return "test", nil
|
|
}
|
|
|
|
func (k ks) CurrentClusterName() (string, error) {
|
|
return "test", nil
|
|
}
|
|
|
|
func (k ks) CurrentNamespaceName() (string, error) {
|
|
return "test", nil
|
|
}
|
|
|
|
func (k ks) ClusterNames() ([]string, error) {
|
|
return []string{"test"}, nil
|
|
}
|
|
|
|
func (k ks) NamespaceNames(nn []v1.Namespace) []string {
|
|
return []string{"test"}
|
|
}
|