da36506b89
P0 fixes: - ModelError.Is(): use exact matching instead of substring contains() - shouldClearStickySession: add context param for cancellation/tracing P1 fixes: - TODO stubs: return 501 Not Implemented errors - validateInstanceSignature: deduplicate to shared validateCodeSignature() - Error messages: standardize to English only - http.go: remove pseudo if-else with duplicate branches
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)
|
|
}
|