@anthonyalayo/drizzle-kit 1.0.2 → 1.0.4

package/tests/introspect/singlestore.test.ts

@@ -1,275 +0,0 @@
-import Docker from 'dockerode';
-import 'dotenv/config';
-import { SQL, sql } from 'drizzle-orm';
-import {
-	bigint,
-	char,
-	decimal,
-	double,
-	float,
-	int,
-	mediumint,
-	singlestoreTable,
-	smallint,
-	text,
-	tinyint,
-	varchar,
-} from 'drizzle-orm/singlestore-core';
-import * as fs from 'fs';
-import getPort from 'get-port';
-import { Connection, createConnection } from 'mysql2/promise';
-import { introspectSingleStoreToFile } from 'tests/schemaDiffer';
-import { v4 as uuid } from 'uuid';
-import { afterAll, beforeAll, beforeEach, expect, test } from 'vitest';
-
-let client: Connection;
-let singlestoreContainer: Docker.Container;
-
-async function createDockerDB(): Promise<string> {
-	const docker = new Docker();
-	const port = await getPort({ port: 3306 });
-	const image = 'ghcr.io/singlestore-labs/singlestoredb-dev:latest';
-
-	const pullStream = await docker.pull(image);
-	await new Promise((resolve, reject) =>
-		docker.modem.followProgress(pullStream, (err) => err ? reject(err) : resolve(err))
-	);
-
-	singlestoreContainer = await docker.createContainer({
-		Image: image,
-		Env: ['ROOT_PASSWORD=singlestore'],
-		name: `drizzle-integration-tests-${uuid()}`,
-		HostConfig: {
-			AutoRemove: true,
-			PortBindings: {
-				'3306/tcp': [{ HostPort: `${port}` }],
-			},
-		},
-	});
-
-	await singlestoreContainer.start();
-	await new Promise((resolve) => setTimeout(resolve, 4000));
-
-	return `singlestore://root:singlestore@localhost:${port}/`;
-}
-
-beforeAll(async () => {
-	const connectionString = process.env.SINGLESTORE_CONNECTION_STRING ?? await createDockerDB();
-
-	const sleep = 1000;
-	let timeLeft = 20000;
-	let connected = false;
-	let lastError: unknown | undefined;
-	do {
-		try {
-			client = await createConnection(connectionString);
-			await client.connect();
-			connected = true;
-			break;
-		} catch (e) {
-			lastError = e;
-			await new Promise((resolve) => setTimeout(resolve, sleep));
-			timeLeft -= sleep;
-		}
-	} while (timeLeft > 0);
-	if (!connected) {
-		console.error('Cannot connect to SingleStore');
-		await client?.end().catch(console.error);
-		await singlestoreContainer?.stop().catch(console.error);
-		throw lastError;
-	}
-});
-
-afterAll(async () => {
-	await client?.end().catch(console.error);
-	await singlestoreContainer?.stop().catch(console.error);
-});
-
-beforeEach(async () => {
-	await client.query(`drop database if exists \`drizzle\`;`);
-	await client.query(`create database \`drizzle\`;`);
-	await client.query(`use \`drizzle\`;`);
-});
-
-if (!fs.existsSync('tests/introspect/singlestore')) {
-	fs.mkdirSync('tests/introspect/singlestore');
-}
-
-// TODO: Unskip this test when generated column is implemented
-/* test.skip('generated always column: link to another column', async () => {
-	const schema = {
-		users: singlestoreTable('users', {
-			id: int('id'),
-			email: text('email'),
-			generatedEmail: text('generatedEmail').generatedAlwaysAs(
-				(): SQL => sql`\`email\``,
-			),
-		}),
-	};
-
-	const { statements, sqlStatements } = await introspectSingleStoreToFile(
-		client,
-		schema,
-		'generated-link-column',
-		'drizzle',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-}); */
-
-// TODO: Unskip this test when generated column is implemented
-/* test.skip('generated always column virtual: link to another column', async () => {
-	const schema = {
-		users: singlestoreTable('users', {
-			id: int('id'),
-			email: text('email'),
-			generatedEmail: text('generatedEmail').generatedAlwaysAs(
-				(): SQL => sql`\`email\``,
-				{ mode: 'virtual' },
-			),
-		}),
-	};
-
-	const { statements, sqlStatements } = await introspectSingleStoreToFile(
-		client,
-		schema,
-		'generated-link-column-virtual',
-		'drizzle',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-}); */
-
-test('Default value of character type column: char', async () => {
-	const schema = {
-		users: singlestoreTable('users', {
-			id: int('id'),
-			sortKey: char('sortKey', { length: 255 }).default('0'),
-		}),
-	};
-
-	const { statements, sqlStatements } = await introspectSingleStoreToFile(
-		client,
-		schema,
-		'default-value-char-column',
-		'drizzle',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-});
-
-test('Default value of character type column: varchar', async () => {
-	const schema = {
-		users: singlestoreTable('users', {
-			id: int('id'),
-			sortKey: varchar('sortKey', { length: 255 }).default('0'),
-		}),
-	};
-
-	const { statements, sqlStatements } = await introspectSingleStoreToFile(
-		client,
-		schema,
-		'default-value-varchar-column',
-		'drizzle',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-});
-
-// TODO: Unskip this test when views are implemented
-/* test('view #1', async () => {
-	const users = singlestoreTable('users', { id: int('id') });
-	const testView = singlestoreView('some_view', { id: int('id') }).as(
-		sql`select \`drizzle\`.\`users\`.\`id\` AS \`id\` from \`drizzle\`.\`users\``,
-	);
-
-	const schema = {
-		users: users,
-		testView,
-	};
-
-	const { statements, sqlStatements } = await introspectSingleStoreToFile(
-		client,
-		schema,
-		'view-1',
-		'drizzle',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-}); */
-
-// TODO: Unskip this test when views are implemented
-/* test('view #2', async () => {
-	const users = singlestoreTable('some_users', { id: int('id') });
-	const testView = singlestoreView('some_view', { id: int('id') }).algorithm('temptable').sqlSecurity('definer').as(
-		sql`SELECT * FROM ${users}`,
-	);
-
-	const schema = {
-		users: users,
-		testView,
-	};
-
-	const { statements, sqlStatements } = await introspectSingleStoreToFile(
-		client,
-		schema,
-		'view-2',
-		'drizzle',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-}); */
-
-test('handle float type', async () => {
-	const schema = {
-		table: singlestoreTable('table', {
-			col1: float(),
-			col2: float({ precision: 2 }),
-			col3: float({ precision: 2, scale: 1 }),
-		}),
-	};
-
-	const { statements, sqlStatements } = await introspectSingleStoreToFile(
-		client,
-		schema,
-		'handle-float-type',
-		'drizzle',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-});
-
-test('handle unsigned numerical types', async () => {
-	const schema = {
-		table: singlestoreTable('table', {
-			col1: int({ unsigned: true }),
-			col2: tinyint({ unsigned: true }),
-			col3: smallint({ unsigned: true }),
-			col4: mediumint({ unsigned: true }),
-			col5: bigint({ mode: 'number', unsigned: true }),
-			col6: float({ unsigned: true }),
-			col7: float({ precision: 2, scale: 1, unsigned: true }),
-			col8: double({ unsigned: true }),
-			col9: double({ precision: 2, scale: 1, unsigned: true }),
-			col10: decimal({ unsigned: true }),
-			col11: decimal({ precision: 2, scale: 1, unsigned: true }),
-		}),
-	};
-
-	const { statements, sqlStatements } = await introspectSingleStoreToFile(
-		client,
-		schema,
-		'handle-unsigned-numerical-types',
-		'drizzle',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-});

package/tests/introspect/sqlite.test.ts

@@ -1,122 +0,0 @@
-import Database from 'better-sqlite3';
-import { SQL, sql } from 'drizzle-orm';
-import { check, int, sqliteTable, sqliteView, text } from 'drizzle-orm/sqlite-core';
-import * as fs from 'fs';
-import { introspectSQLiteToFile } from 'tests/schemaDiffer';
-import { expect, test } from 'vitest';
-
-if (!fs.existsSync('tests/introspect/sqlite')) {
-	fs.mkdirSync('tests/introspect/sqlite');
-}
-
-test('generated always column: link to another column', async () => {
-	const sqlite = new Database(':memory:');
-
-	const schema = {
-		users: sqliteTable('users', {
-			id: int('id'),
-			email: text('email'),
-			generatedEmail: text('generatedEmail').generatedAlwaysAs(
-				(): SQL => sql`\`email\``,
-			),
-		}),
-	};
-
-	const { statements, sqlStatements } = await introspectSQLiteToFile(
-		sqlite,
-		schema,
-		'generated-link-column',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-});
-
-test('generated always column virtual: link to another column', async () => {
-	const sqlite = new Database(':memory:');
-
-	const schema = {
-		users: sqliteTable('users', {
-			id: int('id'),
-			email: text('email'),
-			generatedEmail: text('generatedEmail').generatedAlwaysAs(
-				(): SQL => sql`\`email\``,
-				{ mode: 'virtual' },
-			),
-		}),
-	};
-
-	const { statements, sqlStatements } = await introspectSQLiteToFile(
-		sqlite,
-		schema,
-		'generated-link-column-virtual',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-});
-
-test('instrospect strings with single quotes', async () => {
-	const sqlite = new Database(':memory:');
-
-	const schema = {
-		columns: sqliteTable('columns', {
-			text: text('text').default('escape\'s quotes " '),
-		}),
-	};
-
-	const { statements, sqlStatements } = await introspectSQLiteToFile(
-		sqlite,
-		schema,
-		'introspect-strings-with-single-quotes',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-});
-
-test('introspect checks', async () => {
-	const sqlite = new Database(':memory:');
-
-	const schema = {
-		users: sqliteTable('users', {
-			id: int('id'),
-			name: text('name'),
-			age: int('age'),
-		}, (table) => ({
-			someCheck: check('some_check', sql`${table.age} > 21`),
-		})),
-	};
-
-	const { statements, sqlStatements } = await introspectSQLiteToFile(
-		sqlite,
-		schema,
-		'introspect-checks',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-});
-
-test('view #1', async () => {
-	const sqlite = new Database(':memory:');
-
-	const users = sqliteTable('users', { id: int('id') });
-	const testView = sqliteView('some_view', { id: int('id') }).as(
-		sql`SELECT * FROM ${users}`,
-	);
-
-	const schema = {
-		users: users,
-		testView,
-	};
-
-	const { statements, sqlStatements } = await introspectSQLiteToFile(
-		sqlite,
-		schema,
-		'view-1',
-	);
-
-	expect(statements.length).toBe(0);
-	expect(sqlStatements.length).toBe(0);
-});

package/tests/libsql-checks.test.ts

@@ -1,308 +0,0 @@
-import { sql } from 'drizzle-orm';
-import { check, int, sqliteTable, text } from 'drizzle-orm/sqlite-core';
-import { expect, test } from 'vitest';
-import { diffTestSchemasLibSQL } from './schemaDiffer';
-
-test('create table with check', async (t) => {
-	const to = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-		}, (table) => ({
-			checkConstraint: check('some_check_name', sql`${table.age} > 21`),
-		})),
-	};
-
-	const { sqlStatements, statements } = await diffTestSchemasLibSQL({}, to, []);
-
-	expect(statements.length).toBe(1);
-	expect(statements[0]).toStrictEqual({
-		type: 'sqlite_create_table',
-		tableName: 'users',
-		columns: [
-			{
-				name: 'id',
-				type: 'integer',
-				notNull: true,
-				primaryKey: true,
-				autoincrement: false,
-			},
-			{
-				name: 'age',
-				type: 'integer',
-				notNull: false,
-				primaryKey: false,
-				autoincrement: false,
-			},
-		],
-		compositePKs: [],
-		checkConstraints: ['some_check_name;"users"."age" > 21'],
-		referenceData: [],
-		uniqueConstraints: [],
-	});
-
-	expect(sqlStatements.length).toBe(1);
-	expect(sqlStatements[0]).toBe(`CREATE TABLE \`users\` (
-\t\`id\` integer PRIMARY KEY NOT NULL,
-\t\`age\` integer,
-\tCONSTRAINT "some_check_name" CHECK("users"."age" > 21)
-);\n`);
-});
-
-test('add check contraint to existing table', async (t) => {
-	const to = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-		}, (table) => ({
-			checkConstraint: check('some_check_name', sql`${table.age} > 21`),
-		})),
-	};
-
-	const from = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-		}),
-	};
-
-	const { sqlStatements, statements } = await diffTestSchemasLibSQL(from, to, []);
-
-	expect(statements.length).toBe(1);
-	expect(statements[0]).toStrictEqual({
-		columns: [
-			{
-				autoincrement: false,
-				generated: undefined,
-				name: 'id',
-				notNull: true,
-				primaryKey: true,
-				type: 'integer',
-			},
-			{
-				autoincrement: false,
-				generated: undefined,
-				name: 'age',
-				notNull: false,
-				primaryKey: false,
-				type: 'integer',
-			},
-		],
-		compositePKs: [],
-		referenceData: [],
-		tableName: 'users',
-		type: 'recreate_table',
-		uniqueConstraints: [],
-		checkConstraints: ['some_check_name;"users"."age" > 21'],
-	});
-
-	expect(sqlStatements.length).toBe(6);
-	expect(sqlStatements[0]).toBe('PRAGMA foreign_keys=OFF;');
-	expect(sqlStatements[1]).toBe(`CREATE TABLE \`__new_users\` (
-\t\`id\` integer PRIMARY KEY NOT NULL,
-\t\`age\` integer,
-\tCONSTRAINT "some_check_name" CHECK("__new_users"."age" > 21)
-);\n`);
-	expect(sqlStatements[2]).toBe(`INSERT INTO \`__new_users\`("id", "age") SELECT "id", "age" FROM \`users\`;`);
-	expect(sqlStatements[3]).toBe(`DROP TABLE \`users\`;`);
-	expect(sqlStatements[4]).toBe(`ALTER TABLE \`__new_users\` RENAME TO \`users\`;`);
-	expect(sqlStatements[5]).toBe(`PRAGMA foreign_keys=ON;`);
-});
-
-test('drop check contraint to existing table', async (t) => {
-	const from = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-		}, (table) => ({
-			checkConstraint: check('some_check_name', sql`${table.age} > 21`),
-		})),
-	};
-
-	const to = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-		}),
-	};
-
-	const { sqlStatements, statements } = await diffTestSchemasLibSQL(from, to, []);
-
-	expect(statements.length).toBe(1);
-	expect(statements[0]).toStrictEqual({
-		columns: [
-			{
-				autoincrement: false,
-				generated: undefined,
-				name: 'id',
-				notNull: true,
-				primaryKey: true,
-				type: 'integer',
-			},
-			{
-				autoincrement: false,
-				generated: undefined,
-				name: 'age',
-				notNull: false,
-				primaryKey: false,
-				type: 'integer',
-			},
-		],
-		compositePKs: [],
-		referenceData: [],
-		tableName: 'users',
-		type: 'recreate_table',
-		uniqueConstraints: [],
-		checkConstraints: [],
-	});
-
-	expect(sqlStatements.length).toBe(6);
-	expect(sqlStatements[0]).toBe('PRAGMA foreign_keys=OFF;');
-	expect(sqlStatements[1]).toBe(`CREATE TABLE \`__new_users\` (
-\t\`id\` integer PRIMARY KEY NOT NULL,
-\t\`age\` integer
-);\n`);
-	expect(sqlStatements[2]).toBe(`INSERT INTO \`__new_users\`("id", "age") SELECT "id", "age" FROM \`users\`;`);
-	expect(sqlStatements[3]).toBe(`DROP TABLE \`users\`;`);
-	expect(sqlStatements[4]).toBe(`ALTER TABLE \`__new_users\` RENAME TO \`users\`;`);
-	expect(sqlStatements[5]).toBe(`PRAGMA foreign_keys=ON;`);
-});
-
-test('rename check constraint', async (t) => {
-	const from = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-		}, (table) => ({
-			checkConstraint: check('some_check_name', sql`${table.age} > 21`),
-		})),
-	};
-
-	const to = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-		}, (table) => ({
-			checkConstraint: check('new_some_check_name', sql`${table.age} > 21`),
-		})),
-	};
-
-	const { sqlStatements, statements } = await diffTestSchemasLibSQL(from, to, []);
-
-	expect(statements.length).toBe(1);
-	expect(statements[0]).toStrictEqual({
-		columns: [
-			{
-				autoincrement: false,
-				generated: undefined,
-				name: 'id',
-				notNull: true,
-				primaryKey: true,
-				type: 'integer',
-			},
-			{
-				autoincrement: false,
-				generated: undefined,
-				name: 'age',
-				notNull: false,
-				primaryKey: false,
-				type: 'integer',
-			},
-		],
-		compositePKs: [],
-		referenceData: [],
-		tableName: 'users',
-		type: 'recreate_table',
-		uniqueConstraints: [],
-		checkConstraints: [`new_some_check_name;"users"."age" > 21`],
-	});
-
-	expect(sqlStatements.length).toBe(6);
-	expect(sqlStatements[0]).toBe('PRAGMA foreign_keys=OFF;');
-	expect(sqlStatements[1]).toBe(`CREATE TABLE \`__new_users\` (
-\t\`id\` integer PRIMARY KEY NOT NULL,
-\t\`age\` integer,
-\tCONSTRAINT "new_some_check_name" CHECK("__new_users"."age" > 21)
-);\n`);
-	expect(sqlStatements[2]).toBe(`INSERT INTO \`__new_users\`("id", "age") SELECT "id", "age" FROM \`users\`;`);
-	expect(sqlStatements[3]).toBe(`DROP TABLE \`users\`;`);
-	expect(sqlStatements[4]).toBe(`ALTER TABLE \`__new_users\` RENAME TO \`users\`;`);
-	expect(sqlStatements[5]).toBe(`PRAGMA foreign_keys=ON;`);
-});
-
-test('rename check constraint', async (t) => {
-	const from = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-		}, (table) => ({
-			checkConstraint: check('some_check_name', sql`${table.age} > 21`),
-		})),
-	};
-
-	const to = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-		}, (table) => ({
-			checkConstraint: check('some_check_name', sql`${table.age} > 10`),
-		})),
-	};
-
-	const { sqlStatements, statements } = await diffTestSchemasLibSQL(from, to, []);
-
-	expect(statements.length).toBe(1);
-	expect(statements[0]).toStrictEqual({
-		columns: [
-			{
-				autoincrement: false,
-				generated: undefined,
-				name: 'id',
-				notNull: true,
-				primaryKey: true,
-				type: 'integer',
-			},
-			{
-				autoincrement: false,
-				generated: undefined,
-				name: 'age',
-				notNull: false,
-				primaryKey: false,
-				type: 'integer',
-			},
-		],
-		compositePKs: [],
-		referenceData: [],
-		tableName: 'users',
-		type: 'recreate_table',
-		uniqueConstraints: [],
-		checkConstraints: [`some_check_name;"users"."age" > 10`],
-	});
-
-	expect(sqlStatements.length).toBe(6);
-	expect(sqlStatements[0]).toBe('PRAGMA foreign_keys=OFF;');
-	expect(sqlStatements[1]).toBe(`CREATE TABLE \`__new_users\` (
-\t\`id\` integer PRIMARY KEY NOT NULL,
-\t\`age\` integer,
-\tCONSTRAINT "some_check_name" CHECK("__new_users"."age" > 10)
-);\n`);
-	expect(sqlStatements[2]).toBe(`INSERT INTO \`__new_users\`("id", "age") SELECT "id", "age" FROM \`users\`;`);
-	expect(sqlStatements[3]).toBe(`DROP TABLE \`users\`;`);
-	expect(sqlStatements[4]).toBe(`ALTER TABLE \`__new_users\` RENAME TO \`users\`;`);
-	expect(sqlStatements[5]).toBe(`PRAGMA foreign_keys=ON;`);
-});
-
-test('create checks with same names', async (t) => {
-	const to = {
-		users: sqliteTable('users', {
-			id: int('id').primaryKey(),
-			age: int('age'),
-			name: text('name'),
-		}, (table) => ({
-			checkConstraint1: check('some_check_name', sql`${table.age} > 21`),
-			checkConstraint2: check('some_check_name', sql`${table.name} != 'Alex'`),
-		})),
-	};
-
-	await expect(diffTestSchemasLibSQL({}, to, [])).rejects.toThrowError();
-});