ce7cb702218693d08ea6d83464d2457c9250027c

services/codex-bridge/chat_test.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+)
+
+func TestInvalidChat(t *testing.T) {
+	for _, body := range []string{
+		`{`,
+		`{"model":"m","messages":[]}`,
+		`{"model":"m","messages":[{"role":"assistant","content":"text"}]}`,
+		`{"model":"m","messages":[{"role":"user","content":[{"type":"image_url","image_url":{"url":"https://example.com/image"}}]}]}`,
+		`{"model":"m","messages":[{"role":"user","content":null}]}`,
+		`{"model":"m","messages":[{"role":"user","content":"text"}],"tools":[{"type":"function"}]}`,
+	} {
+		t.Run(body, func(t *testing.T) {
+			w := httptest.NewRecorder()
+			(&bridge{binary: "/nonexistent"}).handler().ServeHTTP(w, httptest.NewRequest(http.MethodPost, "/v1/chat/completions", strings.NewReader(body)))
+			if w.Code != http.StatusBadRequest {
+				t.Fatalf("status %d: %s", w.Code, w.Body)
+			}
+		})
+	}
+}
+
+func TestHistoryRoles(t *testing.T) {
+	var req chatRequest
+	err := json.Unmarshal([]byte(`{"model":"m","messages":[{"role":"system","content":"Instruction"},{"role":"user","content":"Question"},{"role":"assistant","content":"Answer"},{"role":"user","content":[{"type":"text","text":"Follow"},{"type":"text","text":"up"}]}]}`), &req)
+	if err != nil {
+		t.Fatal(err)
+	}
+	items, prompt, err := inputs(req)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if prompt != "Followup" || len(items) != 3 {
+		t.Fatalf("%v %q", items, prompt)
+	}
+	for i, role := range []string{"developer", "user", "assistant"} {
+		if items[i]["role"] != role {
+			t.Fatalf("history %v", items)
+		}
+	}
+	if got := items[2]["content"].([]object)[0]; got["type"] != "output_text" || got["text"] != "Answer" {
+		t.Fatalf("assistant content %v", got)
+	}
+}
+
+func TestUnavailableCodex(t *testing.T) {
+	w := httptest.NewRecorder()
+	(&bridge{binary: "/nonexistent"}).handler().ServeHTTP(w, httptest.NewRequest(http.MethodGet, "/v1/models", nil))
+	if w.Code != http.StatusBadGateway {
+		t.Fatalf("status %d", w.Code)
+	}
+	var result struct {
+		Error *rpcError `json:"error"`
+	}
+	if err := json.Unmarshal(w.Body.Bytes(), &result); err != nil || result.Error == nil {
+		t.Fatalf("invalid error: %s", w.Body)
+	}
+}