c548021921
- 同步 OpenAI 图片生成与编辑接口 - 接入图片请求解析、账号调度、转发与用量记录 - 接入图片计费与图片用量落库 - 限制 OAuth 生图仅支持无显式模型和尺寸的基础请求
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)
|
|
}
|