Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/openai_gateway_passthrough_image_intent_test.go

@@ -0,0 +1,59 @@
+package service
+
+import (
+	"context"
+	"io"
+	"net/http"
+	"strings"
+	"testing"
+
+	"github.com/gin-gonic/gin"
+	"github.com/stretchr/testify/require"
+	"github.com/tidwall/gjson"
+)
+
+func TestOpenAIGatewayService_APIKeyPassthrough_ImageIntentPreservesGateAndBilling(t *testing.T) {
+	gin.SetMode(gin.TestMode)
+	body := []byte(`{"model":"gpt-5.4","stream":false,"tools":[{"type":"image_generation","model":"gpt-image-2","size":"2048x1152"}],"input":"draw"}`)
+
+	t.Run("disabled group rejects before upstream", func(t *testing.T) {
+		upstream := &httpUpstreamRecorder{}
+		svc := newOpenAIImageGenerationControlTestService(upstream)
+		c, recorder := newOpenAIImageGenerationControlTestContext(false, "curl/8.0")
+		account := newOpenAIImageGenerationControlTestAccount()
+		account.Extra = map[string]any{"openai_passthrough": true}
+
+		result, err := svc.Forward(context.Background(), c, account, body)
+
+		require.Error(t, err)
+		require.Nil(t, result)
+		require.Equal(t, http.StatusForbidden, recorder.Code)
+		require.Equal(t, "permission_error", gjson.GetBytes(recorder.Body.Bytes(), "error.type").String())
+		require.Nil(t, upstream.lastReq)
+	})
+
+	t.Run("allowed group keeps image billing", func(t *testing.T) {
+		upstream := &httpUpstreamRecorder{resp: &http.Response{
+			StatusCode: http.StatusOK,
+			Header:     http.Header{"Content-Type": []string{"application/json"}},
+			Body: io.NopCloser(strings.NewReader(
+				`{"output":[{"id":"ig_1","type":"image_generation_call","result":"final-image","size":"2048x1152"}],"usage":{"input_tokens":1,"output_tokens":2}}`,
+			)),
+		}}
+		svc := newOpenAIImageGenerationControlTestService(upstream)
+		c, _ := newOpenAIImageGenerationControlTestContext(true, "curl/8.0")
+		account := newOpenAIImageGenerationControlTestAccount()
+		account.Extra = map[string]any{"openai_passthrough": true}
+
+		result, err := svc.Forward(context.Background(), c, account, body)
+
+		require.NoError(t, err)
+		require.NotNil(t, result)
+		require.NotNil(t, upstream.lastReq)
+		require.Equal(t, body, upstream.lastBody)
+		require.Equal(t, 1, result.ImageCount)
+		require.Equal(t, "gpt-image-2", result.BillingModel)
+		require.Equal(t, "2K", result.ImageSize)
+		require.Equal(t, "2048x1152", result.ImageInputSize)
+	})
+}