mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-04-03 15:02:13 +08:00
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAppendOpsUpstreamError_UsesRequestBodyBytesFromContext(t *testing.T) {
|
||
|
|
gin.SetMode(gin.TestMode)
|
||
|
|
rec := httptest.NewRecorder()
|
||
|
|
c, _ := gin.CreateTestContext(rec)
|
||
|
|
|
||
|
|
setOpsUpstreamRequestBody(c, []byte(`{"model":"gpt-5"}`))
|
||
|
|
appendOpsUpstreamError(c, OpsUpstreamErrorEvent{
|
||
|
|
Kind: "http_error",
|
||
|
|
Message: "upstream failed",
|
||
|
|
})
|
||
|
|
|
||
|
|
v, ok := c.Get(OpsUpstreamErrorsKey)
|
||
|
|
require.True(t, ok)
|
||
|
|
events, ok := v.([]*OpsUpstreamErrorEvent)
|
||
|
|
require.True(t, ok)
|
||
|
|
require.Len(t, events, 1)
|
||
|
|
require.Equal(t, `{"model":"gpt-5"}`, events[0].UpstreamRequestBody)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAppendOpsUpstreamError_UsesRequestBodyStringFromContext(t *testing.T) {
|
||
|
|
gin.SetMode(gin.TestMode)
|
||
|
|
rec := httptest.NewRecorder()
|
||
|
|
c, _ := gin.CreateTestContext(rec)
|
||
|
|
|
||
|
|
c.Set(OpsUpstreamRequestBodyKey, `{"model":"gpt-4"}`)
|
||
|
|
appendOpsUpstreamError(c, OpsUpstreamErrorEvent{
|
||
|
|
Kind: "request_error",
|
||
|
|
Message: "dial timeout",
|
||
|
|
})
|
||
|
|
|
||
|
|
v, ok := c.Get(OpsUpstreamErrorsKey)
|
||
|
|
require.True(t, ok)
|
||
|
|
events, ok := v.([]*OpsUpstreamErrorEvent)
|
||
|
|
require.True(t, ok)
|
||
|
|
require.Len(t, events, 1)
|
||
|
|
require.Equal(t, `{"model":"gpt-4"}`, events[0].UpstreamRequestBody)
|
||
|
|
}
|