refactor: sql-tools (#19717)

2025-07-03 10:59:17 -04:00
parent 484529e61e
commit 6044663e26
160 changed files with 1120 additions and 1186 deletions
@@ -0,0 +1,96 @@
+import { transformConstraints } from 'src/sql-tools/transformers/constraint.transformer';
+import { ConstraintType } from 'src/sql-tools/types';
+import { describe, expect, it } from 'vitest';
+
+describe(transformConstraints.name, () => {
+  describe('ConstraintAdd', () => {
+    describe('primary keys', () => {
+      it('should work', () => {
+        expect(
+          transformConstraints({
+            type: 'ConstraintAdd',
+            constraint: {
+              type: ConstraintType.PRIMARY_KEY,
+              name: 'PK_test',
+              tableName: 'table1',
+              columnNames: ['id'],
+              synchronize: true,
+            },
+            reason: 'unknown',
+          }),
+        ).toEqual('ALTER TABLE "table1" ADD CONSTRAINT "PK_test" PRIMARY KEY ("id");');
+      });
+    });
+
+    describe('foreign keys', () => {
+      it('should work', () => {
+        expect(
+          transformConstraints({
+            type: 'ConstraintAdd',
+            constraint: {
+              type: ConstraintType.FOREIGN_KEY,
+              name: 'FK_test',
+              tableName: 'table1',
+              columnNames: ['parentId'],
+              referenceColumnNames: ['id'],
+              referenceTableName: 'table2',
+              synchronize: true,
+            },
+            reason: 'unknown',
+          }),
+        ).toEqual(
+          'ALTER TABLE "table1" ADD CONSTRAINT "FK_test" FOREIGN KEY ("parentId") REFERENCES "table2" ("id") ON UPDATE NO ACTION ON DELETE NO ACTION;',
+        );
+      });
+    });
+
+    describe('unique', () => {
+      it('should work', () => {
+        expect(
+          transformConstraints({
+            type: 'ConstraintAdd',
+            constraint: {
+              type: ConstraintType.UNIQUE,
+              name: 'UQ_test',
+              tableName: 'table1',
+              columnNames: ['id'],
+              synchronize: true,
+            },
+            reason: 'unknown',
+          }),
+        ).toEqual('ALTER TABLE "table1" ADD CONSTRAINT "UQ_test" UNIQUE ("id");');
+      });
+    });
+
+    describe('check', () => {
+      it('should work', () => {
+        expect(
+          transformConstraints({
+            type: 'ConstraintAdd',
+            constraint: {
+              type: ConstraintType.CHECK,
+              name: 'CHK_test',
+              tableName: 'table1',
+              expression: '"id" IS NOT NULL',
+              synchronize: true,
+            },
+            reason: 'unknown',
+          }),
+        ).toEqual('ALTER TABLE "table1" ADD CONSTRAINT "CHK_test" CHECK ("id" IS NOT NULL);');
+      });
+    });
+  });
+
+  describe('ConstraintDrop', () => {
+    it('should work', () => {
+      expect(
+        transformConstraints({
+          type: 'ConstraintDrop',
+          tableName: 'table1',
+          constraintName: 'PK_test',
+          reason: 'unknown',
+        }),
+      ).toEqual(`ALTER TABLE "table1" DROP CONSTRAINT "PK_test";`);
+    });
+  });
+});