Files
sub2api/backend/internal/service/usage_log_cyber_test.go
T
李建琦 6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
2026-08-21 18:30:13 +08:00

26 lines
791 B
Go

package service
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestRequestTypeCyberBlocked(t *testing.T) {
require.True(t, RequestTypeCyberBlocked.IsValid())
require.Equal(t, "cyber", RequestTypeCyberBlocked.String())
rt, err := ParseUsageRequestType("cyber")
require.NoError(t, err)
require.Equal(t, RequestTypeCyberBlocked, rt)
// 显式 cyber 被 EffectiveRequestType 保留(不被 legacy 推导覆盖)
u := &UsageLog{RequestType: RequestTypeCyberBlocked, Stream: true}
require.Equal(t, RequestTypeCyberBlocked, u.EffectiveRequestType())
// Sync 保留 cyber 且不覆盖真实 stream
u.SyncRequestTypeAndLegacyFields()
require.Equal(t, RequestTypeCyberBlocked, u.RequestType)
require.True(t, u.Stream, "cyber 不应覆盖真实 stream 字段")
}