95 lines
1.4 KiB
Go
95 lines
1.4 KiB
Go
|
package menu
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func TestMenuInit(t *testing.T) {
|
||
|
m := NewMenu()
|
||
|
err := m.Put("1", "foo")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
err = m.Put("2", "bar")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
r, err := m.Render(0)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
expect := `1:foo
|
||
|
2:bar`
|
||
|
if r != expect {
|
||
|
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", expect, r)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestMenuBrowse(t *testing.T) {
|
||
|
cfg := DefaultBrowseConfig()
|
||
|
m := NewMenu().WithPageCount(3).WithBrowseConfig(cfg)
|
||
|
err := m.Put("1", "foo")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
err = m.Put("2", "bar")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
r, err := m.Render(0)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
expect := `1:foo
|
||
|
2:bar
|
||
|
11:next`
|
||
|
if r != expect {
|
||
|
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", expect, r)
|
||
|
}
|
||
|
|
||
|
err = m.Put("1", "foo")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
err = m.Put("2", "bar")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
r, err = m.Render(1)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
expect = `1:foo
|
||
|
2:bar
|
||
|
11:next
|
||
|
22:previous`
|
||
|
if r != expect {
|
||
|
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", expect, r)
|
||
|
}
|
||
|
|
||
|
err = m.Put("1", "foo")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
err = m.Put("2", "bar")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
r, err = m.Render(2)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
expect = `1:foo
|
||
|
2:bar
|
||
|
22:previous`
|
||
|
if r != expect {
|
||
|
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", expect, r)
|
||
|
}
|
||
|
|
||
|
_, err = m.Render(3)
|
||
|
if err == nil {
|
||
|
t.Fatalf("expected render fail")
|
||
|
}
|
||
|
}
|