Files
sub2api/backend/internal/service/api_key_service_validation_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

47 lines
1.2 KiB
Go

//go:build unit
package service
import (
"math"
"testing"
"github.com/stretchr/testify/require"
)
func TestValidateCreateAPIKeyRequestNumericLimits(t *testing.T) {
positiveExpiry := 1
require.NoError(t, validateCreateAPIKeyRequest(CreateAPIKeyRequest{
Quota: 1e100, RateLimit5h: 1e100, ExpiresInDays: &positiveExpiry,
}))
require.NoError(t, validateCreateAPIKeyRequest(CreateAPIKeyRequest{}))
invalidExpiry := 0
tests := []CreateAPIKeyRequest{
{Quota: -1},
{Quota: math.NaN()},
{Quota: math.Inf(1)},
{RateLimit5h: -1},
{RateLimit1d: math.NaN()},
{RateLimit7d: math.Inf(-1)},
{ExpiresInDays: &invalidExpiry},
}
for _, req := range tests {
require.Error(t, validateCreateAPIKeyRequest(req))
}
}
func TestValidateUpdateAPIKeyRequestNumericLimits(t *testing.T) {
zero, large, negative, nan, inf := 0.0, 1e100, -1.0, math.NaN(), math.Inf(1)
require.NoError(t, validateUpdateAPIKeyRequest(UpdateAPIKeyRequest{Quota: &zero, RateLimit7d: &large}))
for _, req := range []UpdateAPIKeyRequest{
{Quota: &negative},
{RateLimit5h: &nan},
{RateLimit1d: &inf},
{RateLimit7d: &negative},
} {
require.Error(t, validateUpdateAPIKeyRequest(req))
}
}