wehub-resource-sync
61 lines
1.8 KiB
TypeScript
61 lines
1.8 KiB
TypeScript
import Database from 'better-sqlite3';
|
|
import { sql } from 'drizzle-orm';
|
|
import { type BetterSQLite3Database, drizzle } from 'drizzle-orm/better-sqlite3';
|
|
import { migrate } from 'drizzle-orm/better-sqlite3/migrator';
|
|
import { afterAll, beforeAll, beforeEach, expect, test } from 'vitest';
|
|
import { skipTests } from '~/common';
|
|
import { anotherUsersMigratorTable, tests, usersMigratorTable } from './sqlite-common';
|
|
|
|
const ENABLE_LOGGING = false;
|
|
|
|
let db: BetterSQLite3Database;
|
|
let client: Database.Database;
|
|
|
|
beforeAll(async () => {
|
|
const dbPath = process.env['SQLITE_DB_PATH'] ?? ':memory:';
|
|
client = new Database(dbPath);
|
|
db = drizzle(client, { logger: ENABLE_LOGGING });
|
|
});
|
|
|
|
afterAll(async () => {
|
|
client?.close();
|
|
});
|
|
|
|
beforeEach((ctx) => {
|
|
ctx.sqlite = {
|
|
db,
|
|
};
|
|
});
|
|
|
|
test('migrator', async () => {
|
|
db.run(sql`drop table if exists another_users`);
|
|
db.run(sql`drop table if exists users12`);
|
|
db.run(sql`drop table if exists __drizzle_migrations`);
|
|
|
|
migrate(db, { migrationsFolder: './drizzle2/sqlite' });
|
|
|
|
db.insert(usersMigratorTable).values({ name: 'John', email: 'email' }).run();
|
|
const result = db.select().from(usersMigratorTable).all();
|
|
|
|
db.insert(anotherUsersMigratorTable).values({ name: 'John', email: 'email' }).run();
|
|
const result2 = db.select().from(anotherUsersMigratorTable).all();
|
|
|
|
expect(result).toEqual([{ id: 1, name: 'John', email: 'email' }]);
|
|
expect(result2).toEqual([{ id: 1, name: 'John', email: 'email' }]);
|
|
|
|
db.run(sql`drop table another_users`);
|
|
db.run(sql`drop table users12`);
|
|
db.run(sql`drop table __drizzle_migrations`);
|
|
});
|
|
|
|
skipTests([
|
|
/**
|
|
* doesn't work properly:
|
|
* Expect: should rollback transaction and don't insert/ update data
|
|
* Received: data inserted/ updated
|
|
*/
|
|
'transaction rollback',
|
|
'nested transaction rollback',
|
|
]);
|
|
tests();
|