사소한 리팩토링

test/setup/database.ts

@@ -0,0 +1,53 @@
+import { PostgreSqlContainer, type StartedPostgreSqlContainer } from "@testcontainers/postgresql";
+import { Kysely, PostgresDialect } from "kysely";
+import { Migrator } from "kysely/migration";
+import pg from "pg";
+import type { TestProject } from "vitest/node";
+import migrations from "../../src/lib/server/db/migrations/index.ts";
+
+declare module "vitest" {
+  export interface ProvidedContext {
+    database: {
+      host: string;
+      port: number;
+      user: string;
+      password: string;
+      name: string;
+    };
+  }
+}
+
+let container: StartedPostgreSqlContainer | undefined;
+
+export const setup = async (project: TestProject) => {
+  container = await new PostgreSqlContainer("postgres:17").start();
+
+  const database = {
+    host: container.getHost(),
+    port: container.getPort(),
+    user: container.getUsername(),
+    password: container.getPassword(),
+    name: container.getDatabase(),
+  };
+
+  const db = new Kysely({
+    dialect: new PostgresDialect({
+      pool: new pg.Pool({ ...database, database: database.name }),
+    }),
+  });
+  try {
+    const { error } = await new Migrator({
+      db,
+      provider: { getMigrations: async () => migrations },
+    }).migrateToLatest();
+    if (error) throw error;
+  } finally {
+    await db.destroy();
+  }
+
+  project.provide("database", database);
+};
+
+export const teardown = async () => {
+  await container?.stop();
+};