summaryrefslogtreecommitdiff
path: root/internal/llm/openai_http_test.go
diff options
context:
space:
mode:
authorPaul Buetow <paul@buetow.org>2025-09-05 20:55:01 +0300
committerPaul Buetow <paul@buetow.org>2025-09-05 20:55:01 +0300
commit95e0633abaf5779c17c133f94037f38b73c72d3e (patch)
tree64c11d662aa9e1cb2ea94ccdd6c452d53dbf71e6 /internal/llm/openai_http_test.go
parentd99bdf981dbee7e038e4a8e262504c1a15047c38 (diff)
tests: add more negative provider cases and table-driven LSP coverage; assert headers; add indent postprocess test
Diffstat (limited to 'internal/llm/openai_http_test.go')
-rw-r--r--internal/llm/openai_http_test.go28
1 files changed, 28 insertions, 0 deletions
diff --git a/internal/llm/openai_http_test.go b/internal/llm/openai_http_test.go
index 45f0c99..808bb2b 100644
--- a/internal/llm/openai_http_test.go
+++ b/internal/llm/openai_http_test.go
@@ -63,6 +63,34 @@ func TestOpenAI_ChatStream_SSE_ErrorChunk(t *testing.T) {
}
}
+func TestOpenAI_Chat_NoChoices_Error(t *testing.T) {
+ srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+ _ = json.NewEncoder(w).Encode(map[string]any{"choices": []any{}})
+ }))
+ defer srv.Close()
+ c := newOpenAI(srv.URL, "g", "KEY", f64p(0.2)).(openAIClient)
+ c.httpClient = srv.Client()
+ if _, err := c.Chat(context.Background(), []Message{{Role:"user", Content:"hi"}}); err == nil {
+ t.Fatalf("expected error when choices empty")
+ }
+}
+
+func TestOpenAI_ChatStream_SSE_EmptyDelta_NoError(t *testing.T) {
+ srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+ w.Header().Set("Content-Type", "text/event-stream")
+ io.WriteString(w, "data: {\\\"choices\\\":[{\\\"delta\\\":{\\\"content\\\":\\\"\\\"}}]}\\n\\n")
+ io.WriteString(w, "data: [DONE]\\n")
+ }))
+ defer srv.Close()
+ c := newOpenAI(srv.URL, "g", "KEY", f64p(0.2)).(openAIClient)
+ c.httpClient = srv.Client()
+ var got string
+ if err := c.ChatStream(context.Background(), []Message{{Role:"user", Content:"hi"}}, func(s string){ got += s }); err != nil {
+ t.Fatalf("unexpected error for empty delta: %v", err)
+ }
+ if got != "" { t.Fatalf("expected no output for empty delta, got %q", got) }
+}
+
func TestOpenAI_Chat_DecodeError_StatusOK(t *testing.T) {
// Return status 200 but invalid JSON body; Chat should return an error
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {