chore: import upstream snapshot with attribution

2026-07-13 12:10:05 +08:00
commit d37d8d293b
1388 changed files with 484182 additions and 0 deletions
@@ -0,0 +1,100 @@
+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();