chore(core): Rename Data Store module to Data Table (no-changelog) (#18675)

packages/cli/src/modules/data-table/__tests__/sql-utils.test.ts

@@ -0,0 +1,121 @@
+import type { DataStoreRows } from 'n8n-workflow';
+
+import { addColumnQuery, deleteColumnQuery, splitRowsByExistence } from '../utils/sql-utils';
+
+describe('sql-utils', () => {
+	describe('addColumnQuery', () => {
+		it('should generate a valid SQL query for adding columns to a table, sqlite', () => {
+			const tableName = 'data_store_user_abc';
+			const column = { name: 'email', type: 'number' as const };
+
+			const query = addColumnQuery(tableName, column, 'sqlite');
+
+			expect(query).toBe('ALTER TABLE "data_store_user_abc" ADD "email" REAL');
+		});
+
+		it('should generate a valid SQL query for adding columns to a table, postgres', () => {
+			const tableName = 'data_store_user_abc';
+			const column = { name: 'email', type: 'number' as const };
+
+			const query = addColumnQuery(tableName, column, 'postgres');
+
+			expect(query).toBe('ALTER TABLE "data_store_user_abc" ADD "email" DOUBLE PRECISION');
+		});
+
+		it('should generate a valid SQL query for adding columns to a table, mysql', () => {
+			const tableName = 'data_store_user_abc';
+			const column = { name: 'email', type: 'number' as const };
+
+			const query = addColumnQuery(tableName, column, 'mysql');
+
+			expect(query).toBe('ALTER TABLE `data_store_user_abc` ADD `email` DOUBLE');
+		});
+
+		it('should generate a valid SQL query for adding columns to a table, mariadb', () => {
+			const tableName = 'data_store_user_abc';
+			const column = { name: 'email', type: 'number' as const };
+
+			const query = addColumnQuery(tableName, column, 'mariadb');
+
+			expect(query).toBe('ALTER TABLE `data_store_user_abc` ADD `email` DOUBLE');
+		});
+	});
+
+	describe('deleteColumnQuery', () => {
+		it('should generate a valid SQL query for deleting columns from a table', () => {
+			const tableName = 'data_store_user_abc';
+			const column = 'email';
+
+			const query = deleteColumnQuery(tableName, column, 'sqlite');
+
+			expect(query).toBe('ALTER TABLE "data_store_user_abc" DROP COLUMN "email"');
+		});
+	});
+
+	describe('splitRowsByExistence', () => {
+		it('should correctly separate rows into insert and update based on matchFields', () => {
+			const existing = [
+				{ name: 'Alice', age: 30 },
+				{ name: 'Bob', age: 25 },
+			];
+			const matchFields = ['name'];
+			const rows: DataStoreRows = [
+				{ name: 'Alice', age: 30 },
+				{ name: 'Bob', age: 26 },
+				{ name: 'Charlie', age: 35 },
+			];
+
+			const { rowsToInsert, rowsToUpdate } = splitRowsByExistence(existing, matchFields, rows);
+
+			expect(rowsToUpdate).toEqual([
+				{ name: 'Alice', age: 30 },
+				{ name: 'Bob', age: 26 },
+			]);
+			expect(rowsToInsert).toEqual([{ name: 'Charlie', age: 35 }]);
+		});
+
+		it('should treat rows as new if matchFields combination does not exist', () => {
+			const existing = [{ name: 'Bob', city: 'Berlin' }];
+			const matchFields = ['name', 'city'];
+			const rows: DataStoreRows = [{ name: 'Alice', city: 'Berlin' }];
+
+			const { rowsToInsert, rowsToUpdate } = splitRowsByExistence(existing, matchFields, rows);
+
+			expect(rowsToUpdate).toEqual([]);
+			expect(rowsToInsert).toEqual([{ name: 'Alice', city: 'Berlin' }]);
+		});
+
+		it('should insert all rows if existing set is empty', () => {
+			const existing: Array<Record<string, unknown>> = [];
+			const matchFields = ['name'];
+			const rows: DataStoreRows = [{ name: 'Alice' }, { name: 'Bob' }];
+
+			const { rowsToInsert, rowsToUpdate } = splitRowsByExistence(existing, matchFields, rows);
+
+			expect(rowsToUpdate).toEqual([]);
+			expect(rowsToInsert).toEqual(rows);
+		});
+
+		it('should update all rows if all keys match existing', () => {
+			const existing = [{ name: 'Alice' }, { name: 'Bob' }];
+			const matchFields = ['name'];
+			const rows: DataStoreRows = [{ name: 'Alice' }, { name: 'Bob' }];
+
+			const { rowsToInsert, rowsToUpdate } = splitRowsByExistence(existing, matchFields, rows);
+
+			expect(rowsToInsert).toEqual([]);
+			expect(rowsToUpdate).toEqual(rows);
+		});
+
+		it('should handle empty input rows', () => {
+			const existing = [{ name: 'Alice' }];
+			const matchFields = ['name'];
+			const rows: DataStoreRows = [];
+
+			const { rowsToInsert, rowsToUpdate } = splitRowsByExistence(existing, matchFields, rows);
+
+			expect(rowsToInsert).toEqual([]);
+			expect(rowsToUpdate).toEqual([]);
+		});
+	});
+});