Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/setting_panel_rate_limit_test.go

@@ -0,0 +1,182 @@
+package service
+
+import (
+	"context"
+	"sync"
+	"testing"
+
+	"github.com/Wei-Shaw/sub2api/internal/config"
+	"github.com/stretchr/testify/require"
+)
+
+type panelRateLimitSettingRepo struct {
+	mu            sync.Mutex
+	values        map[string]string
+	getValueErr   error
+	getValueCalls int
+}
+
+func (r *panelRateLimitSettingRepo) Get(_ context.Context, key string) (*Setting, error) {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	value, ok := r.values[key]
+	if !ok {
+		return nil, ErrSettingNotFound
+	}
+	return &Setting{Key: key, Value: value}, nil
+}
+
+func (r *panelRateLimitSettingRepo) GetValue(_ context.Context, key string) (string, error) {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	r.getValueCalls++
+	if r.getValueErr != nil {
+		return "", r.getValueErr
+	}
+	value, ok := r.values[key]
+	if !ok {
+		return "", ErrSettingNotFound
+	}
+	return value, nil
+}
+
+func (r *panelRateLimitSettingRepo) Set(_ context.Context, key, value string) error {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	if r.values == nil {
+		r.values = make(map[string]string)
+	}
+	r.values[key] = value
+	return nil
+}
+
+func (r *panelRateLimitSettingRepo) GetMultiple(_ context.Context, keys []string) (map[string]string, error) {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	out := make(map[string]string, len(keys))
+	for _, key := range keys {
+		if value, ok := r.values[key]; ok {
+			out[key] = value
+		}
+	}
+	return out, nil
+}
+
+func (r *panelRateLimitSettingRepo) SetMultiple(_ context.Context, settings map[string]string) error {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	if r.values == nil {
+		r.values = make(map[string]string)
+	}
+	for key, value := range settings {
+		r.values[key] = value
+	}
+	return nil
+}
+
+func (r *panelRateLimitSettingRepo) GetAll(_ context.Context) (map[string]string, error) {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	out := make(map[string]string, len(r.values))
+	for key, value := range r.values {
+		out[key] = value
+	}
+	return out, nil
+}
+
+func (r *panelRateLimitSettingRepo) Delete(_ context.Context, key string) error {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	delete(r.values, key)
+	return nil
+}
+
+func newPanelRateLimitTestService(repo SettingRepository) *SettingService {
+	return NewSettingService(repo, &config.Config{})
+}
+
+func TestGetPanelRateLimitSettingsDefaults(t *testing.T) {
+	svc := newPanelRateLimitTestService(&panelRateLimitSettingRepo{})
+
+	settings, err := svc.GetPanelRateLimitSettings(context.Background())
+	require.NoError(t, err)
+	require.Equal(t, DefaultPanelRateLimitSettings(), settings)
+}
+
+func TestGetPanelRateLimitSettingsInvalidJSONFallsBack(t *testing.T) {
+	repo := &panelRateLimitSettingRepo{values: map[string]string{
+		SettingKeyPanelRateLimitSettings: "{not-json",
+	}}
+	svc := newPanelRateLimitTestService(repo)
+
+	settings, err := svc.GetPanelRateLimitSettings(context.Background())
+	require.NoError(t, err)
+	require.Equal(t, DefaultPanelRateLimitSettings(), settings)
+}
+
+func TestGetPanelRateLimitSettingsNormalizesValues(t *testing.T) {
+	repo := &panelRateLimitSettingRepo{values: map[string]string{
+		SettingKeyPanelRateLimitSettings: `{"enabled":true,"user_rpm":-5,"heavy_rpm":999999999,"exempt_admin":false,"public_ip_rpm":10}`,
+	}}
+	svc := newPanelRateLimitTestService(repo)
+
+	settings, err := svc.GetPanelRateLimitSettings(context.Background())
+	require.NoError(t, err)
+	require.True(t, settings.Enabled)
+	require.Equal(t, 0, settings.UserRPM)
+	require.Equal(t, panelRateLimitRPMMax, settings.HeavyRPM)
+	require.Equal(t, 10, settings.PublicIPRPM)
+	require.False(t, settings.ExemptAdmin)
+}
+
+func TestSetPanelRateLimitSettingsValidation(t *testing.T) {
+	svc := newPanelRateLimitTestService(&panelRateLimitSettingRepo{})
+
+	require.Error(t, svc.SetPanelRateLimitSettings(context.Background(), nil))
+	require.Error(t, svc.SetPanelRateLimitSettings(context.Background(), &PanelRateLimitSettings{UserRPM: -1}))
+	require.Error(t, svc.SetPanelRateLimitSettings(context.Background(), &PanelRateLimitSettings{HeavyRPM: panelRateLimitRPMMax + 1}))
+}
+
+func TestSetPanelRateLimitSettingsRoundTripAndCacheRefresh(t *testing.T) {
+	repo := &panelRateLimitSettingRepo{}
+	svc := newPanelRateLimitTestService(repo)
+
+	// 先填充缓存（默认值）
+	cached := svc.GetPanelRateLimitSettingsCached(context.Background())
+	require.Equal(t, *DefaultPanelRateLimitSettings(), cached)
+
+	want := &PanelRateLimitSettings{
+		Enabled:     true,
+		UserRPM:     120,
+		HeavyRPM:    30,
+		ExemptAdmin: false,
+		PublicIPRPM: 60,
+	}
+	require.NoError(t, svc.SetPanelRateLimitSettings(context.Background(), want))
+
+	// 写入后无需等待 TTL，缓存立即反映新值
+	cached = svc.GetPanelRateLimitSettingsCached(context.Background())
+	require.Equal(t, *want, cached)
+
+	// DB 中持久化的值可直读
+	stored, err := svc.GetPanelRateLimitSettings(context.Background())
+	require.NoError(t, err)
+	require.Equal(t, want, stored)
+}
+
+func TestGetPanelRateLimitSettingsCachedAvoidsRepeatedDBReads(t *testing.T) {
+	repo := &panelRateLimitSettingRepo{values: map[string]string{
+		SettingKeyPanelRateLimitSettings: `{"enabled":true,"user_rpm":100,"heavy_rpm":20,"exempt_admin":true,"public_ip_rpm":50}`,
+	}}
+	svc := newPanelRateLimitTestService(repo)
+
+	for i := 0; i < 5; i++ {
+		settings := svc.GetPanelRateLimitSettingsCached(context.Background())
+		require.Equal(t, 100, settings.UserRPM)
+	}
+
+	repo.mu.Lock()
+	calls := repo.getValueCalls
+	repo.mu.Unlock()
+	require.Equal(t, 1, calls, "TTL 内应只读一次 DB")
+}