n8n/packages/cli/test/integration/workflows.controller.test.ts

import type { SuperAgentTest } from 'supertest';
import type { IPinData } from 'n8n-workflow';

import type { User } from '@db/entities/User';
import * as UserManagementHelpers from '@/UserManagement/UserManagementHelper';

import * as utils from './shared/utils';
import * as testDb from './shared/testDb';
import { makeWorkflow, MOCK_PINDATA } from './shared/utils';
import * as Db from '@/Db';

let ownerShell: User;
let authOwnerAgent: SuperAgentTest;

beforeAll(async () => {
	const app = await utils.initTestServer({ endpointGroups: ['workflows'] });
	const globalOwnerRole = await testDb.getGlobalOwnerRole();
	ownerShell = await testDb.createUserShell(globalOwnerRole);
	authOwnerAgent = utils.createAgent(app, { auth: true, user: ownerShell });

	// mock whether sharing is enabled or not
	jest.spyOn(UserManagementHelpers, 'isSharingEnabled').mockReturnValue(false);
});

beforeEach(async () => {
	await testDb.truncate(['Workflow', 'SharedWorkflow']);
});

afterAll(async () => {
	await testDb.terminate();
});

describe('POST /workflows', () => {
	test('should store pin data for node in workflow', async () => {
		const workflow = makeWorkflow({ withPinData: true });

		const response = await authOwnerAgent.post('/workflows').send(workflow);

		expect(response.statusCode).toBe(200);

		const { pinData } = response.body.data as { pinData: IPinData };

		expect(pinData).toMatchObject(MOCK_PINDATA);
	});

	test('should set pin data to null if no pin data', async () => {
		const workflow = makeWorkflow({ withPinData: false });

		const response = await authOwnerAgent.post('/workflows').send(workflow);

		expect(response.statusCode).toBe(200);

		const { pinData } = response.body.data as { pinData: IPinData };

		expect(pinData).toBeNull();
	});
});

describe('GET /workflows/:id', () => {
	test('should return pin data', async () => {
		const workflow = makeWorkflow({ withPinData: true });

		const workflowCreationResponse = await authOwnerAgent.post('/workflows').send(workflow);

		const { id } = workflowCreationResponse.body.data as { id: string };

		const sw = await Db.collections.SharedWorkflow.find();

		const workflowRetrievalResponse = await authOwnerAgent.get(`/workflows/${id}`);

		expect(workflowRetrievalResponse.statusCode).toBe(200);

		const { pinData } = workflowRetrievalResponse.body.data as { pinData: IPinData };

		expect(pinData).toMatchObject(MOCK_PINDATA);
	});
});