admin_service_delete_test.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814
  1. //go:build unit
  2. package service
  3. import (
  4. "context"
  5. "errors"
  6. "testing"
  7. "time"
  8. "github.com/Wei-Shaw/sub2api/internal/pkg/pagination"
  9. "github.com/stretchr/testify/require"
  10. )
  11. type userRepoStub struct {
  12. user *User
  13. usersByID map[int64]*User
  14. getErr error
  15. createErr error
  16. deleteErr error
  17. exists bool
  18. existsErr error
  19. aliasExists bool
  20. aliasErr error
  21. guardedCreates int
  22. nextID int64
  23. created []*User
  24. updated []*User
  25. deletedIDs []int64
  26. usersByEmail map[string]*User
  27. getByEmailErr error
  28. getByEmailMisses int
  29. domainCounts map[string]int
  30. domainCountErr error
  31. domainLimitErr error
  32. domainLimitedCreates int
  33. }
  34. func (s *userRepoStub) CountUsersByEmailDomain(_ context.Context, domain string) (int, error) {
  35. if s.domainCountErr != nil {
  36. return 0, s.domainCountErr
  37. }
  38. return s.domainCounts[domain], nil
  39. }
  40. func (s *userRepoStub) CreateWithEmailAliasGuardAndDomainLimit(ctx context.Context, user *User, domain string) error {
  41. s.domainLimitedCreates++
  42. if s.domainLimitErr != nil {
  43. return s.domainLimitErr
  44. }
  45. if s.domainCounts[domain] > 0 {
  46. return ErrEmailDomainRegistrationLimit
  47. }
  48. return s.CreateWithEmailAliasGuard(ctx, user)
  49. }
  50. func (s *userRepoStub) Create(ctx context.Context, user *User) error {
  51. if s.createErr != nil {
  52. return s.createErr
  53. }
  54. if s.nextID != 0 && user.ID == 0 {
  55. user.ID = s.nextID
  56. }
  57. s.created = append(s.created, user)
  58. if s.usersByEmail == nil {
  59. s.usersByEmail = make(map[string]*User)
  60. }
  61. s.usersByEmail[user.Email] = user
  62. s.user = user
  63. return nil
  64. }
  65. func (s *userRepoStub) CreateWithEmailAliasGuard(ctx context.Context, user *User) error {
  66. s.guardedCreates++
  67. if s.aliasErr != nil {
  68. return s.aliasErr
  69. }
  70. if s.aliasExists {
  71. return ErrEmailExists
  72. }
  73. return s.Create(ctx, user)
  74. }
  75. func (s *userRepoStub) GetByID(ctx context.Context, id int64) (*User, error) {
  76. if s.getErr != nil {
  77. return nil, s.getErr
  78. }
  79. if s.usersByID != nil {
  80. if user, ok := s.usersByID[id]; ok {
  81. return user, nil
  82. }
  83. return nil, ErrUserNotFound
  84. }
  85. if s.user == nil {
  86. return nil, ErrUserNotFound
  87. }
  88. return s.user, nil
  89. }
  90. func (s *userRepoStub) GetByEmail(ctx context.Context, email string) (*User, error) {
  91. if s.getByEmailErr != nil {
  92. return nil, s.getByEmailErr
  93. }
  94. if s.getByEmailMisses > 0 {
  95. s.getByEmailMisses--
  96. return nil, ErrUserNotFound
  97. }
  98. if s.usersByEmail != nil {
  99. if user, ok := s.usersByEmail[email]; ok {
  100. return user, nil
  101. }
  102. }
  103. if s.user != nil && s.user.Email == email {
  104. return s.user, nil
  105. }
  106. return nil, ErrUserNotFound
  107. }
  108. func (s *userRepoStub) GetFirstAdmin(ctx context.Context) (*User, error) {
  109. panic("unexpected GetFirstAdmin call")
  110. }
  111. func (s *userRepoStub) Update(ctx context.Context, user *User, fields UserUpdateFields) error {
  112. s.updated = append(s.updated, user)
  113. if s.usersByEmail == nil {
  114. s.usersByEmail = make(map[string]*User)
  115. }
  116. s.usersByEmail[user.Email] = user
  117. s.user = user
  118. return nil
  119. }
  120. func (s *userRepoStub) Delete(ctx context.Context, id int64) error {
  121. s.deletedIDs = append(s.deletedIDs, id)
  122. return s.deleteErr
  123. }
  124. func (s *userRepoStub) GetUserAvatar(ctx context.Context, userID int64) (*UserAvatar, error) {
  125. panic("unexpected GetUserAvatar call")
  126. }
  127. func (s *userRepoStub) UpsertUserAvatar(ctx context.Context, userID int64, input UpsertUserAvatarInput) (*UserAvatar, error) {
  128. panic("unexpected UpsertUserAvatar call")
  129. }
  130. func (s *userRepoStub) DeleteUserAvatar(ctx context.Context, userID int64) error {
  131. panic("unexpected DeleteUserAvatar call")
  132. }
  133. func (s *userRepoStub) List(ctx context.Context, params pagination.PaginationParams) ([]User, *pagination.PaginationResult, error) {
  134. panic("unexpected List call")
  135. }
  136. func (s *userRepoStub) ListWithFilters(ctx context.Context, params pagination.PaginationParams, filters UserListFilters) ([]User, *pagination.PaginationResult, error) {
  137. panic("unexpected ListWithFilters call")
  138. }
  139. func (s *userRepoStub) GetLatestUsedAtByUserIDs(ctx context.Context, userIDs []int64) (map[int64]*time.Time, error) {
  140. panic("unexpected GetLatestUsedAtByUserIDs call")
  141. }
  142. func (s *userRepoStub) GetLatestUsedAtByUserID(ctx context.Context, userID int64) (*time.Time, error) {
  143. panic("unexpected GetLatestUsedAtByUserID call")
  144. }
  145. func (s *userRepoStub) UpdateUserLastActiveAt(ctx context.Context, userID int64, activeAt time.Time) error {
  146. panic("unexpected UpdateUserLastActiveAt call")
  147. }
  148. func (s *userRepoStub) UpdateBalance(ctx context.Context, id int64, amount float64) error {
  149. panic("unexpected UpdateBalance call")
  150. }
  151. func (s *userRepoStub) DeductBalance(ctx context.Context, id int64, amount float64) error {
  152. panic("unexpected DeductBalance call")
  153. }
  154. func (s *userRepoStub) AdjustBalance(ctx context.Context, id int64, delta float64) (BalanceChange, error) {
  155. panic("unexpected AdjustBalance call")
  156. }
  157. func (s *userRepoStub) SetBalance(ctx context.Context, id int64, value float64) (BalanceChange, error) {
  158. panic("unexpected SetBalance call")
  159. }
  160. func (s *userRepoStub) UpdateConcurrency(ctx context.Context, id int64, amount int) error {
  161. panic("unexpected UpdateConcurrency call")
  162. }
  163. func (s *userRepoStub) BatchSetConcurrency(context.Context, []int64, int) (int, error) { return 0, nil }
  164. func (s *userRepoStub) BatchAddConcurrency(context.Context, []int64, int) (int, error) { return 0, nil }
  165. func (s *userRepoStub) BatchUpdateLimits(context.Context, []int64, *int, *int) (int, error) {
  166. return 0, nil
  167. }
  168. func (s *userRepoStub) ExistsByEmail(ctx context.Context, email string) (bool, error) {
  169. if s.existsErr != nil {
  170. return false, s.existsErr
  171. }
  172. return s.exists, nil
  173. }
  174. func (s *userRepoStub) ExistsByEmailAlias(ctx context.Context, email string) (bool, error) {
  175. if s.aliasErr != nil {
  176. return false, s.aliasErr
  177. }
  178. return s.aliasExists, nil
  179. }
  180. func (s *userRepoStub) RemoveGroupFromAllowedGroups(ctx context.Context, groupID int64) (int64, error) {
  181. panic("unexpected RemoveGroupFromAllowedGroups call")
  182. }
  183. func (s *userRepoStub) RemoveGroupFromUserAllowedGroups(ctx context.Context, userID int64, groupID int64) error {
  184. panic("unexpected RemoveGroupFromUserAllowedGroups call")
  185. }
  186. func (s *userRepoStub) AddGroupToAllowedGroups(ctx context.Context, userID int64, groupID int64) error {
  187. panic("unexpected AddGroupToAllowedGroups call")
  188. }
  189. func (s *userRepoStub) ListUserAuthIdentities(ctx context.Context, userID int64) ([]UserAuthIdentityRecord, error) {
  190. panic("unexpected ListUserAuthIdentities call")
  191. }
  192. func (s *userRepoStub) UnbindUserAuthProvider(context.Context, int64, string) error {
  193. panic("unexpected UnbindUserAuthProvider call")
  194. }
  195. func (s *userRepoStub) UpdateTotpSecret(ctx context.Context, userID int64, encryptedSecret *string) error {
  196. panic("unexpected UpdateTotpSecret call")
  197. }
  198. func (s *userRepoStub) EnableTotp(ctx context.Context, userID int64) error {
  199. panic("unexpected EnableTotp call")
  200. }
  201. func (s *userRepoStub) DisableTotp(ctx context.Context, userID int64) error {
  202. panic("unexpected DisableTotp call")
  203. }
  204. func (s *userRepoStub) GetByIDIncludeDeleted(ctx context.Context, id int64) (*User, error) {
  205. return s.GetByID(ctx, id)
  206. }
  207. type groupRepoStub struct {
  208. affectedUserIDs []int64
  209. deleteErr error
  210. deleteCalls []int64
  211. guardedCalls []int64
  212. }
  213. func (s *groupRepoStub) Create(ctx context.Context, group *Group) error {
  214. panic("unexpected Create call")
  215. }
  216. func (s *groupRepoStub) GetByID(ctx context.Context, id int64) (*Group, error) {
  217. panic("unexpected GetByID call")
  218. }
  219. func (s *groupRepoStub) GetByIDLite(ctx context.Context, id int64) (*Group, error) {
  220. panic("unexpected GetByIDLite call")
  221. }
  222. func (s *groupRepoStub) Update(ctx context.Context, group *Group) error {
  223. panic("unexpected Update call")
  224. }
  225. func (s *groupRepoStub) Delete(ctx context.Context, id int64) error {
  226. panic("unexpected Delete call")
  227. }
  228. func (s *groupRepoStub) DeleteCascade(ctx context.Context, id int64) ([]int64, error) {
  229. s.deleteCalls = append(s.deleteCalls, id)
  230. return s.affectedUserIDs, s.deleteErr
  231. }
  232. func (s *groupRepoStub) DeleteCascadeIfEmpty(ctx context.Context, id int64) ([]int64, error) {
  233. s.guardedCalls = append(s.guardedCalls, id)
  234. return s.affectedUserIDs, s.deleteErr
  235. }
  236. func (s *groupRepoStub) List(ctx context.Context, params pagination.PaginationParams) ([]Group, *pagination.PaginationResult, error) {
  237. panic("unexpected List call")
  238. }
  239. func (s *groupRepoStub) ListWithFilters(ctx context.Context, params pagination.PaginationParams, platform, status, search string, isExclusive *bool) ([]Group, *pagination.PaginationResult, error) {
  240. panic("unexpected ListWithFilters call")
  241. }
  242. func (s *groupRepoStub) ListActive(ctx context.Context) ([]Group, error) {
  243. panic("unexpected ListActive call")
  244. }
  245. func (s *groupRepoStub) ListActiveByPlatform(ctx context.Context, platform string) ([]Group, error) {
  246. panic("unexpected ListActiveByPlatform call")
  247. }
  248. func (s *groupRepoStub) ExistsByName(ctx context.Context, name string) (bool, error) {
  249. panic("unexpected ExistsByName call")
  250. }
  251. func (s *groupRepoStub) GetAccountCount(ctx context.Context, groupID int64) (int64, int64, error) {
  252. panic("unexpected GetAccountCount call")
  253. }
  254. func (s *groupRepoStub) DeleteAccountGroupsByGroupID(ctx context.Context, groupID int64) (int64, error) {
  255. panic("unexpected DeleteAccountGroupsByGroupID call")
  256. }
  257. func (s *groupRepoStub) BindAccountsToGroup(ctx context.Context, groupID int64, accountIDs []int64) error {
  258. panic("unexpected BindAccountsToGroup call")
  259. }
  260. func (s *groupRepoStub) GetAccountIDsByGroupIDs(ctx context.Context, groupIDs []int64) ([]int64, error) {
  261. panic("unexpected GetAccountIDsByGroupIDs call")
  262. }
  263. func (s *groupRepoStub) UpdateSortOrders(ctx context.Context, updates []GroupSortOrderUpdate) error {
  264. return nil
  265. }
  266. type deleteGroupAPIKeyRepoStub struct {
  267. apiKeyRepoStubForGroupUpdate
  268. keys []string
  269. listErr error
  270. listGroupIDs []int64
  271. }
  272. func (s *deleteGroupAPIKeyRepoStub) ListKeysByGroupID(ctx context.Context, groupID int64) ([]string, error) {
  273. s.listGroupIDs = append(s.listGroupIDs, groupID)
  274. if s.listErr != nil {
  275. return nil, s.listErr
  276. }
  277. return s.keys, nil
  278. }
  279. type proxyRepoStub struct {
  280. deleteErr error
  281. countErr error
  282. accountCount int64
  283. deletedIDs []int64
  284. }
  285. func (s *proxyRepoStub) Create(ctx context.Context, proxy *Proxy) error {
  286. panic("unexpected Create call")
  287. }
  288. func (s *proxyRepoStub) GetByID(ctx context.Context, id int64) (*Proxy, error) {
  289. panic("unexpected GetByID call")
  290. }
  291. func (s *proxyRepoStub) ListByIDs(ctx context.Context, ids []int64) ([]Proxy, error) {
  292. panic("unexpected ListByIDs call")
  293. }
  294. func (s *proxyRepoStub) Update(ctx context.Context, proxy *Proxy) error {
  295. panic("unexpected Update call")
  296. }
  297. func (s *proxyRepoStub) Delete(ctx context.Context, id int64) error {
  298. s.deletedIDs = append(s.deletedIDs, id)
  299. return s.deleteErr
  300. }
  301. func (s *proxyRepoStub) List(ctx context.Context, params pagination.PaginationParams) ([]Proxy, *pagination.PaginationResult, error) {
  302. panic("unexpected List call")
  303. }
  304. func (s *proxyRepoStub) ListWithFilters(ctx context.Context, params pagination.PaginationParams, protocol, status, search string) ([]Proxy, *pagination.PaginationResult, error) {
  305. panic("unexpected ListWithFilters call")
  306. }
  307. func (s *proxyRepoStub) ListActive(ctx context.Context) ([]Proxy, error) {
  308. panic("unexpected ListActive call")
  309. }
  310. func (s *proxyRepoStub) ListActiveWithAccountCount(ctx context.Context) ([]ProxyWithAccountCount, error) {
  311. panic("unexpected ListActiveWithAccountCount call")
  312. }
  313. func (s *proxyRepoStub) ListWithFiltersAndAccountCount(ctx context.Context, params pagination.PaginationParams, protocol, status, search string) ([]ProxyWithAccountCount, *pagination.PaginationResult, error) {
  314. panic("unexpected ListWithFiltersAndAccountCount call")
  315. }
  316. func (s *proxyRepoStub) ExistsByHostPortAuth(ctx context.Context, host string, port int, username, password string) (bool, error) {
  317. panic("unexpected ExistsByHostPortAuth call")
  318. }
  319. func (s *proxyRepoStub) CountAccountsByProxyID(ctx context.Context, proxyID int64) (int64, error) {
  320. if s.countErr != nil {
  321. return 0, s.countErr
  322. }
  323. return s.accountCount, nil
  324. }
  325. func (s *proxyRepoStub) ListAccountSummariesByProxyID(ctx context.Context, proxyID int64) ([]ProxyAccountSummary, error) {
  326. panic("unexpected ListAccountSummariesByProxyID call")
  327. }
  328. func (s *proxyRepoStub) SweepExpiredProxies(_ context.Context, _ time.Time) (int64, error) {
  329. return 0, nil
  330. }
  331. func (s *proxyRepoStub) ListAllForFallback(_ context.Context) ([]Proxy, error) {
  332. return nil, nil
  333. }
  334. func (s *proxyRepoStub) CountExpired(_ context.Context) (int64, error) {
  335. return 0, nil
  336. }
  337. func (s *proxyRepoStub) CountExpiringSoon(_ context.Context, _ time.Time) (int64, error) {
  338. return 0, nil
  339. }
  340. type redeemRepoStub struct {
  341. deleteErrByID map[int64]error
  342. deletedIDs []int64
  343. batchUpdateIDs []int64
  344. batchUpdateFields RedeemCodeBatchUpdateFields
  345. batchUpdateResult int64
  346. batchUpdateErr error
  347. batchUpdateCalled bool
  348. }
  349. func (s *redeemRepoStub) Create(ctx context.Context, code *RedeemCode) error {
  350. panic("unexpected Create call")
  351. }
  352. func (s *redeemRepoStub) CreateBatch(ctx context.Context, codes []RedeemCode) error {
  353. panic("unexpected CreateBatch call")
  354. }
  355. func (s *redeemRepoStub) GetByID(ctx context.Context, id int64) (*RedeemCode, error) {
  356. panic("unexpected GetByID call")
  357. }
  358. func (s *redeemRepoStub) GetByCode(ctx context.Context, code string) (*RedeemCode, error) {
  359. panic("unexpected GetByCode call")
  360. }
  361. func (s *redeemRepoStub) Update(ctx context.Context, code *RedeemCode) error {
  362. panic("unexpected Update call")
  363. }
  364. func (s *redeemRepoStub) BatchUpdate(ctx context.Context, ids []int64, fields RedeemCodeBatchUpdateFields) (int64, error) {
  365. s.batchUpdateCalled = true
  366. s.batchUpdateIDs = append([]int64(nil), ids...)
  367. s.batchUpdateFields = fields
  368. if s.batchUpdateErr != nil {
  369. return 0, s.batchUpdateErr
  370. }
  371. if s.batchUpdateResult != 0 {
  372. return s.batchUpdateResult, nil
  373. }
  374. return int64(len(ids)), nil
  375. }
  376. func (s *redeemRepoStub) Delete(ctx context.Context, id int64) error {
  377. s.deletedIDs = append(s.deletedIDs, id)
  378. if s.deleteErrByID != nil {
  379. if err, ok := s.deleteErrByID[id]; ok {
  380. return err
  381. }
  382. }
  383. return nil
  384. }
  385. func (s *redeemRepoStub) Use(ctx context.Context, id, userID int64) error {
  386. panic("unexpected Use call")
  387. }
  388. func (s *redeemRepoStub) List(ctx context.Context, params pagination.PaginationParams) ([]RedeemCode, *pagination.PaginationResult, error) {
  389. panic("unexpected List call")
  390. }
  391. func (s *redeemRepoStub) ListWithFilters(ctx context.Context, params pagination.PaginationParams, codeType, status, search string) ([]RedeemCode, *pagination.PaginationResult, error) {
  392. panic("unexpected ListWithFilters call")
  393. }
  394. func (s *redeemRepoStub) ListByUser(ctx context.Context, userID int64, limit int) ([]RedeemCode, error) {
  395. panic("unexpected ListByUser call")
  396. }
  397. func (s *redeemRepoStub) ListByUserPaginated(ctx context.Context, userID int64, params pagination.PaginationParams, codeType string) ([]RedeemCode, *pagination.PaginationResult, error) {
  398. panic("unexpected ListByUserPaginated call")
  399. }
  400. func (s *redeemRepoStub) SumPositiveBalanceByUser(ctx context.Context, userID int64) (float64, error) {
  401. panic("unexpected SumPositiveBalanceByUser call")
  402. }
  403. type subscriptionInvalidateCall struct {
  404. userID int64
  405. groupID int64
  406. }
  407. type billingCacheStub struct {
  408. invalidations chan subscriptionInvalidateCall
  409. }
  410. func newBillingCacheStub(buffer int) *billingCacheStub {
  411. return &billingCacheStub{invalidations: make(chan subscriptionInvalidateCall, buffer)}
  412. }
  413. func (s *billingCacheStub) GetUserBalance(ctx context.Context, userID int64) (float64, error) {
  414. panic("unexpected GetUserBalance call")
  415. }
  416. func (s *billingCacheStub) SetUserBalance(ctx context.Context, userID int64, balance float64) error {
  417. panic("unexpected SetUserBalance call")
  418. }
  419. func (s *billingCacheStub) DeductUserBalance(ctx context.Context, userID int64, amount float64) error {
  420. panic("unexpected DeductUserBalance call")
  421. }
  422. func (s *billingCacheStub) InvalidateUserBalance(ctx context.Context, userID int64) error {
  423. panic("unexpected InvalidateUserBalance call")
  424. }
  425. func (s *billingCacheStub) GetSubscriptionCache(ctx context.Context, userID, groupID int64) (*SubscriptionCacheData, error) {
  426. panic("unexpected GetSubscriptionCache call")
  427. }
  428. func (s *billingCacheStub) SetSubscriptionCache(ctx context.Context, userID, groupID int64, data *SubscriptionCacheData) error {
  429. panic("unexpected SetSubscriptionCache call")
  430. }
  431. func (s *billingCacheStub) UpdateSubscriptionUsage(ctx context.Context, userID, groupID int64, cost float64) error {
  432. panic("unexpected UpdateSubscriptionUsage call")
  433. }
  434. func (s *billingCacheStub) InvalidateSubscriptionCache(ctx context.Context, userID, groupID int64) error {
  435. s.invalidations <- subscriptionInvalidateCall{userID: userID, groupID: groupID}
  436. return nil
  437. }
  438. func (s *billingCacheStub) GetAPIKeyRateLimit(ctx context.Context, keyID int64) (*APIKeyRateLimitCacheData, error) {
  439. panic("unexpected GetAPIKeyRateLimit call")
  440. }
  441. func (s *billingCacheStub) SetAPIKeyRateLimit(ctx context.Context, keyID int64, data *APIKeyRateLimitCacheData) error {
  442. panic("unexpected SetAPIKeyRateLimit call")
  443. }
  444. func (s *billingCacheStub) UpdateAPIKeyRateLimitUsage(ctx context.Context, keyID int64, cost float64) error {
  445. panic("unexpected UpdateAPIKeyRateLimitUsage call")
  446. }
  447. func (s *billingCacheStub) InvalidateAPIKeyRateLimit(ctx context.Context, keyID int64) error {
  448. panic("unexpected InvalidateAPIKeyRateLimit call")
  449. }
  450. func (s *billingCacheStub) GetUserPlatformQuotaCache(ctx context.Context, userID int64, platform string) (*UserPlatformQuotaCacheEntry, bool, error) {
  451. panic("unexpected GetUserPlatformQuotaCache call")
  452. }
  453. func (s *billingCacheStub) SetUserPlatformQuotaCache(ctx context.Context, userID int64, platform string, entry *UserPlatformQuotaCacheEntry, ttl time.Duration) error {
  454. panic("unexpected SetUserPlatformQuotaCache call")
  455. }
  456. func (s *billingCacheStub) DeleteUserPlatformQuotaCache(ctx context.Context, userID int64, platform string) error {
  457. panic("unexpected DeleteUserPlatformQuotaCache call")
  458. }
  459. func (s *billingCacheStub) IncrUserPlatformQuotaUsageCache(ctx context.Context, userID int64, platform string, cost float64, ttl time.Duration, markDirty bool) error {
  460. panic("unexpected IncrUserPlatformQuotaUsageCache call")
  461. }
  462. func (s *billingCacheStub) PopDirtyUserPlatformQuotaKeys(ctx context.Context, n int) ([]UserPlatformQuotaKey, error) {
  463. panic("unexpected PopDirtyUserPlatformQuotaKeys call")
  464. }
  465. func (s *billingCacheStub) ReaddDirtyUserPlatformQuotaKeys(ctx context.Context, keys []UserPlatformQuotaKey) error {
  466. panic("unexpected ReaddDirtyUserPlatformQuotaKeys call")
  467. }
  468. func (s *billingCacheStub) BatchGetUserPlatformQuotaCache(ctx context.Context, keys []UserPlatformQuotaKey) ([]*UserPlatformQuotaCacheEntry, error) {
  469. panic("unexpected BatchGetUserPlatformQuotaCache call")
  470. }
  471. func waitForInvalidations(t *testing.T, ch <-chan subscriptionInvalidateCall, expected int) []subscriptionInvalidateCall {
  472. t.Helper()
  473. calls := make([]subscriptionInvalidateCall, 0, expected)
  474. timeout := time.After(2 * time.Second)
  475. for len(calls) < expected {
  476. select {
  477. case call := <-ch:
  478. calls = append(calls, call)
  479. case <-timeout:
  480. t.Fatalf("timeout waiting for %d invalidations, got %d", expected, len(calls))
  481. }
  482. }
  483. return calls
  484. }
  485. func TestAdminService_DeleteUser_Success(t *testing.T) {
  486. repo := &userRepoStub{user: &User{ID: 7, Role: RoleUser}}
  487. svc := &adminServiceImpl{userRepo: repo}
  488. err := svc.DeleteUser(context.Background(), 7)
  489. require.NoError(t, err)
  490. require.Equal(t, []int64{7}, repo.deletedIDs)
  491. }
  492. func TestAdminService_DeleteUser_DeletesOwnedAPIKeys(t *testing.T) {
  493. repo := &userRepoStub{user: &User{ID: 7, Role: RoleUser}}
  494. apiKeyRepo := &apiKeyRepoStub{
  495. allowListByUserID: true,
  496. listByUserIDKeys: []APIKey{
  497. {ID: 11, UserID: 7, Key: "sk-user-1"},
  498. {ID: 12, UserID: 7, Key: "sk-user-2"},
  499. },
  500. }
  501. invalidator := &authCacheInvalidatorStub{}
  502. svc := &adminServiceImpl{
  503. userRepo: repo,
  504. apiKeyRepo: apiKeyRepo,
  505. authCacheInvalidator: invalidator,
  506. }
  507. err := svc.DeleteUser(context.Background(), 7)
  508. require.NoError(t, err)
  509. require.Equal(t, []int64{7}, repo.deletedIDs)
  510. require.Equal(t, []int64{7}, apiKeyRepo.listByUserIDCalls)
  511. require.Equal(t, []int64{11, 12}, apiKeyRepo.deletedIDs)
  512. require.ElementsMatch(t, []string{"sk-user-1", "sk-user-2"}, invalidator.keys)
  513. require.Equal(t, []int64{7}, invalidator.userIDs)
  514. }
  515. func TestAdminService_DeleteUser_NotFound(t *testing.T) {
  516. repo := &userRepoStub{getErr: ErrUserNotFound}
  517. svc := &adminServiceImpl{userRepo: repo}
  518. err := svc.DeleteUser(context.Background(), 404)
  519. require.ErrorIs(t, err, ErrUserNotFound)
  520. require.Empty(t, repo.deletedIDs)
  521. }
  522. func TestAdminService_DeleteUser_AdminGuard(t *testing.T) {
  523. repo := &userRepoStub{user: &User{ID: 1, Role: RoleAdmin}}
  524. svc := &adminServiceImpl{userRepo: repo}
  525. err := svc.DeleteUser(context.Background(), 1)
  526. require.Error(t, err)
  527. require.ErrorContains(t, err, "cannot delete admin user")
  528. require.Empty(t, repo.deletedIDs)
  529. }
  530. func TestAdminService_DeleteUser_DeleteError(t *testing.T) {
  531. deleteErr := errors.New("delete failed")
  532. repo := &userRepoStub{
  533. user: &User{ID: 9, Role: RoleUser},
  534. deleteErr: deleteErr,
  535. }
  536. svc := &adminServiceImpl{userRepo: repo}
  537. err := svc.DeleteUser(context.Background(), 9)
  538. require.ErrorIs(t, err, deleteErr)
  539. require.Equal(t, []int64{9}, repo.deletedIDs)
  540. }
  541. func TestAdminService_DeleteGroup_Success_WithCacheInvalidation(t *testing.T) {
  542. cache := newBillingCacheStub(2)
  543. repo := &groupRepoStub{affectedUserIDs: []int64{11, 12}}
  544. svc := &adminServiceImpl{
  545. groupRepo: repo,
  546. billingCacheService: &BillingCacheService{cache: cache},
  547. }
  548. err := svc.DeleteGroup(context.Background(), 5)
  549. require.NoError(t, err)
  550. require.Equal(t, []int64{5}, repo.deleteCalls)
  551. calls := waitForInvalidations(t, cache.invalidations, 2)
  552. require.ElementsMatch(t, []subscriptionInvalidateCall{
  553. {userID: 11, groupID: 5},
  554. {userID: 12, groupID: 5},
  555. }, calls)
  556. }
  557. func TestAdminService_DeleteGroup_InvalidatesAuthCacheForBoundKeys(t *testing.T) {
  558. repo := &groupRepoStub{}
  559. apiKeyRepo := &deleteGroupAPIKeyRepoStub{keys: []string{"k1", "k2"}}
  560. invalidator := &authCacheInvalidatorStub{}
  561. svc := &adminServiceImpl{
  562. groupRepo: repo,
  563. apiKeyRepo: apiKeyRepo,
  564. authCacheInvalidator: invalidator,
  565. }
  566. err := svc.DeleteGroup(context.Background(), 5)
  567. require.NoError(t, err)
  568. require.Equal(t, []int64{5}, repo.deleteCalls)
  569. require.Equal(t, []int64{5}, apiKeyRepo.listGroupIDs)
  570. require.Equal(t, []string{"k1", "k2"}, invalidator.keys)
  571. }
  572. func TestAdminService_DeleteGroup_NotFound(t *testing.T) {
  573. repo := &groupRepoStub{deleteErr: ErrGroupNotFound}
  574. svc := &adminServiceImpl{groupRepo: repo}
  575. err := svc.DeleteGroup(context.Background(), 99)
  576. require.ErrorIs(t, err, ErrGroupNotFound)
  577. }
  578. func TestAdminService_DeleteGroup_Error(t *testing.T) {
  579. deleteErr := errors.New("delete failed")
  580. repo := &groupRepoStub{deleteErr: deleteErr}
  581. svc := &adminServiceImpl{groupRepo: repo}
  582. err := svc.DeleteGroup(context.Background(), 42)
  583. require.ErrorIs(t, err, deleteErr)
  584. }
  585. func TestAdminService_DeleteGroupIfEmpty_UsesGuardedCascade(t *testing.T) {
  586. repo := &groupRepoStub{deleteErr: ErrGroupNotEmpty}
  587. svc := &adminServiceImpl{groupRepo: repo, emptyGroupDeleteRepo: repo}
  588. err := svc.DeleteGroupIfEmpty(context.Background(), 42)
  589. require.ErrorIs(t, err, ErrGroupNotEmpty)
  590. require.Equal(t, []int64{42}, repo.guardedCalls)
  591. require.Empty(t, repo.deleteCalls)
  592. }
  593. func TestAdminService_DeleteGroupIfEmpty_MissingCapabilityReturnsError(t *testing.T) {
  594. svc := &adminServiceImpl{groupRepo: &groupRepoStub{}}
  595. require.ErrorContains(t, svc.DeleteGroupIfEmpty(context.Background(), 42), "guarded group deletion is unavailable")
  596. }
  597. func TestAdminService_DeleteProxy_Success(t *testing.T) {
  598. repo := &proxyRepoStub{}
  599. svc := &adminServiceImpl{proxyRepo: repo}
  600. err := svc.DeleteProxy(context.Background(), 7)
  601. require.NoError(t, err)
  602. require.Equal(t, []int64{7}, repo.deletedIDs)
  603. }
  604. func TestAdminService_DeleteProxy_Idempotent(t *testing.T) {
  605. repo := &proxyRepoStub{}
  606. svc := &adminServiceImpl{proxyRepo: repo}
  607. err := svc.DeleteProxy(context.Background(), 404)
  608. require.NoError(t, err)
  609. require.Equal(t, []int64{404}, repo.deletedIDs)
  610. }
  611. func TestAdminService_DeleteProxy_InUse(t *testing.T) {
  612. repo := &proxyRepoStub{accountCount: 2}
  613. svc := &adminServiceImpl{proxyRepo: repo}
  614. err := svc.DeleteProxy(context.Background(), 77)
  615. require.ErrorIs(t, err, ErrProxyInUse)
  616. require.Empty(t, repo.deletedIDs)
  617. }
  618. func TestAdminService_DeleteProxy_Error(t *testing.T) {
  619. deleteErr := errors.New("delete failed")
  620. repo := &proxyRepoStub{deleteErr: deleteErr}
  621. svc := &adminServiceImpl{proxyRepo: repo}
  622. err := svc.DeleteProxy(context.Background(), 33)
  623. require.ErrorIs(t, err, deleteErr)
  624. }
  625. func TestAdminService_DeleteRedeemCode_Success(t *testing.T) {
  626. repo := &redeemRepoStub{}
  627. svc := &adminServiceImpl{redeemCodeRepo: repo}
  628. err := svc.DeleteRedeemCode(context.Background(), 10)
  629. require.NoError(t, err)
  630. require.Equal(t, []int64{10}, repo.deletedIDs)
  631. }
  632. func TestAdminService_DeleteRedeemCode_Idempotent(t *testing.T) {
  633. repo := &redeemRepoStub{}
  634. svc := &adminServiceImpl{redeemCodeRepo: repo}
  635. err := svc.DeleteRedeemCode(context.Background(), 999)
  636. require.NoError(t, err)
  637. require.Equal(t, []int64{999}, repo.deletedIDs)
  638. }
  639. func TestAdminService_DeleteRedeemCode_Error(t *testing.T) {
  640. deleteErr := errors.New("delete failed")
  641. repo := &redeemRepoStub{deleteErrByID: map[int64]error{1: deleteErr}}
  642. svc := &adminServiceImpl{redeemCodeRepo: repo}
  643. err := svc.DeleteRedeemCode(context.Background(), 1)
  644. require.ErrorIs(t, err, deleteErr)
  645. require.Equal(t, []int64{1}, repo.deletedIDs)
  646. }
  647. func TestAdminService_BatchDeleteRedeemCodes_Success(t *testing.T) {
  648. repo := &redeemRepoStub{}
  649. svc := &adminServiceImpl{redeemCodeRepo: repo}
  650. deleted, err := svc.BatchDeleteRedeemCodes(context.Background(), []int64{1, 2, 3})
  651. require.NoError(t, err)
  652. require.Equal(t, int64(3), deleted)
  653. require.Equal(t, []int64{1, 2, 3}, repo.deletedIDs)
  654. }
  655. func TestAdminService_BatchDeleteRedeemCodes_PartialFailures(t *testing.T) {
  656. repo := &redeemRepoStub{
  657. deleteErrByID: map[int64]error{
  658. 2: errors.New("db error"),
  659. },
  660. }
  661. svc := &adminServiceImpl{redeemCodeRepo: repo}
  662. deleted, err := svc.BatchDeleteRedeemCodes(context.Background(), []int64{1, 2, 3})
  663. require.NoError(t, err)
  664. require.Equal(t, int64(2), deleted)
  665. require.Equal(t, []int64{1, 2, 3}, repo.deletedIDs)
  666. }