初版功能完成

internal/model/model_test.go

@@ -0,0 +1,75 @@
+package model
+
+import (
+	"encoding/json"
+	"math"
+	"strings"
+	"testing"
+)
+
+func TestNodeTypesAreCompleteAndValid(t *testing.T) {
+	t.Parallel()
+	if got, want := len(NodeTypes), 2; got != want {
+		t.Fatalf("NodeTypes length = %d, want %d", got, want)
+	}
+	for _, nodeType := range NodeTypes {
+		if !nodeType.Valid() {
+			t.Fatalf("listed node type %q is invalid", nodeType)
+		}
+	}
+	if _, err := ParseNodeType("server"); err == nil {
+		t.Fatal("ParseNodeType accepted the Server, which is not a Windows Node role")
+	}
+}
+
+func TestNodeStatusesAreCompleteAndValid(t *testing.T) {
+	t.Parallel()
+	for _, status := range []NodeStatus{NodeOnline, NodeUnstable, NodeOffline} {
+		if !status.Valid() {
+			t.Fatalf("listed Node status %q is invalid", status)
+		}
+	}
+	if NodeStatus("UNKNOWN").Valid() {
+		t.Fatal("unknown Node status accepted")
+	}
+}
+
+func TestSessionStatusesAreCompleteAndValid(t *testing.T) {
+	t.Parallel()
+	if got, want := len(SessionStatuses), 7; got != want {
+		t.Fatalf("SessionStatuses length = %d, want %d", got, want)
+	}
+	seen := make(map[SessionStatus]struct{}, len(SessionStatuses))
+	for _, status := range SessionStatuses {
+		if !status.Valid() {
+			t.Fatalf("listed Session status %q is invalid", status)
+		}
+		if _, duplicate := seen[status]; duplicate {
+			t.Fatalf("duplicate Session status %q", status)
+		}
+		seen[status] = struct{}{}
+	}
+	if _, err := ParseSessionStatus("RUNNING"); err == nil {
+		t.Fatal("ParseSessionStatus accepted a non-spec state")
+	}
+}
+
+func TestSessionIDsUseLosslessJSONDecimalStrings(t *testing.T) {
+	t.Parallel()
+	session := Session{ID: math.MaxUint64, Status: SessionActive}
+	raw, err := json.Marshal(session)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !strings.Contains(string(raw), `"session_id":"18446744073709551615"`) {
+		t.Fatalf("SessionID was not encoded losslessly for JavaScript clients: %s", raw)
+	}
+	var decoded Session
+	if err := json.Unmarshal(raw, &decoded); err != nil || decoded.ID != session.ID {
+		t.Fatalf("SessionID round trip = %d, %v", decoded.ID, err)
+	}
+	eventRaw, err := json.Marshal(EventLog{SessionID: math.MaxUint64})
+	if err != nil || !strings.Contains(string(eventRaw), `"session_id":"18446744073709551615"`) {
+		t.Fatalf("Event SessionID was not encoded losslessly: %s, %v", eventRaw, err)
+	}
+}