create a websocket game server

game-server/internal/protocol/protocol_test.go

@@ -0,0 +1,71 @@
+package protocol
+
+import (
+	"encoding/json"
+	"testing"
+)
+
+func TestParseJoinRoundtrip(t *testing.T) {
+	opcode, args, err := ParseMessage([]byte(`[1,"game-token","user-token"]`))
+	if err != nil {
+		t.Fatalf("parse: %v", err)
+	}
+	if opcode != OpJoin {
+		t.Fatalf("opcode = %d, want %d", opcode, OpJoin)
+	}
+	gt, ut, err := ParseJoin(args)
+	if err != nil {
+		t.Fatalf("parse join: %v", err)
+	}
+	if gt != "game-token" || ut != "user-token" {
+		t.Fatalf("got (%q, %q)", gt, ut)
+	}
+}
+
+func TestParseMessageErrors(t *testing.T) {
+	for _, data := range []string{`{}`, `[]`, `["1"]`, `not json`} {
+		if _, _, err := ParseMessage([]byte(data)); err == nil {
+			t.Fatalf("expected error for %s", data)
+		}
+	}
+}
+
+func TestBuildEnd(t *testing.T) {
+	data := BuildEnd(7, EndCodeJoinTimeout, "join timeout exceeded")
+	var v []any
+	if err := json.Unmarshal(data, &v); err != nil {
+		t.Fatalf("unmarshal: %v", err)
+	}
+	want := []any{float64(2), float64(7), float64(-1), "join timeout exceeded"}
+	if len(v) != len(want) {
+		t.Fatalf("got %v, want %v", v, want)
+	}
+	for i := range want {
+		if v[i] != want[i] {
+			t.Fatalf("got %v, want %v", v, want)
+		}
+	}
+}
+
+func TestBuildStateAndAck(t *testing.T) {
+	if got := string(BuildState(3, StatePlacingShips)); got != `[3,3,1]` {
+		t.Fatalf("BuildState = %s", got)
+	}
+	if got := string(BuildAck(3)); got != `[4,3]` {
+		t.Fatalf("BuildAck = %s", got)
+	}
+}
+
+func TestParseAck(t *testing.T) {
+	_, args, err := ParseMessage([]byte(`[4,12]`))
+	if err != nil {
+		t.Fatalf("parse: %v", err)
+	}
+	seq, err := ParseAck(args)
+	if err != nil {
+		t.Fatalf("parse ack: %v", err)
+	}
+	if seq != 12 {
+		t.Fatalf("seq = %d, want 12", seq)
+	}
+}