chore: reset cognito after each test

compose.yml

@@ -1,6 +1,6 @@
 services:
   magnito:
-    image: frourio/magnito:0.19.0
+    image: frourio/magnito:0.20.0
     ports:
       - 5050:5050
       - 5051:5051

server/tests/api/apiClient.ts

@@ -16,7 +16,7 @@ import {
   COGNITO_USER_POOL_ID,
 } from 'service/envValues';
 import { ulid } from 'ulid';
-import { TEST_PORT } from './utils';
+import { TEST_PORT, TEST_USERNAME_PREFIX } from './utils';
 
 const baseURL = `http://127.0.0.1:${TEST_PORT}${API_BASE_PATH}`;
 
@@ -27,7 +27,7 @@ export const noCookieClient = api(
 type Tokens = { idToken: string; accessToken: string };
 
 export const createCognitoUser = async (): Promise<Tokens> => {
-  const userName = `test-${ulid()}`;
+  const userName = `${TEST_USERNAME_PREFIX}-${ulid()}`;
   const password = `Test-user-${ulid()}`;
   const command1 = new AdminCreateUserCommand({
     UserPoolId: COGNITO_USER_POOL_ID,
@@ -56,7 +56,7 @@ export const createCognitoUser = async (): Promise<Tokens> => {
 };
 
 export const createGoogleUser = async (): Promise<Tokens> => {
-  const userName = `test-${ulid()}`;
+  const userName = `${TEST_USERNAME_PREFIX}-${ulid()}`;
   const codeVerifier = ulid();
   const user = await fetch(`${COGNITO_POOL_ENDPOINT}/public/socialUsers`, {
     method: 'POST',

server/tests/api/utils.ts

@@ -4,6 +4,8 @@ import { SERVER_PORT } from 'service/envValues';
 
 export const TEST_PORT = SERVER_PORT - 1;
 
+export const TEST_USERNAME_PREFIX = 'test-username-prefix';
+
 export const GET = (api: { get: unknown; $path: () => string }): string => `GET: ${api.$path()}`;
 
 export const POST = (api: { post: unknown; $path: () => string }): string => `POST: ${api.$path()}`;

server/tests/setup.ts

@@ -1,13 +1,18 @@
+import {
+  AdminDeleteUserCommand,
+  ListUsersCommand,
+} from '@aws-sdk/client-cognito-identity-provider';
 import { DeleteObjectsCommand, ListObjectsV2Command } from '@aws-sdk/client-s3';
 import { exec } from 'child_process';
 import type { FastifyInstance } from 'fastify';
 import { init } from 'service/app';
-import { S3_BUCKET } from 'service/envValues';
+import { cognitoClient } from 'service/cognito';
+import { COGNITO_USER_POOL_ID, S3_BUCKET } from 'service/envValues';
 import { prismaClient } from 'service/prismaClient';
 import { s3Client } from 'service/s3Client';
 import util from 'util';
 import { afterAll, afterEach, beforeAll, beforeEach } from 'vitest';
-import { TEST_PORT } from './api/utils';
+import { TEST_PORT, TEST_USERNAME_PREFIX } from './api/utils';
 
 let server: FastifyInstance;
 
@@ -27,21 +32,49 @@ beforeEach(async (info) => {
   await util.promisify(exec)('npx prisma migrate reset --force');
 });
 
+const resetS3 = async (): Promise<void> => {
+  const objects = await s3Client.send(new ListObjectsV2Command({ Bucket: S3_BUCKET }));
+  const keys = objects.Contents?.map(({ Key }) => ({ Key })) ?? [];
+
+  if (keys.length === 0) return;
+
+  await s3Client.send(new DeleteObjectsCommand({ Bucket: S3_BUCKET, Delete: { Objects: keys } }));
+};
+
+const resetCognito = async (): Promise<void> => {
+  const { Users } = await cognitoClient.send(
+    new ListUsersCommand({ UserPoolId: COGNITO_USER_POOL_ID }),
+  );
+
+  if (!Users) return;
+
+  const testUsers = Users.filter((u) => u.Username?.startsWith(TEST_USERNAME_PREFIX));
+
+  for (let i = 0; ; i += 1) {
+    const targets = testUsers.slice(i * 10, (i + 1) * 10);
+
+    if (targets.length === 0) return;
+
+    await Promise.all(
+      targets.map((u) =>
+        cognitoClient.send(
+          new AdminDeleteUserCommand({ Username: u.Username, UserPoolId: COGNITO_USER_POOL_ID }),
+        ),
+      ),
+    );
+  }
+};
+
 afterEach(async (info) => {
   if (unneededServer(info.task.file)) return;
 
   await prismaClient.$disconnect();
+  await resetS3();
+  await resetCognito();
 });
 
 afterAll(async (info) => {
   if (unneededServer(info.file)) return;
 
   await server.close();
-
-  const objects = await s3Client.send(new ListObjectsV2Command({ Bucket: S3_BUCKET }));
-  const keys = objects.Contents?.map(({ Key }) => ({ Key })) ?? [];
-
-  if (keys.length === 0) return;
-
-  await s3Client.send(new DeleteObjectsCommand({ Bucket: S3_BUCKET, Delete: { Objects: keys } }));
 });