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

backend/internal/service/admin_service_batch_limits_test.go

@@ -0,0 +1,92 @@
+//go:build unit
+
+package service
+
+import (
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+type batchLimitsUserRepoStub struct {
+	*userRepoStub
+	calls       int
+	userIDs     []int64
+	concurrency *int
+	rpmLimit    *int
+	affected    int
+	err         error
+}
+
+func (s *batchLimitsUserRepoStub) BatchUpdateLimits(_ context.Context, userIDs []int64, concurrency, rpmLimit *int) (int, error) {
+	s.calls++
+	s.userIDs = append([]int64(nil), userIDs...)
+	s.concurrency = cloneBatchLimitValue(concurrency)
+	s.rpmLimit = cloneBatchLimitValue(rpmLimit)
+	return s.affected, s.err
+}
+
+func cloneBatchLimitValue(value *int) *int {
+	if value == nil {
+		return nil
+	}
+	cloned := *value
+	return &cloned
+}
+
+func TestAdminServiceBatchUpdateLimitsPassesOnlyProvidedFields(t *testing.T) {
+	concurrency := 0
+	repo := &batchLimitsUserRepoStub{
+		userRepoStub: &userRepoStub{},
+		affected:     2,
+	}
+	invalidator := &authCacheInvalidatorStub{}
+	service := &adminServiceImpl{userRepo: repo, authCacheInvalidator: invalidator}
+
+	affected, err := service.BatchUpdateLimits(
+		context.Background(),
+		[]int64{3, 0, 3, 7, -1},
+		&concurrency,
+		nil,
+	)
+
+	require.NoError(t, err)
+	require.Equal(t, 2, affected)
+	require.Equal(t, []int64{3, 7}, repo.userIDs)
+	require.Equal(t, pointerToInt(0), repo.concurrency)
+	require.Nil(t, repo.rpmLimit)
+	require.Equal(t, []int64{3, 7}, invalidator.userIDs)
+}
+
+func TestAdminServiceBatchUpdateLimitsDoesNotInvalidateCacheOnRepositoryError(t *testing.T) {
+	rpmLimit := 60
+	repo := &batchLimitsUserRepoStub{
+		userRepoStub: &userRepoStub{},
+		err:          errors.New("database unavailable"),
+	}
+	invalidator := &authCacheInvalidatorStub{}
+	service := &adminServiceImpl{userRepo: repo, authCacheInvalidator: invalidator}
+
+	affected, err := service.BatchUpdateLimits(context.Background(), []int64{1, 2}, nil, &rpmLimit)
+
+	require.EqualError(t, err, "database unavailable")
+	require.Zero(t, affected)
+	require.Empty(t, invalidator.userIDs)
+}
+
+func TestAdminServiceBatchUpdateLimitsRequiresAField(t *testing.T) {
+	repo := &batchLimitsUserRepoStub{userRepoStub: &userRepoStub{}}
+	service := &adminServiceImpl{userRepo: repo, authCacheInvalidator: &authCacheInvalidatorStub{}}
+
+	affected, err := service.BatchUpdateLimits(context.Background(), []int64{1}, nil, nil)
+
+	require.Error(t, err)
+	require.Zero(t, affected)
+	require.Zero(t, repo.calls)
+}
+
+func pointerToInt(value int) *int {
+	return &value
+}