| package sseparser |
|
|
| import ( |
| "testing" |
|
|
| chatgpt_types "aurora/typings/chatgpt" |
| ) |
|
|
| func TestDataPayloads_SingleDataLine(t *testing.T) { |
| line := "data: {\"key\":\"value\"}\n\n" |
| payloads := DataPayloads(line) |
| if len(payloads) != 1 { |
| t.Fatalf("expected 1 payload, got %d", len(payloads)) |
| } |
| if payloads[0] != `{"key":"value"}` { |
| t.Errorf("unexpected payload: %s", payloads[0]) |
| } |
| } |
|
|
| func TestDataPayloads_MultipleDataLines(t *testing.T) { |
| line := "data: {\"a\":1}\ndata: {\"b\":2}\n\n" |
| payloads := DataPayloads(line) |
| if len(payloads) != 2 { |
| t.Fatalf("expected 2 payloads, got %d", len(payloads)) |
| } |
| } |
|
|
| func TestDataPayloads_DoneMarker(t *testing.T) { |
| line := "data: [DONE]\n\n" |
| payloads := DataPayloads(line) |
| if len(payloads) != 1 || payloads[0] != "[DONE]" { |
| t.Fatalf("expected [DONE], got %v", payloads) |
| } |
| } |
|
|
| func TestDataPayloads_EmptyLine(t *testing.T) { |
| payloads := DataPayloads("") |
| if len(payloads) != 0 { |
| t.Fatalf("expected 0 payloads, got %d", len(payloads)) |
| } |
| } |
|
|
| func TestSplitDataPayloads_JSON(t *testing.T) { |
| payloads := SplitDataPayloads(`{"key":"value"}`) |
| if len(payloads) != 1 || payloads[0] != `{"key":"value"}` { |
| t.Fatalf("unexpected: %v", payloads) |
| } |
| } |
|
|
| func TestSplitDataPayloads_Done(t *testing.T) { |
| payloads := SplitDataPayloads("[DONE]") |
| if len(payloads) != 1 || payloads[0] != "[DONE]" { |
| t.Fatalf("unexpected: %v", payloads) |
| } |
| } |
|
|
| func TestEventName_Present(t *testing.T) { |
| line := "event: response.created\ndata: {}\n\n" |
| name, ok := EventName(line) |
| if !ok || name != "response.created" { |
| t.Fatalf("expected response.created, got %q (ok=%v)", name, ok) |
| } |
| } |
|
|
| func TestEventName_Absent(t *testing.T) { |
| line := "data: {}\n\n" |
| name, ok := EventName(line) |
| if ok { |
| t.Fatalf("expected no event, got %q", name) |
| } |
| } |
|
|
| func TestNumberToInt64_Float64(t *testing.T) { |
| v, ok := NumberToInt64(float64(42.5)) |
| if !ok || v != 42 { |
| t.Fatalf("expected 42, got %d (ok=%v)", v, ok) |
| } |
| } |
|
|
| func TestNumberToInt64_Int(t *testing.T) { |
| v, ok := NumberToInt64(int(100)) |
| if !ok || v != 100 { |
| t.Fatalf("expected 100, got %d (ok=%v)", v, ok) |
| } |
| } |
|
|
| func TestNumberToInt64_String(t *testing.T) { |
| _, ok := NumberToInt64("not a number") |
| if ok { |
| t.Fatal("expected false for string input") |
| } |
| } |
|
|
| func TestChannelFromValue_Nested(t *testing.T) { |
| value := map[string]interface{}{ |
| "delta": map[string]interface{}{ |
| "channel": "test-channel", |
| }, |
| } |
| ch := ChannelFromValue(value) |
| if ch != "test-channel" { |
| t.Fatalf("expected test-channel, got %q", ch) |
| } |
| } |
|
|
| func TestChannelFromValue_Empty(t *testing.T) { |
| ch := ChannelFromValue(map[string]interface{}{}) |
| if ch != "" { |
| t.Fatalf("expected empty, got %q", ch) |
| } |
| } |
|
|
| func TestFirstStringPart(t *testing.T) { |
| parts := []interface{}{"hello", "world"} |
| if s := FirstStringPart(parts); s != "hello" { |
| t.Fatalf("expected hello, got %q", s) |
| } |
| if s := FirstStringPart(nil); s != "" { |
| t.Fatalf("expected empty, got %q", s) |
| } |
| } |
|
|
| func TestNormalizeContentDelta(t *testing.T) { |
| if d := NormalizeContentDelta("", "hello"); d != "hello" { |
| t.Fatalf("expected hello, got %q", d) |
| } |
| if d := NormalizeContentDelta("hel", "hello"); d != "lo" { |
| t.Fatalf("expected lo, got %q", d) |
| } |
| if d := NormalizeContentDelta("hello", ""); d != "" { |
| t.Fatalf("expected empty, got %q", d) |
| } |
| } |
|
|
| func TestIsUsableConversationResponse(t *testing.T) { |
| |
| if IsUsableConversationResponse(chatgpt_types.ChatGPTResponse{}) { |
| t.Fatal("empty response should not be usable") |
| } |
| } |
|
|
