n8n-enterprise-unlocked/packages/nodes-base/nodes/Google/Drive/test/v2/node/helpers.ts

import get from 'lodash/get';
import { constructExecutionMetaData } from 'n8n-core';
import type { IDataObject, IExecuteFunctions, IGetNodeParameterOptions, INode } from 'n8n-workflow';
import { Readable } from 'stream';

export const driveNode: INode = {
	id: '11',
	name: 'Google Drive node',
	typeVersion: 3,
	type: 'n8n-nodes-base.googleDrive',
	position: [42, 42],
	parameters: {},
};

export const createMockExecuteFunction = (
	nodeParameters: IDataObject,
	node: INode,
	continueOnFail = false,
) => {
	const fakeExecuteFunction = {
		getNodeParameter(
			parameterName: string,
			_itemIndex: number,
			fallbackValue?: IDataObject,
			options?: IGetNodeParameterOptions,
		) {
			const parameter = options?.extractValue ? `${parameterName}.value` : parameterName;
			return get(nodeParameters, parameter, fallbackValue);
		},
		getNode() {
			return node;
		},
		helpers: {
			constructExecutionMetaData,
			returnJsonArray: () => [],
			prepareBinaryData: () => {},
			httpRequest: () => {},
		},
		continueOnFail: () => continueOnFail,
	} as unknown as IExecuteFunctions;
	return fakeExecuteFunction;
};

export function createTestStream(byteSize: number) {
	let bytesSent = 0;
	const CHUNK_SIZE = 64 * 1024; // 64kB chunks (default NodeJS highWaterMark)

	return new Readable({
		read() {
			const remainingBytes = byteSize - bytesSent;

			if (remainingBytes <= 0) {
				this.push(null);
				return;
			}

			const chunkSize = Math.min(CHUNK_SIZE, remainingBytes);
			const chunk = Buffer.alloc(chunkSize, 'A'); // Test data just a string of "A"

			bytesSent += chunkSize;
			this.push(chunk);
		},
	});
}