mirror/ditto
mirror/ditto
1
0
Fork 0
mirror of https://gitlab.com/soapbox-pub/ditto.git synced 2025-12-06 11:29:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: drop postgres tables dynamically

Browse source
This commit is contained in:
Alex Gleason 2024-09-25 13:13:58 -05:00
parent f44b9267ad
commit 606aeb3236
No known key found for this signature in database
GPG key ID: 7211D1F99744FBB7
1 changed files with 9 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
src/test.ts
View file

@ -5,6 +5,7 @@ import { Conf } from '@/config.ts';
import { DittoDB } from '@/db/DittoDB.ts';
import { EventsDB } from '@/storages/EventsDB.ts';
import { purifyEvent } from '@/utils/purify.ts';
import { sql } from 'kysely';
/** Import an event fixture by name in tests. */
export async function eventFixture(name: string): Promise<NostrEvent> {
@ -34,7 +35,6 @@ export function genEvent(t: Partial<NostrEvent> = {}, sk: Uint8Array = generateS
/** Create a database for testing. It uses `TEST_DATABASE_URL`, or creates an in-memory database by default. */
export async function createTestDB() {
const { testDatabaseUrl } = Conf;
const { protocol } = new URL(testDatabaseUrl);
const { kysely } = DittoDB.create(testDatabaseUrl, { poolSize: 1 });
await DittoDB.migrate(kysely);
@ -49,27 +49,15 @@ export async function createTestDB() {
store,
kysely,
[Symbol.asyncDispose]: async () => {
// If we're testing against real Postgres, we will reuse the database
// between tests, so we should drop the tables to keep each test fresh.
if (['postgres:', 'postgresql:'].includes(protocol)) {
for (
const table of [
'author_stats',
'event_stats',
'event_zaps',
'kysely_migration',
'kysely_migration_lock',
'nip46_tokens',
'pubkey_domains',
'nostr_events',
'event_zaps',
'author_search',
]
) {
await kysely.schema.dropTable(table).ifExists().cascade().execute();
const { rows } = await sql<
{ tablename: string }
>`select tablename from pg_tables where schemaname = current_schema()`.execute(kysely);
for (const { tablename } of rows) {
await kysely.schema.dropTable(tablename).ifExists().cascade().execute();
}
await kysely.destroy();
}
},
};
}