n8n-enterprise-unlocked/packages/nodes-base/nodes/Google/BigQuery/test/v2/node/executeQuery.test.ts

import type { IHttpRequestMethods, INodeTypes } from 'n8n-workflow';

import { executeWorkflow } from '@test/nodes/ExecuteWorkflow';
import { setup, workflowToTests } from '@test/nodes/Helpers';
import type { WorkflowTestData } from '@test/nodes/types';

import * as transport from '../../../v2/transport';

jest.mock('../../../v2/transport', () => {
	const originalModule = jest.requireActual('../../../v2/transport');
	return {
		...originalModule,
		googleBigQueryApiRequest: jest.fn(async (method: IHttpRequestMethods, resource: string) => {
			if (resource === '/v2/projects/test-project/jobs' && method === 'POST') {
				return {
					jobReference: {
						jobId: 'job_123',
					},
					status: {
						state: 'DONE',
					},
				};
			}
			if (resource === '/v2/projects/test-project/queries/job_123' && method === 'GET') {
				return {};
			}
		}),
		googleBigQueryApiRequestAllItems: jest.fn(async () => ({ rows: [], schema: {} })),
	};
});

describe('Test Google BigQuery V2, executeQuery', () => {
	const workflows = ['nodes/Google/BigQuery/test/v2/node/executeQuery.workflow.json'];
	const tests = workflowToTests(workflows);
	const nodeTypes = setup(tests);

	const testNode = async (testData: WorkflowTestData, types: INodeTypes) => {
		const { result } = await executeWorkflow(testData, types);

		expect(transport.googleBigQueryApiRequest).toHaveBeenCalledTimes(1);
		expect(transport.googleBigQueryApiRequestAllItems).toHaveBeenCalledTimes(1);
		expect(transport.googleBigQueryApiRequest).toHaveBeenCalledWith(
			'POST',
			'/v2/projects/test-project/jobs',
			{
				configuration: {
					query: {
						query: 'SELECT * FROM bigquery_node_dev_test_dataset.test_json;',
						useLegacySql: false,
					},
				},
			},
		);
		expect(transport.googleBigQueryApiRequestAllItems).toHaveBeenCalledWith(
			'GET',
			'/v2/projects/test-project/queries/job_123',
			undefined,
			{ location: undefined, maxResults: 1000, timeoutMs: 10000 },
		);

		expect(result.finished).toEqual(true);
	};

	for (const testData of tests) {
		test(testData.description, async () => await testNode(testData, nodeTypes));
	}
});