Add projects

backend/internal/client/grokbuild/grokbuild_test.go

@@ -0,0 +1,50 @@
+package grokbuild
+
+import "testing"
+
+func TestIsGrokShellUserAgent(t *testing.T) {
+	tests := []struct {
+		name string
+		ua   string
+		want bool
+	}{
+		{"shell", "grok-shell/0.2.119 (macos; aarch64)", true},
+		{"pager", "grok-pager/0.2.119 grok-shell/0.2.119 (macos; aarch64)", true},
+		{"case insensitive", "GROK-PAGER/1.0 GROK-SHELL/1.0", true},
+		{"ordinary client", "curl/8.7.1", false},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			if got := IsGrokShellUserAgent(test.ua); got != test.want {
+				t.Fatalf("IsGrokShellUserAgent(%q) = %t, want %t", test.ua, got, test.want)
+			}
+		})
+	}
+}
+
+func TestBuildResponse(t *testing.T) {
+	response := BuildResponse([]ModelInfo{
+		{ID: "grok-4", DisplayName: "Grok 4", ContextLength: 256000, ReasoningLevels: []string{"high"}},
+		{ID: "plain-model", ContextLength: 0},
+	})
+
+	if response.Object != "list" || len(response.Data) != 2 {
+		t.Fatalf("response envelope = %#v", response)
+	}
+	entry := response.Data[0]
+	if entry.ID != "grok-4" || entry.Model != "grok-4" || entry.Name != "Grok 4" {
+		t.Fatalf("entry identity = %#v", entry)
+	}
+	if entry.ContextWindow != 256000 {
+		t.Fatalf("entry context = %#v", entry)
+	}
+	if entry.APIBackend != "responses" || !entry.SupportedInAPI {
+		t.Fatalf("entry fixed fields = %#v", entry)
+	}
+	if len(entry.ReasoningEfforts) != 1 || entry.ReasoningEfforts[0].Value != "high" {
+		t.Fatalf("reasoning efforts = %#v", entry.ReasoningEfforts)
+	}
+	if response.Data[1].Name != "plain-model" || response.Data[1].ContextWindow != 0 || response.Data[1].ReasoningEfforts != nil {
+		t.Fatalf("fallback/omitempty mapping = %#v", response.Data[1])
+	}
+}