n8n/packages/nodes-base/nodes/Postgres/test/v2/runQueries.test.ts
2023-04-03 18:18:01 +03:00

54 lines
1.3 KiB
TypeScript

import { constructExecutionMetaData } from 'n8n-core';
import type { IDataObject, INode } from 'n8n-workflow';
import type { PgpDatabase } from '../../v2/helpers/interfaces';
import { configureQueryRunner } from '../../v2/helpers/utils';
import pgPromise from 'pg-promise';
const node: INode = {
id: '1',
name: 'Postgres node',
typeVersion: 2,
type: 'n8n-nodes-base.postgres',
position: [60, 760],
parameters: {
operation: 'executeQuery',
},
};
const createMockDb = (returnData: IDataObject | IDataObject[]) => {
return {
async any() {
return returnData;
},
async multi() {
return returnData;
},
async tx() {
return returnData;
},
async task() {
return returnData;
},
} as unknown as PgpDatabase;
};
describe('Test PostgresV2, runQueries', () => {
it('should execute, should return success true', async () => {
const pgp = pgPromise();
const db = createMockDb([]);
const dbMultiSpy = jest.spyOn(db, 'multi');
const runQueries = configureQueryRunner(node, constructExecutionMetaData, false, pgp, db);
const result = await runQueries([{ query: 'SELECT * FROM table', values: [] }], [], {});
expect(result).toBeDefined();
expect(result).toHaveLength(1);
expect(result).toEqual([{ json: { success: true } }]);
expect(dbMultiSpy).toHaveBeenCalledWith('SELECT * FROM table');
});
});