| package openai |
|
|
| import ( |
| "bytes" |
| "context" |
| "encoding/json" |
| "net/http" |
| "net/http/httptest" |
| "testing" |
|
|
| "github.com/go-chi/chi/v5" |
|
|
| "ds2api/internal/account" |
| "ds2api/internal/auth" |
| "ds2api/internal/config" |
| ) |
|
|
| func newResolverWithConfigJSON(t *testing.T, cfgJSON string) (*config.Store, *auth.Resolver) { |
| t.Helper() |
| t.Setenv("DS2API_CONFIG_JSON", cfgJSON) |
| store := config.LoadStore() |
| pool := account.NewPool(store) |
| resolver := auth.NewResolver(store, pool, func(_ context.Context, _ config.Account) (string, error) { |
| return "unused", nil |
| }) |
| return store, resolver |
| } |
|
|
| func TestEmbeddingsRouteContract(t *testing.T) { |
| store, resolver := newResolverWithConfigJSON(t, `{"embeddings":{"provider":"deterministic"}}`) |
| h := &openAITestSurface{Store: store, Auth: resolver} |
| r := chi.NewRouter() |
| registerOpenAITestRoutes(r, h) |
|
|
| t.Run("unauthorized", func(t *testing.T) { |
| body := bytes.NewBufferString(`{"model":"gpt-4o","input":"hello"}`) |
| req := httptest.NewRequest(http.MethodPost, "/v1/embeddings", body) |
| req.Header.Set("Content-Type", "application/json") |
| rec := httptest.NewRecorder() |
| r.ServeHTTP(rec, req) |
| if rec.Code != http.StatusUnauthorized { |
| t.Fatalf("expected 401, got %d body=%s", rec.Code, rec.Body.String()) |
| } |
| }) |
|
|
| t.Run("ok", func(t *testing.T) { |
| body := bytes.NewBufferString(`{"model":"gpt-4o","input":["a","b"]}`) |
| req := httptest.NewRequest(http.MethodPost, "/v1/embeddings", body) |
| req.Header.Set("Authorization", "Bearer test-token") |
| req.Header.Set("Content-Type", "application/json") |
| rec := httptest.NewRecorder() |
| r.ServeHTTP(rec, req) |
| if rec.Code != http.StatusOK { |
| t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String()) |
| } |
| var out map[string]any |
| if err := json.Unmarshal(rec.Body.Bytes(), &out); err != nil { |
| t.Fatalf("decode response failed: %v", err) |
| } |
| if out["object"] != "list" { |
| t.Fatalf("unexpected object: %#v", out["object"]) |
| } |
| data, _ := out["data"].([]any) |
| if len(data) != 2 { |
| t.Fatalf("expected 2 embeddings, got %d", len(data)) |
| } |
| }) |
| } |
|
|
| func TestEmbeddingsRouteProviderMissing(t *testing.T) { |
| store, resolver := newResolverWithConfigJSON(t, `{}`) |
| h := &openAITestSurface{Store: store, Auth: resolver} |
| r := chi.NewRouter() |
| registerOpenAITestRoutes(r, h) |
|
|
| body := bytes.NewBufferString(`{"model":"gpt-4o","input":"hello"}`) |
| req := httptest.NewRequest(http.MethodPost, "/v1/embeddings", body) |
| req.Header.Set("Authorization", "Bearer test-token") |
| req.Header.Set("Content-Type", "application/json") |
| rec := httptest.NewRecorder() |
| r.ServeHTTP(rec, req) |
| if rec.Code != http.StatusNotImplemented { |
| t.Fatalf("expected 501, got %d body=%s", rec.Code, rec.Body.String()) |
| } |
| var out map[string]any |
| if err := json.Unmarshal(rec.Body.Bytes(), &out); err != nil { |
| t.Fatalf("decode response failed: %v", err) |
| } |
| errObj, _ := out["error"].(map[string]any) |
| if _, ok := errObj["code"]; !ok { |
| t.Fatalf("expected error.code in response: %#v", out) |
| } |
| if _, ok := errObj["param"]; !ok { |
| t.Fatalf("expected error.param in response: %#v", out) |
| } |
| } |
|
|
