alighasami
3d5eaf9445
Some checks failed
Security: Sync from Public / sync-from-public (push) Has been cancelled
Test: Benchmark Nightly / build (push) Has been cancelled
Test: Benchmark Nightly / Notify Cats on failure (push) Has been cancelled
CI: Python / Checks (push) Has been cancelled
Test: Evals Python / Workflow Comparison Python (push) Has been cancelled
Util: Check Docs URLs / check-docs-urls (push) Has been cancelled
Test: Visual Storybook / Cloudflare Pages (push) Has been cancelled
Test: E2E Performance / build-and-test-performance (push) Has been cancelled
Test: Workflows Nightly / Run Workflow Tests (push) Has been cancelled
Util: Cleanup CI Docker Images / Delete stale CI images (push) Has been cancelled
Test: Benchmark Destroy Env / build (push) Has been cancelled
Util: Update Node Popularity / update-popularity (push) Has been cancelled
Test: E2E Coverage Weekly / Coverage Tests (push) Has been cancelled
40 lines
1.1 KiB
TypeScript
40 lines
1.1 KiB
TypeScript
import { TaskRunnersConfig } from '@n8n/config';
|
|
import { Container } from '@n8n/di';
|
|
import request from 'supertest';
|
|
import type TestAgent from 'supertest/lib/agent';
|
|
|
|
import { TaskBrokerServer } from '@/task-runners/task-broker/task-broker-server';
|
|
|
|
export interface TestTaskBrokerServer {
|
|
server: TaskBrokerServer;
|
|
agent: TestAgent;
|
|
config: TaskRunnersConfig;
|
|
}
|
|
|
|
/**
|
|
* Sets up a Task Broker Server for testing purposes. The server needs
|
|
* to be started and stopped manually.
|
|
*
|
|
* @example
|
|
* const { server, agent, config } = setupBrokerTestServer();
|
|
*
|
|
* beforeAll(async () => await server.start());
|
|
* afterAll(async () => await server.stop());
|
|
*/
|
|
export const setupBrokerTestServer = (
|
|
config: Partial<TaskRunnersConfig> = {},
|
|
): TestTaskBrokerServer => {
|
|
const runnerConfig = Container.get(TaskRunnersConfig);
|
|
Object.assign(runnerConfig, config);
|
|
runnerConfig.port = 0; // Use any port
|
|
|
|
const taskBrokerServer = Container.get(TaskBrokerServer);
|
|
const agent = request.agent(taskBrokerServer.app);
|
|
|
|
return {
|
|
server: taskBrokerServer,
|
|
agent,
|
|
config: runnerConfig,
|
|
};
|
|
};
|