internal/llm/openai_sse_negative_test.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

package llm

import (
    "context"
    "io"
    "net/http"
    "net/http/httptest"
    "testing"
)

func TestOpenAI_ChatStream_SSE_MalformedChunk(t *testing.T) {
    // Malformed JSON chunk should be skipped; no onDelta calls; no error.
    srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Content-Type", "text/event-stream")
        io.WriteString(w, "data: {not json}\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 malformed chunk: %v", err)
    }
    if got != "" { t.Fatalf("expected no deltas for malformed chunk, got %q", got) }
}