mudhorn/n8n
1
0
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2024-11-14 08:34:07 -08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add task broker tests

Browse source
This commit is contained in:
Iván Ovejero 2024-11-07 14:02:44 +01:00
parent 647bd61d05
commit b3db0c8a93
No known key found for this signature in database
1 changed files with 50 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
packages/cli/src/runners/__tests__/task-broker.test.ts
View file

@ -1,6 +1,8 @@
import type { TaskRunnersConfig } from '@n8n/config';
import type { RunnerMessage, TaskResultData } from '@n8n/task-runner';
import { mock } from 'jest-mock-extended';
import type { INodeTypeBaseDescription } from 'n8n-workflow';
import Container from 'typedi';
import { TaskRejectError } from '../errors';
import { TaskBroker } from '../task-broker.service';
@ -206,6 +208,54 @@ describe('TaskBroker', () => {
expect(taskBroker.acceptOffer).toHaveBeenCalled();
expect(taskBroker.getPendingTaskOffers()).toHaveLength(0);
});
['internal_childprocess', 'internal_launcher'].forEach((mode: TaskRunnersConfig['mode']) => {
it(`should manage lifecycle in \`${mode}\` mode`, async () => {
const RunnerLifecycleManager = {
ensureRunnerAvailable: jest.fn().mockResolvedValue(undefined),
updateLastActivityTime: jest.fn(),
};
jest.mock('@/runners/runner-lifecycle-manager', () => ({ RunnerLifecycleManager }));
jest.spyOn(Container, 'get').mockReturnValue(RunnerLifecycleManager);
const taskBroker = new TaskBroker(mock(), mock<TaskRunnersConfig>({ mode }));
const request: TaskRequest = {
requestId: 'request1',
requesterId: 'requester1',
taskType: 'test',
};
await taskBroker.taskRequested(request);
expect(RunnerLifecycleManager.ensureRunnerAvailable).toHaveBeenCalled();
expect(RunnerLifecycleManager.updateLastActivityTime).toHaveBeenCalled();
});
});
it('should not manage lifecycle in `external` mode', async () => {
const RunnerLifecycleManager = {
ensureRunnerAvailable: jest.fn().mockResolvedValue(undefined),
updateLastActivityTime: jest.fn(),
};
jest.mock('@/runners/runner-lifecycle-manager', () => ({ RunnerLifecycleManager }));
jest.spyOn(Container, 'get').mockReturnValue(RunnerLifecycleManager);
const taskBroker = new TaskBroker(mock(), mock<TaskRunnersConfig>({ mode: 'external' }));
const request: TaskRequest = {
requestId: 'request1',
requesterId: 'requester1',
taskType: 'test',
};
await taskBroker.taskRequested(request);
expect(RunnerLifecycleManager.ensureRunnerAvailable).not.toHaveBeenCalled();
expect(RunnerLifecycleManager.updateLastActivityTime).not.toHaveBeenCalled();
});
});
describe('taskOffered', () => {