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

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([]);
		});
	});
});