mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-14 12:02:27 +08:00
64 lines
2.5 KiB
Go
64 lines
2.5 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/config"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
func TestBuildUpstreamRequestToolChangeBetas(t *testing.T) {
|
|
const mid = "mid-conversation-tool-changes-2026-07-01"
|
|
const inline = "inline-tools-2026-09-15"
|
|
for _, tc := range []struct {
|
|
name, header string
|
|
drop map[string]struct{}
|
|
wantMid, wantInline bool
|
|
}{
|
|
{"both", mid + "," + inline, nil, true, true},
|
|
{"mid only", mid, nil, true, false},
|
|
{"inline only", inline, nil, false, true},
|
|
{"absent", "", nil, false, false},
|
|
{"unknown", inline + "-other", nil, false, false},
|
|
{"duplicate", inline + ", " + inline + "," + mid, nil, true, true},
|
|
{"filter inline", mid + "," + inline, map[string]struct{}{inline: {}}, true, false},
|
|
{"filter mid", mid + "," + inline, map[string]struct{}{mid: {}}, false, true},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
c, _ := gin.CreateTestContext(httptest.NewRecorder())
|
|
c.Request = httptest.NewRequest(http.MethodPost, "/v1/messages", nil)
|
|
c.Request.Header.Set("anthropic-beta", tc.header+",unrelated-beta")
|
|
if tc.drop != nil {
|
|
c.Set(betaPolicyFilterSetKey, tc.drop)
|
|
}
|
|
account := &Account{ID: 701, Platform: PlatformAnthropic, Type: AccountTypeOAuth,
|
|
Credentials: map[string]any{"access_token": "test-token"}, Status: StatusActive, Schedulable: true}
|
|
body := []byte(`{"model":"claude-opus-5-5","max_tokens":1024,"messages":[{"role":"user","content":[{"type":"tool_addition","tool":{"type":"tool_reference","tool_name":"lookup"}},{"type":"text","text":"Use lookup"}]}]}`)
|
|
svc := &GatewayService{cfg: &config.Config{}}
|
|
req, _, err := svc.buildUpstreamRequest(context.Background(), c, account, body,
|
|
"test-token", "oauth", "claude-opus-5-5", false, true)
|
|
require.NoError(t, err)
|
|
out := readUpstreamBodyForTest(t, req)
|
|
require.JSONEq(t, gjson.GetBytes(body, "messages").Raw, gjson.GetBytes(out, "messages").Raw)
|
|
header := getHeaderRaw(req.Header, "anthropic-beta")
|
|
require.Equal(t, tc.wantMid, containsBetaToken(header, mid))
|
|
require.Equal(t, tc.wantInline, containsBetaToken(header, inline))
|
|
require.False(t, containsBetaToken(header, "unrelated-beta"))
|
|
require.False(t, containsBetaToken(header, inline+"-other"))
|
|
require.True(t, containsBetaToken(header, "oauth-2025-04-20"))
|
|
if tc.wantInline {
|
|
require.Equal(t, 1, strings.Count(header, inline))
|
|
}
|
|
})
|
|
}
|
|
}
|