mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-24 23:43:46 +08:00
erio
7c0de8b1a1
Update model mapping tests, rate limit pre-check tests, thinking suffix tests, and whitelist tests to reflect the new default mapping and simplified applyThinkingModelSuffix behavior.
69 lines
1.9 KiB
Go
69 lines
1.9 KiB
Go
//go:build unit
|
||
|
||
package service
|
||
|
||
import (
|
||
"testing"
|
||
)
|
||
|
||
func TestApplyThinkingModelSuffix(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
mappedModel string
|
||
thinkingEnabled bool
|
||
expected string
|
||
}{
|
||
// Thinking 未开启：保持原样
|
||
{
|
||
name: "thinking disabled - claude-sonnet-4-5 unchanged",
|
||
mappedModel: "claude-sonnet-4-5",
|
||
thinkingEnabled: false,
|
||
expected: "claude-sonnet-4-5",
|
||
},
|
||
{
|
||
name: "thinking disabled - other model unchanged",
|
||
mappedModel: "claude-opus-4-6-thinking",
|
||
thinkingEnabled: false,
|
||
expected: "claude-opus-4-6-thinking",
|
||
},
|
||
|
||
// Thinking 开启 + claude-sonnet-4-5：Sonnet 4.5 已统一映射到 4.6，不再特殊处理
|
||
{
|
||
name: "thinking enabled - claude-sonnet-4-5 unchanged (mapped to 4.6 elsewhere)",
|
||
mappedModel: "claude-sonnet-4-5",
|
||
thinkingEnabled: true,
|
||
expected: "claude-sonnet-4-5",
|
||
},
|
||
|
||
// Thinking 开启 + 其他模型：保持原样
|
||
{
|
||
name: "thinking enabled - claude-sonnet-4-5-thinking unchanged",
|
||
mappedModel: "claude-sonnet-4-5-thinking",
|
||
thinkingEnabled: true,
|
||
expected: "claude-sonnet-4-5-thinking",
|
||
},
|
||
{
|
||
name: "thinking enabled - claude-opus-4-6-thinking unchanged",
|
||
mappedModel: "claude-opus-4-6-thinking",
|
||
thinkingEnabled: true,
|
||
expected: "claude-opus-4-6-thinking",
|
||
},
|
||
{
|
||
name: "thinking enabled - gemini model unchanged",
|
||
mappedModel: "gemini-3-flash",
|
||
thinkingEnabled: true,
|
||
expected: "gemini-3-flash",
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
result := applyThinkingModelSuffix(tt.mappedModel, tt.thinkingEnabled)
|
||
if result != tt.expected {
|
||
t.Errorf("applyThinkingModelSuffix(%q, %v) = %q, want %q",
|
||
tt.mappedModel, tt.thinkingEnabled, result, tt.expected)
|
||
}
|
||
})
|
||
}
|
||
}
|