feat(sync): full code sync from release
This commit is contained in:
47
backend/internal/service/ops_retry_context_test.go
Normal file
47
backend/internal/service/ops_retry_context_test.go
Normal file
@@ -0,0 +1,47 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestNewOpsRetryContext_SetsHTTPTransportAndRequestHeaders(t *testing.T) {
|
||||
errorLog := &OpsErrorLogDetail{
|
||||
OpsErrorLog: OpsErrorLog{
|
||||
RequestPath: "/openai/v1/responses",
|
||||
},
|
||||
UserAgent: "ops-retry-agent/1.0",
|
||||
RequestHeaders: `{
|
||||
"anthropic-beta":"beta-v1",
|
||||
"ANTHROPIC-VERSION":"2023-06-01",
|
||||
"authorization":"Bearer should-not-forward"
|
||||
}`,
|
||||
}
|
||||
|
||||
c, w := newOpsRetryContext(context.Background(), errorLog)
|
||||
require.NotNil(t, c)
|
||||
require.NotNil(t, w)
|
||||
require.NotNil(t, c.Request)
|
||||
|
||||
require.Equal(t, "/openai/v1/responses", c.Request.URL.Path)
|
||||
require.Equal(t, "application/json", c.Request.Header.Get("Content-Type"))
|
||||
require.Equal(t, "ops-retry-agent/1.0", c.Request.Header.Get("User-Agent"))
|
||||
require.Equal(t, "beta-v1", c.Request.Header.Get("anthropic-beta"))
|
||||
require.Equal(t, "2023-06-01", c.Request.Header.Get("anthropic-version"))
|
||||
require.Empty(t, c.Request.Header.Get("authorization"), "未在白名单内的敏感头不应被重放")
|
||||
require.Equal(t, OpenAIClientTransportHTTP, GetOpenAIClientTransport(c))
|
||||
}
|
||||
|
||||
func TestNewOpsRetryContext_InvalidHeadersJSONStillSetsHTTPTransport(t *testing.T) {
|
||||
errorLog := &OpsErrorLogDetail{
|
||||
RequestHeaders: "{invalid-json",
|
||||
}
|
||||
|
||||
c, _ := newOpsRetryContext(context.Background(), errorLog)
|
||||
require.NotNil(t, c)
|
||||
require.NotNil(t, c.Request)
|
||||
require.Equal(t, "/", c.Request.URL.Path)
|
||||
require.Equal(t, OpenAIClientTransportHTTP, GetOpenAIClientTransport(c))
|
||||
}
|
||||
Reference in New Issue
Block a user