[drizzle-kit] added tests

Author: OleksiiKH0240

drizzle-kit/tests/postgres/pg-columns.test.ts

@@ -211,7 +211,7 @@ test('alter column type to custom type', async (t) => {
 });
 
 // https://github.com/drizzle-team/drizzle-orm/issues/4245
-test('alter text type to jsonb type', async (t) => {
+test('alter text type to jsonb type', async () => {
 	const schema1 = {
 		table1: pgTable('table1', {
 			column1: text(),
@@ -245,6 +245,61 @@ test('alter text type to jsonb type', async (t) => {
 	expect(res[0]).toBeDefined();
 });
 
+// https://github.com/drizzle-team/drizzle-orm/issues/3589
+test('alter integer type to text type with fk constraints', async () => {
+	const users1 = pgTable('users', {
+		id: serial().primaryKey(),
+	});
+
+	const schema1 = {
+		users1,
+		sessions: pgTable('sessions', {
+			id: text().primaryKey(),
+			userId: integer().notNull().references(() => users1.id),
+		}),
+		content: pgTable('content', {
+			id: text().primaryKey(),
+			userId: integer().notNull().references(() => users1.id),
+		}),
+	};
+
+	const { sqlStatements: st1, next: n1 } = await diff({}, schema1, []);
+	await push({ db, to: schema1 });
+	await db.query('insert into "users" values (1);');
+	await db.query('insert into "sessions" values (1,1);');
+	await db.query('insert into "content" values (1,1);');
+
+	const users2 = pgTable('users', {
+		id: text().primaryKey(),
+	});
+	const schema2 = {
+		users2,
+		sessions: pgTable('sessions', {
+			id: text().primaryKey(),
+			userId: text().notNull().references(() => users2.id),
+		}),
+		content: pgTable('content', {
+			id: text().primaryKey(),
+			userId: text().notNull().references(() => users2.id),
+		}),
+	};
+
+	const { sqlStatements: st2 } = await diff(n1, schema2, []);
+	const { sqlStatements: pst2 } = await push({ db, to: schema2 });
+
+	const expectedSt2 = [
+		'ALTER TABLE "sessions" DROP CONSTRAINT "sessions_userId_users_id_fkey";',
+		'ALTER TABLE "content" DROP CONSTRAINT "content_userId_users_id_fkey";',
+		'ALTER TABLE "users" ALTER COLUMN "id" SET DATA TYPE text;',
+		'ALTER TABLE "sessions" ALTER COLUMN "userId" SET DATA TYPE text;',
+		'ALTER TABLE "content" ALTER COLUMN "userId" SET DATA TYPE text;',
+		'ALTER TABLE "sessions" ADD CONSTRAINT "sessions_userId_users_id_fkey" FOREIGN KEY ("userId") REFERENCES "users"("id");',
+		'ALTER TABLE "content" ADD CONSTRAINT "content_userId_users_id_fkey" FOREIGN KEY ("userId") REFERENCES "users"("id");',
+	];
+	expect(st2).toStrictEqual(expectedSt2);
+	expect(pst2).toStrictEqual(expectedSt2);
+});
+
 test('alter table add composite pk', async (t) => {
 	const schema1 = {
 		table: pgTable('table', {
@@ -277,6 +332,43 @@ test('alter table add composite pk', async (t) => {
 	expect(pst).toStrictEqual(st0);
 });
 
+// https://github.com/drizzle-team/drizzle-orm/issues/3496
+test('remove/add pk', async (t) => {
+	const Step = pgTable('Step', {
+		id: bigint({ mode: 'number' }).primaryKey(),
+	});
+	const schema1 = {
+		Step1: Step,
+		Branch: pgTable('Branch', {
+			id: bigint({ mode: 'number' }).primaryKey(),
+			stepId: bigint({ mode: 'number' }).references(() => Step.id, { onDelete: 'cascade' }),
+		}),
+	};
+
+	const { next: n1 } = await diff({}, schema1, []);
+	await push({ db, to: schema1 });
+
+	const schema2 = {
+		Step,
+		Branch: pgTable('Branch', {
+			stepId: bigint({ mode: 'number' }).primaryKey().references(() => Step.id, { onDelete: 'cascade' }),
+		}),
+	};
+
+	const { sqlStatements: st2 } = await diff(n1, schema2, []);
+	const { sqlStatements: pst2 } = await push({ db, to: schema2 });
+
+	const expectedSt2 = [
+		[
+			'ALTER TABLE "Branch" DROP CONSTRAINT "Branch_pkey";',
+			'ALTER TABLE "Branch" DROP COLUMN "id";',
+			'ALTER TABLE "Branch" ADD PRIMARY KEY ("stepId");',
+		],
+	];
+	expect(st2).toStrictEqual(expectedSt2);
+	expect(pst2).toStrictEqual(expectedSt2);
+});
+
 test('rename table rename column #1', async (t) => {
 	const schema1 = {
 		users: pgTable('users', {

drizzle-kit/tests/postgres/pg-defaults.test.ts

@@ -389,6 +389,7 @@ test('numeric arrays', async () => {
 	expect.soft(res24).toStrictEqual([]);
 });
 
+// https://github.com/drizzle-team/drizzle-orm/issues/3582
 test('real + real arrays', async () => {
 	const res1 = await diffDefault(_, real().default(1000.123), '1000.123');
 

drizzle-kit/tests/postgres/pull.test.ts

@@ -340,13 +340,14 @@ test('generated column: link to another jsonb column', async () => {
 	expect(sqlStatements.length).toBe(0);
 });
 
+// https://github.com/drizzle-team/drizzle-orm/issues/3593
 // https://github.com/drizzle-team/drizzle-orm/issues/4349
 // https://github.com/drizzle-team/drizzle-orm/issues/4632
 // https://github.com/drizzle-team/drizzle-orm/issues/4644
 // https://github.com/drizzle-team/drizzle-orm/issues/4730
 // https://github.com/drizzle-team/drizzle-orm/issues/4760
 // https://github.com/drizzle-team/drizzle-orm/issues/4916
-test('introspect all column types', async () => {
+test.only('introspect all column types', async () => {
 	const myEnum = pgEnum('my_enum', ['a', 'b', 'c']);
 	const schema = {
 		enum_: myEnum,
@@ -366,6 +367,7 @@ test('introspect all column types', async () => {
 			text3: text('text3').default(''),
 			varchar: varchar('varchar', { length: 25 }).default('abc'),
 			varchar1: varchar('varchar1', { length: 25 }).default(''),
+			varchar2: varchar('varchar2').default(sql`md5((random())::text)`),
 			char: char('char', { length: 3 }).default('abc'),
 			char1: char('char1', { length: 3 }).default(''),
 			serial: serial('serial'),