From 100d9d2034d581f05cd84acd935ebcfaad431f1f Mon Sep 17 00:00:00 2001 From: yangjianbo Date: Mon, 29 Dec 2025 14:29:51 +0800 Subject: [PATCH] =?UTF-8?q?test(=E6=9C=8D=E5=8A=A1=E5=B1=82):=20=E6=B7=BB?= =?UTF-8?q?=E5=8A=A0=E5=88=A0=E9=99=A4=E6=96=B9=E6=B3=95=E7=9A=84=E5=8D=95?= =?UTF-8?q?=E5=85=83=E6=B5=8B=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增 AccountService.Delete 方法的 4 个测试用例 - 新增 ApiKeyService.Delete 方法的 4 个测试用例 - 使用 stub 模式隔离数据库依赖，验证权限检查和缓存清理逻辑 - 添加详细的中文注释说明测试设计和预期行为 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 --- .../service/account_service_delete_test.go | 207 +++++++++++++++++ .../service/api_key_service_delete_test.go | 208 ++++++++++++++++++ 2 files changed, 415 insertions(+) create mode 100644 backend/internal/service/account_service_delete_test.go create mode 100644 backend/internal/service/api_key_service_delete_test.go diff --git a/backend/internal/service/account_service_delete_test.go b/backend/internal/service/account_service_delete_test.go new file mode 100644 index 00000000..93888bf5 --- /dev/null +++ b/backend/internal/service/account_service_delete_test.go @@ -0,0 +1,207 @@ +//go:build unit + +// 账号服务删除方法的单元测试 +// 测试 AccountService.Delete 方法在各种场景下的行为 + +package service + +import ( + "context" + "errors" + "testing" + "time" + + "github.com/Wei-Shaw/sub2api/internal/pkg/pagination" + "github.com/stretchr/testify/require" +) + +// accountRepoStub 是 AccountRepository 接口的测试桩实现。 +// 用于隔离测试 AccountService.Delete 方法，避免依赖真实数据库。 +// +// 设计说明： +// - exists: 模拟 ExistsByID 返回的存在性结果 +// - existsErr: 模拟 ExistsByID 返回的错误 +// - deleteErr: 模拟 Delete 返回的错误 +// - deletedIDs: 记录被调用删除的账号 ID，用于断言验证 +type accountRepoStub struct { + exists bool // ExistsByID 的返回值 + existsErr error // ExistsByID 的错误返回值 + deleteErr error // Delete 的错误返回值 + deletedIDs []int64 // 记录已删除的账号 ID 列表 +} + +// 以下方法在本测试中不应被调用，使用 panic 确保测试失败时能快速定位问题 + +func (s *accountRepoStub) Create(ctx context.Context, account *Account) error { + panic("unexpected Create call") +} + +func (s *accountRepoStub) GetByID(ctx context.Context, id int64) (*Account, error) { + panic("unexpected GetByID call") +} + +// ExistsByID 返回预设的存在性检查结果。 +// 这是 Delete 方法调用的第一个仓储方法，用于验证账号是否存在。 +func (s *accountRepoStub) ExistsByID(ctx context.Context, id int64) (bool, error) { + return s.exists, s.existsErr +} + +func (s *accountRepoStub) GetByCRSAccountID(ctx context.Context, crsAccountID string) (*Account, error) { + panic("unexpected GetByCRSAccountID call") +} + +func (s *accountRepoStub) Update(ctx context.Context, account *Account) error { + panic("unexpected Update call") +} + +// Delete 记录被删除的账号 ID 并返回预设的错误。 +// 通过 deletedIDs 可以验证删除操作是否被正确调用。 +func (s *accountRepoStub) Delete(ctx context.Context, id int64) error { + s.deletedIDs = append(s.deletedIDs, id) + return s.deleteErr +} + +// 以下是接口要求实现但本测试不关心的方法 + +func (s *accountRepoStub) List(ctx context.Context, params pagination.PaginationParams) ([]Account, *pagination.PaginationResult, error) { + panic("unexpected List call") +} + +func (s *accountRepoStub) ListWithFilters(ctx context.Context, params pagination.PaginationParams, platform, accountType, status, search string) ([]Account, *pagination.PaginationResult, error) { + panic("unexpected ListWithFilters call") +} + +func (s *accountRepoStub) ListByGroup(ctx context.Context, groupID int64) ([]Account, error) { + panic("unexpected ListByGroup call") +} + +func (s *accountRepoStub) ListActive(ctx context.Context) ([]Account, error) { + panic("unexpected ListActive call") +} + +func (s *accountRepoStub) ListByPlatform(ctx context.Context, platform string) ([]Account, error) { + panic("unexpected ListByPlatform call") +} + +func (s *accountRepoStub) UpdateLastUsed(ctx context.Context, id int64) error { + panic("unexpected UpdateLastUsed call") +} + +func (s *accountRepoStub) BatchUpdateLastUsed(ctx context.Context, updates map[int64]time.Time) error { + panic("unexpected BatchUpdateLastUsed call") +} + +func (s *accountRepoStub) SetError(ctx context.Context, id int64, errorMsg string) error { + panic("unexpected SetError call") +} + +func (s *accountRepoStub) SetSchedulable(ctx context.Context, id int64, schedulable bool) error { + panic("unexpected SetSchedulable call") +} + +func (s *accountRepoStub) BindGroups(ctx context.Context, accountID int64, groupIDs []int64) error { + panic("unexpected BindGroups call") +} + +func (s *accountRepoStub) ListSchedulable(ctx context.Context) ([]Account, error) { + panic("unexpected ListSchedulable call") +} + +func (s *accountRepoStub) ListSchedulableByGroupID(ctx context.Context, groupID int64) ([]Account, error) { + panic("unexpected ListSchedulableByGroupID call") +} + +func (s *accountRepoStub) ListSchedulableByPlatform(ctx context.Context, platform string) ([]Account, error) { + panic("unexpected ListSchedulableByPlatform call") +} + +func (s *accountRepoStub) ListSchedulableByGroupIDAndPlatform(ctx context.Context, groupID int64, platform string) ([]Account, error) { + panic("unexpected ListSchedulableByGroupIDAndPlatform call") +} + +func (s *accountRepoStub) SetRateLimited(ctx context.Context, id int64, resetAt time.Time) error { + panic("unexpected SetRateLimited call") +} + +func (s *accountRepoStub) SetOverloaded(ctx context.Context, id int64, until time.Time) error { + panic("unexpected SetOverloaded call") +} + +func (s *accountRepoStub) ClearRateLimit(ctx context.Context, id int64) error { + panic("unexpected ClearRateLimit call") +} + +func (s *accountRepoStub) UpdateSessionWindow(ctx context.Context, id int64, start, end *time.Time, status string) error { + panic("unexpected UpdateSessionWindow call") +} + +func (s *accountRepoStub) UpdateExtra(ctx context.Context, id int64, updates map[string]any) error { + panic("unexpected UpdateExtra call") +} + +func (s *accountRepoStub) BulkUpdate(ctx context.Context, ids []int64, updates AccountBulkUpdate) (int64, error) { + panic("unexpected BulkUpdate call") +} + +// TestAccountService_Delete_NotFound 测试删除不存在的账号时返回正确的错误。 +// 预期行为： +// - ExistsByID 返回 false（账号不存在） +// - 返回 ErrAccountNotFound 错误 +// - Delete 方法不被调用（deletedIDs 为空） +func TestAccountService_Delete_NotFound(t *testing.T) { + repo := &accountRepoStub{exists: false} + svc := &AccountService{accountRepo: repo} + + err := svc.Delete(context.Background(), 55) + require.ErrorIs(t, err, ErrAccountNotFound) + require.Empty(t, repo.deletedIDs) // 验证删除操作未被调用 +} + +// TestAccountService_Delete_CheckError 测试存在性检查失败时的错误处理。 +// 预期行为： +// - ExistsByID 返回数据库错误 +// - 返回包含 "check account" 的错误信息 +// - Delete 方法不被调用 +func TestAccountService_Delete_CheckError(t *testing.T) { + repo := &accountRepoStub{existsErr: errors.New("db down")} + svc := &AccountService{accountRepo: repo} + + err := svc.Delete(context.Background(), 55) + require.Error(t, err) + require.ErrorContains(t, err, "check account") // 验证错误信息包含上下文 + require.Empty(t, repo.deletedIDs) +} + +// TestAccountService_Delete_DeleteError 测试删除操作失败时的错误处理。 +// 预期行为： +// - ExistsByID 返回 true（账号存在） +// - Delete 被调用但返回错误 +// - 返回包含 "delete account" 的错误信息 +// - deletedIDs 记录了尝试删除的 ID +func TestAccountService_Delete_DeleteError(t *testing.T) { + repo := &accountRepoStub{ + exists: true, + deleteErr: errors.New("delete failed"), + } + svc := &AccountService{accountRepo: repo} + + err := svc.Delete(context.Background(), 55) + require.Error(t, err) + require.ErrorContains(t, err, "delete account") + require.Equal(t, []int64{55}, repo.deletedIDs) // 验证删除操作被调用 +} + +// TestAccountService_Delete_Success 测试删除操作成功的场景。 +// 预期行为： +// - ExistsByID 返回 true（账号存在） +// - Delete 成功执行 +// - 返回 nil 错误 +// - deletedIDs 记录了被删除的 ID +func TestAccountService_Delete_Success(t *testing.T) { + repo := &accountRepoStub{exists: true} + svc := &AccountService{accountRepo: repo} + + err := svc.Delete(context.Background(), 55) + require.NoError(t, err) + require.Equal(t, []int64{55}, repo.deletedIDs) // 验证正确的 ID 被删除 +} diff --git a/backend/internal/service/api_key_service_delete_test.go b/backend/internal/service/api_key_service_delete_test.go new file mode 100644 index 00000000..a531d0b8 --- /dev/null +++ b/backend/internal/service/api_key_service_delete_test.go @@ -0,0 +1,208 @@ +//go:build unit + +// API Key 服务删除方法的单元测试 +// 测试 ApiKeyService.Delete 方法在各种场景下的行为， +// 包括权限验证、缓存清理和错误处理 + +package service + +import ( + "context" + "errors" + "testing" + "time" + + "github.com/Wei-Shaw/sub2api/internal/pkg/pagination" + "github.com/stretchr/testify/require" +) + +// apiKeyRepoStub 是 ApiKeyRepository 接口的测试桩实现。 +// 用于隔离测试 ApiKeyService.Delete 方法，避免依赖真实数据库。 +// +// 设计说明： +// - ownerID: 模拟 GetOwnerID 返回的所有者 ID +// - ownerErr: 模拟 GetOwnerID 返回的错误（如 ErrApiKeyNotFound） +// - deleteErr: 模拟 Delete 返回的错误 +// - deletedIDs: 记录被调用删除的 API Key ID，用于断言验证 +type apiKeyRepoStub struct { + ownerID int64 // GetOwnerID 的返回值 + ownerErr error // GetOwnerID 的错误返回值 + deleteErr error // Delete 的错误返回值 + deletedIDs []int64 // 记录已删除的 API Key ID 列表 +} + +// 以下方法在本测试中不应被调用，使用 panic 确保测试失败时能快速定位问题 + +func (s *apiKeyRepoStub) Create(ctx context.Context, key *ApiKey) error { + panic("unexpected Create call") +} + +func (s *apiKeyRepoStub) GetByID(ctx context.Context, id int64) (*ApiKey, error) { + panic("unexpected GetByID call") +} + +// GetOwnerID 返回预设的所有者 ID 或错误。 +// 这是 Delete 方法调用的第一个仓储方法，用于验证调用者是否为 API Key 的所有者。 +func (s *apiKeyRepoStub) GetOwnerID(ctx context.Context, id int64) (int64, error) { + return s.ownerID, s.ownerErr +} + +func (s *apiKeyRepoStub) GetByKey(ctx context.Context, key string) (*ApiKey, error) { + panic("unexpected GetByKey call") +} + +func (s *apiKeyRepoStub) Update(ctx context.Context, key *ApiKey) error { + panic("unexpected Update call") +} + +// Delete 记录被删除的 API Key ID 并返回预设的错误。 +// 通过 deletedIDs 可以验证删除操作是否被正确调用。 +func (s *apiKeyRepoStub) Delete(ctx context.Context, id int64) error { + s.deletedIDs = append(s.deletedIDs, id) + return s.deleteErr +} + +// 以下是接口要求实现但本测试不关心的方法 + +func (s *apiKeyRepoStub) ListByUserID(ctx context.Context, userID int64, params pagination.PaginationParams) ([]ApiKey, *pagination.PaginationResult, error) { + panic("unexpected ListByUserID call") +} + +func (s *apiKeyRepoStub) VerifyOwnership(ctx context.Context, userID int64, apiKeyIDs []int64) ([]int64, error) { + panic("unexpected VerifyOwnership call") +} + +func (s *apiKeyRepoStub) CountByUserID(ctx context.Context, userID int64) (int64, error) { + panic("unexpected CountByUserID call") +} + +func (s *apiKeyRepoStub) ExistsByKey(ctx context.Context, key string) (bool, error) { + panic("unexpected ExistsByKey call") +} + +func (s *apiKeyRepoStub) ListByGroupID(ctx context.Context, groupID int64, params pagination.PaginationParams) ([]ApiKey, *pagination.PaginationResult, error) { + panic("unexpected ListByGroupID call") +} + +func (s *apiKeyRepoStub) SearchApiKeys(ctx context.Context, userID int64, keyword string, limit int) ([]ApiKey, error) { + panic("unexpected SearchApiKeys call") +} + +func (s *apiKeyRepoStub) ClearGroupIDByGroupID(ctx context.Context, groupID int64) (int64, error) { + panic("unexpected ClearGroupIDByGroupID call") +} + +func (s *apiKeyRepoStub) CountByGroupID(ctx context.Context, groupID int64) (int64, error) { + panic("unexpected CountByGroupID call") +} + +// apiKeyCacheStub 是 ApiKeyCache 接口的测试桩实现。 +// 用于验证删除操作时缓存清理逻辑是否被正确调用。 +// +// 设计说明： +// - invalidated: 记录被清除缓存的用户 ID 列表 +type apiKeyCacheStub struct { + invalidated []int64 // 记录调用 DeleteCreateAttemptCount 时传入的用户 ID +} + +// GetCreateAttemptCount 返回 0，表示用户未超过创建次数限制 +func (s *apiKeyCacheStub) GetCreateAttemptCount(ctx context.Context, userID int64) (int, error) { + return 0, nil +} + +// IncrementCreateAttemptCount 空实现，本测试不验证此行为 +func (s *apiKeyCacheStub) IncrementCreateAttemptCount(ctx context.Context, userID int64) error { + return nil +} + +// DeleteCreateAttemptCount 记录被清除缓存的用户 ID。 +// 删除 API Key 时会调用此方法清除用户的创建尝试计数缓存。 +func (s *apiKeyCacheStub) DeleteCreateAttemptCount(ctx context.Context, userID int64) error { + s.invalidated = append(s.invalidated, userID) + return nil +} + +// IncrementDailyUsage 空实现，本测试不验证此行为 +func (s *apiKeyCacheStub) IncrementDailyUsage(ctx context.Context, apiKey string) error { + return nil +} + +// SetDailyUsageExpiry 空实现，本测试不验证此行为 +func (s *apiKeyCacheStub) SetDailyUsageExpiry(ctx context.Context, apiKey string, ttl time.Duration) error { + return nil +} + +// TestApiKeyService_Delete_OwnerMismatch 测试非所有者尝试删除时返回权限错误。 +// 预期行为： +// - GetOwnerID 返回所有者 ID 为 1 +// - 调用者 userID 为 2（不匹配） +// - 返回 ErrInsufficientPerms 错误 +// - Delete 方法不被调用 +// - 缓存不被清除 +func TestApiKeyService_Delete_OwnerMismatch(t *testing.T) { + repo := &apiKeyRepoStub{ownerID: 1} + cache := &apiKeyCacheStub{} + svc := &ApiKeyService{apiKeyRepo: repo, cache: cache} + + err := svc.Delete(context.Background(), 10, 2) // API Key ID=10, 调用者 userID=2 + require.ErrorIs(t, err, ErrInsufficientPerms) + require.Empty(t, repo.deletedIDs) // 验证删除操作未被调用 + require.Empty(t, cache.invalidated) // 验证缓存未被清除 +} + +// TestApiKeyService_Delete_Success 测试所有者成功删除 API Key 的场景。 +// 预期行为： +// - GetOwnerID 返回所有者 ID 为 7 +// - 调用者 userID 为 7（匹配） +// - Delete 成功执行 +// - 缓存被正确清除（使用 ownerID） +// - 返回 nil 错误 +func TestApiKeyService_Delete_Success(t *testing.T) { + repo := &apiKeyRepoStub{ownerID: 7} + cache := &apiKeyCacheStub{} + svc := &ApiKeyService{apiKeyRepo: repo, cache: cache} + + err := svc.Delete(context.Background(), 42, 7) // API Key ID=42, 调用者 userID=7 + require.NoError(t, err) + require.Equal(t, []int64{42}, repo.deletedIDs) // 验证正确的 API Key 被删除 + require.Equal(t, []int64{7}, cache.invalidated) // 验证所有者的缓存被清除 +} + +// TestApiKeyService_Delete_NotFound 测试删除不存在的 API Key 时返回正确的错误。 +// 预期行为： +// - GetOwnerID 返回 ErrApiKeyNotFound 错误 +// - 返回 ErrApiKeyNotFound 错误（被 fmt.Errorf 包装） +// - Delete 方法不被调用 +// - 缓存不被清除 +func TestApiKeyService_Delete_NotFound(t *testing.T) { + repo := &apiKeyRepoStub{ownerErr: ErrApiKeyNotFound} + cache := &apiKeyCacheStub{} + svc := &ApiKeyService{apiKeyRepo: repo, cache: cache} + + err := svc.Delete(context.Background(), 99, 1) + require.ErrorIs(t, err, ErrApiKeyNotFound) + require.Empty(t, repo.deletedIDs) + require.Empty(t, cache.invalidated) +} + +// TestApiKeyService_Delete_DeleteFails 测试删除操作失败时的错误处理。 +// 预期行为： +// - GetOwnerID 返回正确的所有者 ID +// - 所有权验证通过 +// - 缓存被清除（在删除之前） +// - Delete 被调用但返回错误 +// - 返回包含 "delete api key" 的错误信息 +func TestApiKeyService_Delete_DeleteFails(t *testing.T) { + repo := &apiKeyRepoStub{ + ownerID: 3, + deleteErr: errors.New("delete failed"), + } + cache := &apiKeyCacheStub{} + svc := &ApiKeyService{apiKeyRepo: repo, cache: cache} + + err := svc.Delete(context.Background(), 3, 3) // API Key ID=3, 调用者 userID=3 + require.Error(t, err) + require.ErrorContains(t, err, "delete api key") + require.Equal(t, []int64{3}, repo.deletedIDs) // 验证删除操作被调用 + require.Equal(t, []int64{3}, cache.invalidated) // 验证缓存已被清除（即使删除失败） +}