浅念
42 lines
1.6 KiB
TypeScript
42 lines
1.6 KiB
TypeScript
import { MySqlContainer } from '@testcontainers/mysql';
|
|
import type { StartedTestContainer } from 'testcontainers';
|
|
import { execSync } from 'node:child_process';
|
|
import { getPrisma, disconnectPrisma } from '../src/lib/prisma.js';
|
|
import { createApp } from '../src/app.js';
|
|
import { bootstrapAdmin } from '../src/services/auth.service.js';
|
|
|
|
// 注:MySqlContainer 默认 user=root password=test
|
|
let container: StartedTestContainer | null = null;
|
|
|
|
export async function setupApp() {
|
|
if (!container) {
|
|
container = await new MySqlContainer('mysql:8').withDatabase('t').start();
|
|
process.env.DATABASE_URL = `mysql://root:test@${container.getHost()}:${container.getMappedPort(3306)}/t`;
|
|
process.env.INGEST_TOKEN = 'dev-ingest-token-1234';
|
|
process.env.JWT_SECRET = 'dev-jwt-secret-1234';
|
|
process.env.ADMIN_USERNAME = 'admin';
|
|
process.env.ADMIN_PASSWORD = 'admin-password-1234';
|
|
execSync('npx prisma migrate deploy', { stdio: 'inherit', env: process.env });
|
|
}
|
|
const prisma = getPrisma();
|
|
await bootstrapAdmin(); // 确保 admin 存在(auth 测试需要)
|
|
const app = createApp();
|
|
return { app, prisma };
|
|
}
|
|
|
|
export async function resetDb() {
|
|
const prisma = getPrisma();
|
|
await prisma.$executeRawUnsafe('SET FOREIGN_KEY_CHECKS=0');
|
|
for (const t of ['chat_messages', 'inquiries', 'customers', 'users']) {
|
|
await prisma.$executeRawUnsafe(`DELETE FROM ${t}`);
|
|
}
|
|
await prisma.$executeRawUnsafe('SET FOREIGN_KEY_CHECKS=1');
|
|
await bootstrapAdmin(); // resetDb 清了 users,重新 seed admin
|
|
}
|
|
|
|
export async function teardown() {
|
|
await disconnectPrisma();
|
|
await container?.stop();
|
|
container = null;
|
|
}
|