chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,60 @@
|
||||
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();
|
||||
Reference in New Issue
Block a user