mirror of
https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git
synced 2025-12-17 18:12:04 +00:00
feat(core): Add health endpoint for task broker server (no-changelog) (#11920)
This commit is contained in:
@@ -0,0 +1,40 @@
|
||||
import { TaskRunnersConfig } from '@n8n/config';
|
||||
import request from 'supertest';
|
||||
import type TestAgent from 'supertest/lib/agent';
|
||||
import Container from 'typedi';
|
||||
|
||||
import { TaskRunnerServer } from '@/runners/task-runner-server';
|
||||
|
||||
export interface TestTaskBrokerServer {
|
||||
server: TaskRunnerServer;
|
||||
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.enabled = true;
|
||||
runnerConfig.port = 0; // Use any port
|
||||
|
||||
const taskRunnerServer = Container.get(TaskRunnerServer);
|
||||
const agent = request.agent(taskRunnerServer.app);
|
||||
|
||||
return {
|
||||
server: taskRunnerServer,
|
||||
agent,
|
||||
config: runnerConfig,
|
||||
};
|
||||
};
|
||||
Reference in New Issue
Block a user