|
|
|
|
@@ -29,22 +29,31 @@ export async function setupTestEnvironment(): Promise<void> {
|
|
|
|
|
* Cleanup test database
|
|
|
|
|
* Disconnects Prisma client and cleans all data
|
|
|
|
|
*/
|
|
|
|
|
async function deleteManyIfTableExists(
|
|
|
|
|
prisma: PrismaClient,
|
|
|
|
|
tableName: string,
|
|
|
|
|
deleteFn: () => Promise<unknown>
|
|
|
|
|
) {
|
|
|
|
|
if (await verifyTableExists(prisma, tableName)) {
|
|
|
|
|
await deleteFn()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
export async function cleanupTestDatabase(prisma: PrismaClient) {
|
|
|
|
|
try {
|
|
|
|
|
// Delete in dependency order
|
|
|
|
|
await prisma.aiFeedback.deleteMany()
|
|
|
|
|
await prisma.memoryEchoInsight.deleteMany()
|
|
|
|
|
await prisma.noteShare.deleteMany()
|
|
|
|
|
await prisma.note.deleteMany()
|
|
|
|
|
await prisma.label.deleteMany()
|
|
|
|
|
await prisma.notebook.deleteMany()
|
|
|
|
|
await prisma.userAISettings.deleteMany()
|
|
|
|
|
await prisma.systemConfig.deleteMany()
|
|
|
|
|
await prisma.session.deleteMany()
|
|
|
|
|
await prisma.account.deleteMany()
|
|
|
|
|
await prisma.verificationToken.deleteMany()
|
|
|
|
|
await prisma.subscription.deleteMany()
|
|
|
|
|
await prisma.user.deleteMany()
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'AiFeedback', () => prisma.aiFeedback.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'MemoryEchoInsight', () => prisma.memoryEchoInsight.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'NoteShare', () => prisma.noteShare.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'Note', () => prisma.note.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'Label', () => prisma.label.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'Notebook', () => prisma.notebook.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'UserAISettings', () => prisma.userAISettings.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'SystemConfig', () => prisma.systemConfig.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'Session', () => prisma.session.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'Account', () => prisma.account.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'VerificationToken', () => prisma.verificationToken.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'Subscription', () => prisma.subscription.deleteMany())
|
|
|
|
|
await deleteManyIfTableExists(prisma, 'User', () => prisma.user.deleteMany())
|
|
|
|
|
await prisma.$disconnect()
|
|
|
|
|
} catch (error) {
|
|
|
|
|
console.error('Error cleaning up test database:', error)
|
|
|
|
|
@@ -211,6 +220,14 @@ export async function verifyColumnExists(prisma: PrismaClient, tableName: string
|
|
|
|
|
/**
|
|
|
|
|
* Get table schema information (PostgreSQL version)
|
|
|
|
|
*/
|
|
|
|
|
/** Approximate DB size in bytes (PostgreSQL). */
|
|
|
|
|
export async function getDatabaseSize(prisma: PrismaClient): Promise<number> {
|
|
|
|
|
const rows = await prisma.$queryRaw<Array<{ size: bigint }>>`
|
|
|
|
|
SELECT pg_database_size(current_database()) AS size
|
|
|
|
|
`
|
|
|
|
|
return Number(rows[0]?.size ?? 0)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
export async function getTableSchema(prisma: PrismaClient, tableName: string) {
|
|
|
|
|
try {
|
|
|
|
|
const result = await prisma.$queryRawUnsafe<Array<{
|
|
|
|
|
|