748a84d871
- Extract PublicSettingsInjectionPayload named struct with drift test - Add channel_monitor_default_interval_seconds to SSR injection - Add image_output_price to SupportedModelChip - Simplify AppSidebar buildSelfNavItems (admins see available channels) - Add gateway WARN logs for 503 no-available-accounts branches - Wire ChannelMonitorRunner into provideCleanup for graceful shutdown - Add migrations 130/131 (CC template userid fix + mimicry field cleanup) - Clean up fork-only features (sora, claude max simulation, client affinity) - Remove ~320 obsolete i18n keys - Add codexUsage utility, WechatServiceButton, BulkEditAccountModal - Tidy go.sum
106 lines
3.7 KiB
Go
106 lines
3.7 KiB
Go
package service
|
|
|
|
import (
|
|
"bytes"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestOpenAIGatewayServiceParseOpenAIImagesRequest_JSON(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
body := []byte(`{"model":"gpt-image-2","prompt":"draw a cat","size":"1024x1024","quality":"high","stream":true}`)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/images/generations", bytes.NewReader(body))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
rec := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(rec)
|
|
c.Request = req
|
|
|
|
svc := &OpenAIGatewayService{}
|
|
parsed, err := svc.ParseOpenAIImagesRequest(c, body)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, parsed)
|
|
require.Equal(t, "/v1/images/generations", parsed.Endpoint)
|
|
require.Equal(t, "gpt-image-2", parsed.Model)
|
|
require.Equal(t, "draw a cat", parsed.Prompt)
|
|
require.True(t, parsed.Stream)
|
|
require.Equal(t, "1024x1024", parsed.Size)
|
|
require.Equal(t, "1K", parsed.SizeTier)
|
|
require.Equal(t, OpenAIImagesCapabilityNative, parsed.RequiredCapability)
|
|
require.False(t, parsed.Multipart)
|
|
}
|
|
|
|
func TestOpenAIGatewayServiceParseOpenAIImagesRequest_MultipartEdit(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
var body bytes.Buffer
|
|
writer := multipart.NewWriter(&body)
|
|
require.NoError(t, writer.WriteField("model", "gpt-image-2"))
|
|
require.NoError(t, writer.WriteField("prompt", "replace background"))
|
|
require.NoError(t, writer.WriteField("size", "1536x1024"))
|
|
part, err := writer.CreateFormFile("image", "source.png")
|
|
require.NoError(t, err)
|
|
_, err = part.Write([]byte("fake-image-bytes"))
|
|
require.NoError(t, err)
|
|
require.NoError(t, writer.Close())
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/images/edits", bytes.NewReader(body.Bytes()))
|
|
req.Header.Set("Content-Type", writer.FormDataContentType())
|
|
rec := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(rec)
|
|
c.Request = req
|
|
|
|
svc := &OpenAIGatewayService{}
|
|
parsed, err := svc.ParseOpenAIImagesRequest(c, body.Bytes())
|
|
require.NoError(t, err)
|
|
require.NotNil(t, parsed)
|
|
require.Equal(t, "/v1/images/edits", parsed.Endpoint)
|
|
require.True(t, parsed.Multipart)
|
|
require.Equal(t, "gpt-image-2", parsed.Model)
|
|
require.Equal(t, "replace background", parsed.Prompt)
|
|
require.Equal(t, "1536x1024", parsed.Size)
|
|
require.Equal(t, "2K", parsed.SizeTier)
|
|
require.Len(t, parsed.Uploads, 1)
|
|
require.Equal(t, OpenAIImagesCapabilityNative, parsed.RequiredCapability)
|
|
}
|
|
|
|
func TestOpenAIGatewayServiceParseOpenAIImagesRequest_PromptOnlyDefaultsRemainBasic(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
body := []byte(`{"prompt":"draw a cat"}`)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/images/generations", bytes.NewReader(body))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
rec := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(rec)
|
|
c.Request = req
|
|
|
|
svc := &OpenAIGatewayService{}
|
|
parsed, err := svc.ParseOpenAIImagesRequest(c, body)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, parsed)
|
|
require.Equal(t, "gpt-image-2", parsed.Model)
|
|
require.Equal(t, OpenAIImagesCapabilityBasic, parsed.RequiredCapability)
|
|
}
|
|
|
|
func TestOpenAIGatewayServiceParseOpenAIImagesRequest_ExplicitSizeRequiresNativeCapability(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
body := []byte(`{"prompt":"draw a cat","size":"1024x1024"}`)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/images/generations", bytes.NewReader(body))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
rec := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(rec)
|
|
c.Request = req
|
|
|
|
svc := &OpenAIGatewayService{}
|
|
parsed, err := svc.ParseOpenAIImagesRequest(c, body)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, parsed)
|
|
require.Equal(t, OpenAIImagesCapabilityNative, parsed.RequiredCapability)
|
|
}
|