76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
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)
|
|
}
|
|
}
|