ed8a9d975b
H1: incrementUsageBillingAccountQuota now uses shared dailyExpiredExpr/weeklyExpiredExpr
constants (supporting fixed reset mode) instead of hardcoded '24 hours'/'168 hours'
H4: public settings endpoint now maps balance_low_notify_recharge_url
H6: GetWebSearchEmulationMode tolerates legacy bool values (true→enabled)
H7: UpdatePlan validates non-nil patch fields (rejects negative price, empty name, etc.)
H8: UsageTable accountBilled() helper with total_cost ?? 0 null guard
H9: AdminUsageLog TS type adds channel_id + billing_tier
M2: account.go "fixed" literals replaced with thresholdTypeFixed constant
M13: SystemSettings TS type adds web_search_emulation_enabled
UI: QuotaLimitCard title labels now use flex-1 to align with flex-1 input boxes
106 lines
3.0 KiB
Go
106 lines
3.0 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetWebSearchEmulationMode_Enabled(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformAnthropic,
|
|
Type: AccountTypeAPIKey,
|
|
Extra: map[string]any{featureKeyWebSearchEmulation: "enabled"},
|
|
}
|
|
require.Equal(t, WebSearchModeEnabled, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_Disabled(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformAnthropic,
|
|
Type: AccountTypeAPIKey,
|
|
Extra: map[string]any{featureKeyWebSearchEmulation: "disabled"},
|
|
}
|
|
require.Equal(t, WebSearchModeDisabled, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_Default(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformAnthropic,
|
|
Type: AccountTypeAPIKey,
|
|
Extra: map[string]any{featureKeyWebSearchEmulation: "default"},
|
|
}
|
|
require.Equal(t, WebSearchModeDefault, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_UnknownString(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformAnthropic,
|
|
Type: AccountTypeAPIKey,
|
|
Extra: map[string]any{featureKeyWebSearchEmulation: "unknown"},
|
|
}
|
|
require.Equal(t, WebSearchModeDefault, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_OldBoolTrue(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformAnthropic,
|
|
Type: AccountTypeAPIKey,
|
|
Extra: map[string]any{featureKeyWebSearchEmulation: true},
|
|
}
|
|
// bool true → tolerant fallback → enabled (not default)
|
|
require.Equal(t, WebSearchModeEnabled, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_OldBoolFalse(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformAnthropic,
|
|
Type: AccountTypeAPIKey,
|
|
Extra: map[string]any{featureKeyWebSearchEmulation: false},
|
|
}
|
|
require.Equal(t, WebSearchModeDefault, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_NilAccount(t *testing.T) {
|
|
var a *Account
|
|
require.Equal(t, WebSearchModeDefault, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_NilExtra(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformAnthropic,
|
|
Type: AccountTypeAPIKey,
|
|
Extra: nil,
|
|
}
|
|
require.Equal(t, WebSearchModeDefault, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_MissingField(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformAnthropic,
|
|
Type: AccountTypeAPIKey,
|
|
Extra: map[string]any{},
|
|
}
|
|
require.Equal(t, WebSearchModeDefault, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_NonAnthropicPlatform(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformOpenAI,
|
|
Type: AccountTypeAPIKey,
|
|
Extra: map[string]any{featureKeyWebSearchEmulation: "enabled"},
|
|
}
|
|
require.Equal(t, WebSearchModeDefault, a.GetWebSearchEmulationMode())
|
|
}
|
|
|
|
func TestGetWebSearchEmulationMode_NonAPIKeyType(t *testing.T) {
|
|
a := &Account{
|
|
Platform: PlatformAnthropic,
|
|
Type: AccountTypeOAuth,
|
|
Extra: map[string]any{featureKeyWebSearchEmulation: "enabled"},
|
|
}
|
|
require.Equal(t, WebSearchModeDefault, a.GetWebSearchEmulationMode())
|
|
}
|