48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
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))
|
|
}
|