1 files changed, 31 insertions, 0 deletions
diff --git a/internal/tui/tui_test.go b/internal/tui/tui_test.go
new file mode 100644
index 0000000..2aa7b84
--- /dev/null
+++ b/internal/tui/tui_test.go
@@ -0,0 +1,31 @@
+package tui
+
+import (
+	"testing"
+
+	"github.com/lukaszkasprzak/lectio/internal/config"
+)
+
+func TestVersionCycle(t *testing.T) {
+	m := New(config.Config{Versions: []string{"pl", "wuj", "vul"}, DefaultVersion: "pl"})
+	if m.version() != "pl" {
+		t.Fatalf("start = %q", m.version())
+	}
+	m = m.cycleVersion(+1)
+	if m.version() != "wuj" {
+		t.Errorf("after tab = %q", m.version())
+	}
+	m = m.cycleVersion(-1)
+	if m.version() != "pl" {
+		t.Errorf("after shift-tab = %q", m.version())
+	}
+}
+
+func TestOfflineDropsPL(t *testing.T) {
+	m := New(config.Config{Versions: []string{"pl", "wuj", "vul"}, DefaultVersion: "pl", Offline: true})
+	for _, v := range m.versions {
+		if v == "pl" {
+			t.Error("offline model kept pl")
+		}
+	}
+}