refactor: sql-tools (#19717)

2025-12-17 17:23:20 +03:00 · 2025-07-03 10:59:17 -04:00
parent 484529e61e
commit 6044663e26
160 changed files with 1120 additions and 1186 deletions
--- a/server/src/sql-tools/transformers/trigger.transformer.spec.ts
+++ b/server/src/sql-tools/transformers/trigger.transformer.spec.ts
@@ -0,0 +1,91 @@
+import { transformTriggers } from 'src/sql-tools/transformers/trigger.transformer';
+import { describe, expect, it } from 'vitest';
+
+describe(transformTriggers.name, () => {
+  describe('TriggerCreate', () => {
+    it('should work', () => {
+      expect(
+        transformTriggers({
+          type: 'TriggerCreate',
+          trigger: {
+            name: 'trigger1',
+            tableName: 'table1',
+            timing: 'before',
+            actions: ['update'],
+            scope: 'row',
+            functionName: 'function1',
+            synchronize: true,
+          },
+          reason: 'unknown',
+        }),
+      ).toEqual(
+        `CREATE OR REPLACE TRIGGER "trigger1"
+  BEFORE UPDATE ON "table1"
+  FOR EACH ROW
+  EXECUTE FUNCTION function1();`,
+      );
+    });
+
+    it('should work with multiple actions', () => {
+      expect(
+        transformTriggers({
+          type: 'TriggerCreate',
+          trigger: {
+            name: 'trigger1',
+            tableName: 'table1',
+            timing: 'before',
+            actions: ['update', 'delete'],
+            scope: 'row',
+            functionName: 'function1',
+            synchronize: true,
+          },
+          reason: 'unknown',
+        }),
+      ).toEqual(
+        `CREATE OR REPLACE TRIGGER "trigger1"
+  BEFORE UPDATE OR DELETE ON "table1"
+  FOR EACH ROW
+  EXECUTE FUNCTION function1();`,
+      );
+    });
+
+    it('should work with old/new reference table aliases', () => {
+      expect(
+        transformTriggers({
+          type: 'TriggerCreate',
+          trigger: {
+            name: 'trigger1',
+            tableName: 'table1',
+            timing: 'before',
+            actions: ['update'],
+            referencingNewTableAs: 'new',
+            referencingOldTableAs: 'old',
+            scope: 'row',
+            functionName: 'function1',
+            synchronize: true,
+          },
+          reason: 'unknown',
+        }),
+      ).toEqual(
+        `CREATE OR REPLACE TRIGGER "trigger1"
+  BEFORE UPDATE ON "table1"
+  REFERENCING OLD TABLE AS "old" NEW TABLE AS "new"
+  FOR EACH ROW
+  EXECUTE FUNCTION function1();`,
+      );
+    });
+  });
+
+  describe('TriggerDrop', () => {
+    it('should work', () => {
+      expect(
+        transformTriggers({
+          type: 'TriggerDrop',
+          tableName: 'table1',
+          triggerName: 'trigger1',
+          reason: 'unknown',
+        }),
+      ).toEqual(`DROP TRIGGER "trigger1" ON "table1";`);
+    });
+  });
+});