wehub-resource-sync
101 lines
3.4 KiB
TypeScript
101 lines
3.4 KiB
TypeScript
import { D1Database, D1DatabaseAPI } from '@miniflare/d1';
|
|
import { createSQLiteDB } from '@miniflare/shared';
|
|
import { sql } from 'drizzle-orm';
|
|
import type { DrizzleD1Database } from 'drizzle-orm/d1';
|
|
import { drizzle } from 'drizzle-orm/d1';
|
|
import { migrate } from 'drizzle-orm/d1/migrator';
|
|
import { beforeAll, beforeEach, expect, test } from 'vitest';
|
|
import { skipTests } from '~/common';
|
|
import { randomString } from '~/utils';
|
|
import { anotherUsersMigratorTable, tests, usersMigratorTable } from './sqlite-common';
|
|
import { TestCache, TestGlobalCache, tests as cacheTests } from './sqlite-common-cache';
|
|
|
|
const ENABLE_LOGGING = false;
|
|
|
|
let db: DrizzleD1Database;
|
|
let dbGlobalCached: DrizzleD1Database;
|
|
let cachedDb: DrizzleD1Database;
|
|
|
|
beforeAll(async () => {
|
|
const sqliteDb = await createSQLiteDB(':memory:');
|
|
const d1db = new D1Database(new D1DatabaseAPI(sqliteDb));
|
|
db = drizzle(d1db, { logger: ENABLE_LOGGING });
|
|
cachedDb = drizzle(d1db, { logger: ENABLE_LOGGING, cache: new TestCache() });
|
|
dbGlobalCached = drizzle(d1db, { logger: ENABLE_LOGGING, cache: new TestGlobalCache() });
|
|
});
|
|
|
|
beforeEach((ctx) => {
|
|
ctx.sqlite = {
|
|
db,
|
|
};
|
|
ctx.cachedSqlite = {
|
|
db: cachedDb,
|
|
dbGlobalCached,
|
|
};
|
|
});
|
|
|
|
test('migrator', async () => {
|
|
await db.run(sql`drop table if exists another_users`);
|
|
await db.run(sql`drop table if exists users12`);
|
|
await db.run(sql`drop table if exists __drizzle_migrations`);
|
|
|
|
await migrate(db, { migrationsFolder: './drizzle2/sqlite' });
|
|
|
|
await db.insert(usersMigratorTable).values({ name: 'John', email: 'email' }).run();
|
|
const result = await db.select().from(usersMigratorTable).all();
|
|
|
|
await db.insert(anotherUsersMigratorTable).values({ name: 'John', email: 'email' }).run();
|
|
const result2 = await db.select().from(anotherUsersMigratorTable).all();
|
|
|
|
expect(result).toEqual([{ id: 1, name: 'John', email: 'email' }]);
|
|
expect(result2).toEqual([{ id: 1, name: 'John', email: 'email' }]);
|
|
|
|
await db.run(sql`drop table another_users`);
|
|
await db.run(sql`drop table users12`);
|
|
await db.run(sql`drop table __drizzle_migrations`);
|
|
});
|
|
|
|
test('migrator : migrate with custom table', async () => {
|
|
const customTable = randomString();
|
|
await db.run(sql`drop table if exists another_users`);
|
|
await db.run(sql`drop table if exists users12`);
|
|
await db.run(sql`drop table if exists ${sql.identifier(customTable)}`);
|
|
|
|
await migrate(db, { migrationsFolder: './drizzle2/sqlite', migrationsTable: customTable });
|
|
|
|
// test if the custom migrations table was created
|
|
const res = await db.all(sql`select * from ${sql.identifier(customTable)};`);
|
|
expect(res.length > 0).toBeTruthy();
|
|
|
|
// test if the migrated table are working as expected
|
|
await db.insert(usersMigratorTable).values({ name: 'John', email: 'email' });
|
|
const result = await db.select().from(usersMigratorTable);
|
|
expect(result).toEqual([{ id: 1, name: 'John', email: 'email' }]);
|
|
|
|
await db.run(sql`drop table another_users`);
|
|
await db.run(sql`drop table users12`);
|
|
await db.run(sql`drop table ${sql.identifier(customTable)}`);
|
|
});
|
|
|
|
skipTests([
|
|
// Cannot convert 49,50,55 to a BigInt
|
|
'insert bigint values',
|
|
// SyntaxError: Unexpected token , in JSON at position 2
|
|
'json insert',
|
|
'insert many',
|
|
'insert many with returning',
|
|
/**
|
|
* TODO: Fix Bug! The objects should be equal
|
|
*
|
|
* See #528 for more details.
|
|
* Tldr the D1 driver does not execute joins successfully
|
|
*/
|
|
'partial join with alias',
|
|
'full join with alias',
|
|
'select from alias',
|
|
'join view as subquery',
|
|
'cross join',
|
|
]);
|
|
cacheTests();
|
|
tests();
|