admin/n8n-enterprise-unlocked
1
0
Fork 0
mirror of https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git synced 2025-12-17 10:02:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(core): Extract poll context out of NodeExecutionFunctions (no-changelog) (#11449)

Browse Source
This commit is contained in:
कारतोफ्फेलस्क्रिप्ट™
2024-11-04 09:49:52 +01:00
committed by GitHub
parent ce963e8824
commit 2104fa1733
16 changed files with 1222 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
packages/nodes-base/test/nodes/TriggerHelpers.ts
View File

@@ -3,7 +3,7 @@ import { mock } from 'jest-mock-extended';
import get from 'lodash/get';
import merge from 'lodash/merge';
import set from 'lodash/set';
import { getExecutePollFunctions, returnJsonArray, type InstanceSettings } from 'n8n-core';
import { PollContext, returnJsonArray, type InstanceSettings } from 'n8n-core';
import { ScheduledTaskManager } from 'n8n-core/dist/ScheduledTaskManager';
import type {
IBinaryData,
@@ -13,7 +13,6 @@ import type {
INode,
INodeType,
INodeTypes,
IPollFunctions,
ITriggerFunctions,
IWebhookFunctions,
IWorkflowExecuteAdditionalData,
@@ -193,14 +192,15 @@ export async function testPollingTriggerNode(
options.node,
) as INode;
const workflow = mock<Workflow>({
timezone: options.timezone ?? 'Europe/Berlin',
timezone,
nodeTypes: mock<INodeTypes>({
getByNameAndVersion: () => mock<INodeType>({ description: trigger.description }),
}),
getStaticData: () => options.workflowStaticData ?? {},
});
const mode = options.mode ?? 'trigger';
const originalPollingFunctions = getExecutePollFunctions(
const pollContext = new PollContext(
workflow,
node,
mock<IWorkflowExecuteAdditionalData>({
@@ -218,22 +218,13 @@ export async function testPollingTriggerNode(
'init',
);
async function getCredentials<T extends object = ICredentialDataDecryptedObject>(): Promise<T> {
return (options.credential ?? {}) as T;
}
pollContext.getNode = () => node;
pollContext.getCredentials = async <T extends object = ICredentialDataDecryptedObject>() =>
(options.credential ?? {}) as T;
pollContext.getNodeParameter = (parameterName, fallback) =>
get(node.parameters, parameterName) ?? fallback;
const pollingFunctions = mock<IPollFunctions>({
...originalPollingFunctions,
getCredentials,
getTimezone: () => timezone,
getNode: () => node,
getMode: () => mode,
getInstanceId: () => 'instanceId',
getWorkflowStaticData: () => options.workflowStaticData ?? {},
getNodeParameter: (parameterName, fallback) => get(node.parameters, parameterName) ?? fallback,
});
const response = await trigger.poll?.call(pollingFunctions);
const response = await trigger.poll?.call(pollContext);
return {
response,