first commit

packages/cli/test/integration/debug.controller.test.ts

@@ -0,0 +1,59 @@
+import { randomName, mockInstance } from '@n8n/backend-test-utils';
+import type { WorkflowEntity } from '@n8n/db';
+import { generateNanoId, WorkflowRepository } from '@n8n/db';
+import { Container } from '@n8n/di';
+import { InstanceSettings } from 'n8n-core';
+
+import { ActiveWorkflowManager } from '@/active-workflow-manager';
+import { MultiMainSetup } from '@/scaling/multi-main-setup.ee';
+
+import { createOwner } from './shared/db/users';
+import type { SuperAgentTest } from './shared/types';
+import { setupTestServer } from './shared/utils';
+
+describe('DebugController', () => {
+	const workflowRepository = mockInstance(WorkflowRepository);
+	const activeWorkflowManager = mockInstance(ActiveWorkflowManager);
+	const instanceSettings = Container.get(InstanceSettings);
+	instanceSettings.markAsLeader();
+
+	const testServer = setupTestServer({ endpointGroups: ['debug'] });
+	let ownerAgent: SuperAgentTest;
+
+	beforeAll(async () => {
+		const owner = await createOwner();
+		ownerAgent = testServer.authAgentFor(owner);
+		testServer.license.enable('feat:multipleMainInstances');
+	});
+
+	describe('GET /debug/multi-main-setup', () => {
+		test('should return multi-main setup details', async () => {
+			const workflowId = generateNanoId();
+			const webhooks = [{ id: workflowId, name: randomName() }] as WorkflowEntity[];
+			const triggersAndPollers = [{ id: workflowId, name: randomName() }] as WorkflowEntity[];
+			const activationErrors = { [workflowId]: 'Failed to activate' };
+			const { instanceId } = instanceSettings;
+			const leaderKey = 'some-leader-key';
+
+			workflowRepository.findIn.mockResolvedValue(triggersAndPollers);
+			workflowRepository.findWebhookBasedActiveWorkflows.mockResolvedValue(webhooks);
+			activeWorkflowManager.allActiveInMemory.mockReturnValue([workflowId]);
+			activeWorkflowManager.getAllWorkflowActivationErrors.mockResolvedValue(activationErrors);
+
+			jest.spyOn(MultiMainSetup.prototype, 'fetchLeaderKey').mockResolvedValue(leaderKey);
+
+			const response = await ownerAgent.get('/debug/multi-main-setup').expect(200);
+
+			expect(response.body.data).toMatchObject({
+				instanceId,
+				leaderKey,
+				isLeader: true,
+				activeWorkflows: {
+					webhooks,
+					triggersAndPollers,
+				},
+				activationErrors,
+			});
+		});
+	});
+});