Refactored package structure

1 files changed, 14 insertions, 0 deletions

diff --git a/pkg/client/state_test.go b/pkg/client/state_test.go
new file mode 100644
index 0000000..917c211
--- /dev/null
+++ b/pkg/client/state_test.go
@@ -0,0 +1,14 @@
+package client
+
+import "testing"
+
+func TestNewState(t *testing.T) {
+	// Empty login
+	if s := NewState(nil, ""); s.currentMenu != "anonymous" {
+		t.Fatal("a new state without login should init to anonymous")
+	}
+	// logged_in
+	if s := NewState(nil, "test"); s.currentMenu != "logged_in" {
+		t.Fatal("a new state with login should init to logged_in")
+	}
+}