李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
457 lines
16 KiB
Go
457 lines
16 KiB
Go
package routes
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/config"
|
|
"github.com/Wei-Shaw/sub2api/internal/handler"
|
|
servermiddleware "github.com/Wei-Shaw/sub2api/internal/server/middleware"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func newGatewayRoutesTestRouter(platform ...string) *gin.Engine {
|
|
return newGatewayRoutesTestRouterWithConfig(&config.Config{
|
|
Gateway: config.GatewayConfig{
|
|
MaxBodySize: 1024 * 1024,
|
|
TextMaxBodySize: 1024 * 1024,
|
|
},
|
|
}, platform...)
|
|
}
|
|
|
|
func newGatewayRoutesTestRouterWithConfig(cfg *config.Config, platform ...string) *gin.Engine {
|
|
gin.SetMode(gin.TestMode)
|
|
router := gin.New()
|
|
|
|
groupPlatform := service.PlatformOpenAI
|
|
if len(platform) > 0 && platform[0] != "" {
|
|
groupPlatform = platform[0]
|
|
}
|
|
RegisterGatewayRoutes(
|
|
router,
|
|
&handler.Handlers{
|
|
Gateway: &handler.GatewayHandler{},
|
|
OpenAIGateway: &handler.OpenAIGatewayHandler{},
|
|
AsyncImage: handler.NewAsyncImageHandler(nil, nil),
|
|
},
|
|
servermiddleware.APIKeyAuthMiddleware(func(c *gin.Context) {
|
|
groupID := int64(1)
|
|
c.Set(string(servermiddleware.ContextKeyAPIKey), &service.APIKey{
|
|
GroupID: &groupID,
|
|
Group: &service.Group{Platform: groupPlatform},
|
|
})
|
|
c.Next()
|
|
}),
|
|
nil,
|
|
nil,
|
|
nil,
|
|
nil,
|
|
nil,
|
|
cfg,
|
|
)
|
|
|
|
return router
|
|
}
|
|
|
|
func TestGatewayRoutesOpenAIResponsesCompactPathIsRegistered(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter()
|
|
|
|
for _, path := range []string{
|
|
"/v1/responses/compact",
|
|
"/responses/compact",
|
|
"/backend-api/codex/responses",
|
|
"/backend-api/codex/responses/compact",
|
|
} {
|
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(`{"model":"gpt-5"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.NotEqual(t, http.StatusNotFound, w.Code, "path=%s should hit OpenAI responses handler", path)
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesOpenAIAlphaSearchPathsAreRegistered(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter()
|
|
registered := make(map[string]bool)
|
|
for _, route := range router.Routes() {
|
|
if route.Method == http.MethodPost {
|
|
registered[route.Path] = true
|
|
}
|
|
}
|
|
|
|
for _, path := range []string{
|
|
"/v1/alpha/search",
|
|
"/alpha/search",
|
|
"/backend-api/codex/alpha/search",
|
|
} {
|
|
require.True(t, registered[path], "POST %s should be registered", path)
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesAlphaSearchRejectsUnsupportedGroup(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformGrok)
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/alpha/search", strings.NewReader(`{"model":"gpt-5.6-sol"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
|
|
require.Equal(t, http.StatusNotFound, w.Code)
|
|
require.Contains(t, w.Body.String(), "only available for OpenAI and Composite groups")
|
|
}
|
|
|
|
func TestGatewayRoutesOpenAIImagesPathsAreRegistered(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter()
|
|
|
|
for _, path := range []string{
|
|
"/v1/images/generations",
|
|
"/v1/images/edits",
|
|
"/images/generations",
|
|
"/images/edits",
|
|
} {
|
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(`{"model":"gpt-image-2","prompt":"draw a cat"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.NotEqual(t, http.StatusNotFound, w.Code, "path=%s should hit OpenAI images handler", path)
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesAsyncImagesPathsAreRegistered(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter()
|
|
registered := make(map[string]bool)
|
|
for _, route := range router.Routes() {
|
|
registered[route.Method+" "+route.Path] = true
|
|
}
|
|
|
|
for _, route := range []string{
|
|
"POST /v1/images/generations/async",
|
|
"POST /v1/images/edits/async",
|
|
"GET /v1/images/tasks/:task_id",
|
|
"POST /images/generations/async",
|
|
"POST /images/edits/async",
|
|
"GET /images/tasks/:task_id",
|
|
} {
|
|
require.True(t, registered[route], "%s should be registered", route)
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesGrokImagesAndVideosPathsAreRegistered(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformGrok)
|
|
|
|
for _, path := range []string{
|
|
"/v1/images/generations",
|
|
"/v1/images/edits",
|
|
"/images/generations",
|
|
"/images/edits",
|
|
"/v1/videos/generations",
|
|
"/v1/videos",
|
|
"/videos",
|
|
"/videos/generations",
|
|
"/v1/videos/edits",
|
|
"/videos/edits",
|
|
"/v1/videos/extensions",
|
|
"/videos/extensions",
|
|
} {
|
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(`{"model":"grok-imagine","prompt":"draw a cat"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.NotEqual(t, http.StatusNotFound, w.Code, "path=%s should hit Grok media handler", path)
|
|
require.NotContains(t, w.Body.String(), "not supported for this platform")
|
|
}
|
|
|
|
for _, path := range []string{
|
|
"/v1/videos/request-123",
|
|
"/videos/request-123",
|
|
"/v1/videos/generations/request-123",
|
|
"/videos/generations/request-123",
|
|
"/v1/videos/edits/request-123",
|
|
"/videos/edits/request-123",
|
|
"/v1/videos/extensions/request-123",
|
|
"/videos/extensions/request-123",
|
|
"/v1/videos/request-123/content",
|
|
"/videos/request-123/content",
|
|
"/v1/videos/generations/request-123/content",
|
|
"/videos/generations/request-123/content",
|
|
"/v1/videos/edits/request-123/content",
|
|
"/videos/edits/request-123/content",
|
|
"/v1/videos/extensions/request-123/content",
|
|
"/videos/extensions/request-123/content",
|
|
} {
|
|
req := httptest.NewRequest(http.MethodGet, path, nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.NotEqual(t, http.StatusNotFound, w.Code, "path=%s should hit Grok video handler", path)
|
|
require.NotContains(t, w.Body.String(), "not supported for this platform")
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesGrokCustomVoiceCRUDPathsAreRegistered(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformGrok)
|
|
registered := make(map[string]bool)
|
|
for _, route := range router.Routes() {
|
|
registered[route.Method+" "+route.Path] = true
|
|
}
|
|
for _, route := range []string{
|
|
"POST /v1/custom-voices",
|
|
"GET /v1/custom-voices",
|
|
"GET /v1/custom-voices/:voice_id",
|
|
"PATCH /v1/custom-voices/:voice_id",
|
|
"DELETE /v1/custom-voices/:voice_id",
|
|
"GET /v1/custom-voices/:voice_id/audio",
|
|
"POST /custom-voices",
|
|
"GET /custom-voices",
|
|
"GET /custom-voices/:voice_id",
|
|
"PATCH /custom-voices/:voice_id",
|
|
"DELETE /custom-voices/:voice_id",
|
|
"GET /custom-voices/:voice_id/audio",
|
|
} {
|
|
require.True(t, registered[route], "%s should be registered", route)
|
|
}
|
|
}
|
|
|
|
func TestGrokCustomVoiceEndpointUsesRouteTemplateNotRawPath(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
router := gin.New()
|
|
var got string
|
|
capture := func(c *gin.Context) {
|
|
got = grokCustomVoiceEndpoint(c)
|
|
c.Status(http.StatusOK)
|
|
}
|
|
router.GET("/v1/custom-voices/:voice_id/audio", capture)
|
|
router.GET("/v1/custom-voices/:voice_id", capture)
|
|
|
|
for _, tc := range []struct {
|
|
path string
|
|
want string
|
|
}{
|
|
{path: "/v1/custom-voices/voice-123", want: "custom-voices/voice-123"},
|
|
{path: "/v1/custom-voices/voice-123/audio", want: "custom-voices/voice-123/audio"},
|
|
// A voice literally named "audio" matches /:voice_id, not /:voice_id/audio.
|
|
// A raw-path suffix check would turn this profile lookup into an audio download.
|
|
{path: "/v1/custom-voices/audio", want: "custom-voices/audio"},
|
|
{path: "/v1/custom-voices/audio/audio", want: "custom-voices/audio/audio"},
|
|
} {
|
|
got = ""
|
|
req := httptest.NewRequest(http.MethodGet, tc.path, nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusOK, w.Code, "path=%s", tc.path)
|
|
require.Equal(t, tc.want, got, "path=%s", tc.path)
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesCompositeVideoLookupsUseGrokHandler(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformComposite)
|
|
|
|
for _, path := range []string{
|
|
"/v1/videos/request-123",
|
|
"/videos/request-123",
|
|
"/v1/videos/request-123/content",
|
|
"/videos/request-123/content",
|
|
} {
|
|
req := httptest.NewRequest(http.MethodGet, path, nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.NotEqual(t, http.StatusNotFound, w.Code, "path=%s should hit Grok video lookup handler", path)
|
|
require.NotContains(t, w.Body.String(), "not supported for this platform")
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesCompositeMessagesWithGrokModelUsesOpenAIGateway(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformComposite)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/messages", strings.NewReader(`{"model":"grok-4.3","messages":[{"role":"user","content":"hi"}]}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
|
|
require.NotEqual(t, http.StatusNotFound, w.Code)
|
|
require.NotContains(t, w.Body.String(), "not supported")
|
|
require.NotContains(t, w.Body.String(), "OpenAI-compatible endpoint")
|
|
require.NotContains(t, w.Body.String(), "composite groups")
|
|
}
|
|
|
|
func TestGatewayRoutesCompositeChatCompletionsWithGrokModelUsesOpenAIGateway(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformComposite)
|
|
|
|
for _, path := range []string{"/v1/chat/completions", "/chat/completions"} {
|
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(`{"model":"grok-4.3","messages":[{"role":"user","content":"hi"}]}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
|
|
require.NotEqual(t, http.StatusNotFound, w.Code, "path=%s", path)
|
|
require.NotContains(t, w.Body.String(), "not supported")
|
|
require.NotContains(t, w.Body.String(), "OpenAI-compatible endpoint")
|
|
require.NotContains(t, w.Body.String(), "composite groups")
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesNonGrokVideosAreRejectedAtPlatformGate(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformOpenAI)
|
|
|
|
for _, tc := range []struct {
|
|
method string
|
|
path string
|
|
body string
|
|
}{
|
|
{http.MethodPost, "/v1/videos/generations", `{"model":"grok-imagine-video-1.5","prompt":"waves"}`},
|
|
{http.MethodPost, "/v1/videos", `{"model":"grok-imagine-video-1.5","prompt":"waves"}`},
|
|
{http.MethodPost, "/videos", `{"model":"grok-imagine-video-1.5","prompt":"waves"}`},
|
|
{http.MethodPost, "/videos/generations", `{"model":"grok-imagine-video-1.5","prompt":"waves"}`},
|
|
{http.MethodPost, "/v1/videos/edits", `{"model":"grok-imagine-video","prompt":"waves","video":{"url":"https://example.com/in.mp4"}}`},
|
|
{http.MethodPost, "/videos/edits", `{"model":"grok-imagine-video","prompt":"waves","video":{"url":"https://example.com/in.mp4"}}`},
|
|
{http.MethodPost, "/v1/videos/extensions", `{"model":"grok-imagine-video","prompt":"waves","video":{"url":"https://example.com/in.mp4"}}`},
|
|
{http.MethodPost, "/videos/extensions", `{"model":"grok-imagine-video","prompt":"waves","video":{"url":"https://example.com/in.mp4"}}`},
|
|
{http.MethodGet, "/v1/videos/request-123", ""},
|
|
{http.MethodGet, "/videos/request-123", ""},
|
|
{http.MethodGet, "/v1/videos/generations/request-123", ""},
|
|
{http.MethodGet, "/videos/generations/request-123", ""},
|
|
{http.MethodGet, "/v1/videos/edits/request-123", ""},
|
|
{http.MethodGet, "/videos/edits/request-123", ""},
|
|
{http.MethodGet, "/v1/videos/extensions/request-123", ""},
|
|
{http.MethodGet, "/videos/extensions/request-123", ""},
|
|
{http.MethodGet, "/v1/videos/request-123/content", ""},
|
|
{http.MethodGet, "/videos/request-123/content", ""},
|
|
{http.MethodGet, "/v1/videos/generations/request-123/content", ""},
|
|
{http.MethodGet, "/videos/generations/request-123/content", ""},
|
|
{http.MethodGet, "/v1/videos/edits/request-123/content", ""},
|
|
{http.MethodGet, "/videos/edits/request-123/content", ""},
|
|
{http.MethodGet, "/v1/videos/extensions/request-123/content", ""},
|
|
{http.MethodGet, "/videos/extensions/request-123/content", ""},
|
|
} {
|
|
req := httptest.NewRequest(tc.method, tc.path, strings.NewReader(tc.body))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusNotFound, w.Code, "method=%s path=%s", tc.method, tc.path)
|
|
require.Contains(t, w.Body.String(), "Videos API is not supported for this platform")
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesCompositeOpenAIOnlyEndpointsRequireOpenAITarget(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformComposite)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/embeddings", strings.NewReader(`{"model":"gemini-2.5-pro","input":"hello"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusNotFound, w.Code)
|
|
|
|
req = httptest.NewRequest(http.MethodPost, "/v1/embeddings", strings.NewReader(`{"model":"text-embedding-3-small","input":"hello"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w = httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.NotEqual(t, http.StatusNotFound, w.Code)
|
|
}
|
|
|
|
func TestGatewayRoutesGrokAllowsCLICompatibilityEntrypoints(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformGrok)
|
|
|
|
for _, tc := range []struct {
|
|
method string
|
|
path string
|
|
}{
|
|
{http.MethodPost, "/v1/messages"},
|
|
{http.MethodPost, "/v1/chat/completions"},
|
|
{http.MethodPost, "/chat/completions"},
|
|
{http.MethodGet, "/v1/responses"},
|
|
{http.MethodGet, "/responses"},
|
|
{http.MethodGet, "/backend-api/codex/responses"},
|
|
} {
|
|
req := httptest.NewRequest(tc.method, tc.path, strings.NewReader(`{"model":"grok"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.NotEqual(t, http.StatusNotFound, w.Code, "method=%s path=%s", tc.method, tc.path)
|
|
require.NotContains(t, w.Body.String(), "not supported for Grok groups")
|
|
}
|
|
|
|
countTokensRouter := newGatewayRoutesTestRouterWithConfig(&config.Config{
|
|
Gateway: config.GatewayConfig{MaxBodySize: 1024 * 1024},
|
|
}, service.PlatformGrok)
|
|
for _, path := range []string{"/v1/messages/count_tokens", "/messages/count_tokens"} {
|
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(`{"model":"grok","messages":[{"role":"user","content":"hi"}]}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
countTokensRouter.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusOK, w.Code, "path=%s", path)
|
|
var response struct {
|
|
InputTokens int `json:"input_tokens"`
|
|
}
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &response), "path=%s", path)
|
|
require.Positive(t, response.InputTokens, "path=%s", path)
|
|
}
|
|
|
|
for _, path := range []string{
|
|
"/v1/responses",
|
|
"/responses",
|
|
"/backend-api/codex/responses",
|
|
} {
|
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(`{"model":"grok","input":"hi"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.NotEqual(t, http.StatusNotFound, w.Code, "path=%s should still reach Responses handler", path)
|
|
}
|
|
}
|
|
|
|
// TestGatewayRoutesResponsesSubpathRejectsNonConformingSubpaths 端到端锁定不变式：
|
|
// /responses/*subpath 的子路径会被转发到上游同名端点之后，因此不合规的子路径必须
|
|
// 在入口就被拒绝，不得进入调度与转发流程。
|
|
func TestGatewayRoutesResponsesSubpathRejectsNonConformingSubpaths(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter()
|
|
|
|
for _, path := range []string{
|
|
"/v1/responses/../../x/y",
|
|
"/v1/responses/..%2f..%2fx/y",
|
|
"/v1/responses/%2e%2e/%2e%2e/x",
|
|
"/responses/%2e%2e%2fx",
|
|
"/backend-api/codex/responses/..%2f..%2fx",
|
|
`/v1/responses/..\..\x`,
|
|
"/v1/responses/%3fa=b",
|
|
"/v1/responses/x%23frag",
|
|
"/v1/responses/compact%2f..",
|
|
} {
|
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(`{"model":"gpt-5"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusNotFound, w.Code, "path=%s must be rejected at the edge", path)
|
|
require.Contains(t, w.Body.String(), "Unsupported responses subpath", "path=%s", path)
|
|
}
|
|
}
|
|
|
|
func TestGatewayRoutesOpenAICountTokensPathIsRegistered(t *testing.T) {
|
|
router := newGatewayRoutesTestRouter(service.PlatformOpenAI)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/messages/count_tokens", strings.NewReader(`{"model":"claude-sonnet-4-5","messages":[{"role":"user","content":"hi"}]}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(w, req)
|
|
require.NotEqual(t, http.StatusNotFound, w.Code)
|
|
}
|