ci: Setup cypress tasks for resetting DB, and setting up an owner (#4717)

* ci: Setup cypress tasks for resetting DB, and setting up an owner

* address Ivan's comments

packages/cli/src/Db.ts

@@ -32,7 +32,7 @@ import {
 export let isInitialized = false;
 export const collections = {} as IDatabaseCollections;
 
-let connection: Connection;
+export let connection: Connection;
 
 export async function transaction<T>(fn: (entityManager: EntityManager) => Promise<T>): Promise<T> {
 	return connection.transaction(fn);

packages/cli/src/Server.ts

@@ -270,6 +270,10 @@ class App {
 
 		setupErrorMiddleware(this.app);
 
+		if (process.env.E2E_TESTS === 'true') {
+			this.app.use('/e2e', require('./api/e2e.api').e2eController);
+		}
+
 		const urlBaseWebhook = WebhookHelpers.getWebhookBaseUrl();
 		const telemetrySettings: ITelemetrySettings = {
 			enabled: config.getEnv('diagnostics.enabled'),
@@ -431,6 +435,7 @@ class App {
 			'metrics',
 			'icons',
 			'types',
+			'e2e',
 			this.endpointWebhook,
 			this.endpointWebhookTest,
 			this.endpointPresetCredentials,

packages/cli/src/api/e2e.api.ts

@@ -0,0 +1,111 @@
+/* eslint-disable @typescript-eslint/no-unsafe-return */
+/* eslint-disable @typescript-eslint/no-unsafe-call */
+/* eslint-disable @typescript-eslint/no-unsafe-member-access */
+/* eslint-disable @typescript-eslint/no-unsafe-argument */
+/* eslint-disable @typescript-eslint/no-unsafe-assignment */
+/* eslint-disable @typescript-eslint/naming-convention */
+import { Router } from 'express';
+import bodyParser from 'body-parser';
+import { v4 as uuid } from 'uuid';
+import config from '@/config';
+import * as Db from '@/Db';
+import { Role } from '@/databases/entities/Role';
+import { hashPassword } from '@/UserManagement/UserManagementHelper';
+
+if (process.env.E2E_TESTS !== 'true') {
+	console.error('E2E endpoints only allowed during E2E tests');
+	process.exit(1);
+}
+
+const tablesToTruncate = [
+	'shared_workflow',
+	'shared_credentials',
+	'webhook_entity',
+	'workflows_tags',
+	'credentials_entity',
+	'tag_entity',
+	'workflow_entity',
+	'execution_entity',
+	'settings',
+	'installed_packages',
+	'installed_nodes',
+	'user',
+	'role',
+];
+
+const truncateAll = async () => {
+	const { connection } = Db;
+	for (const table of tablesToTruncate) {
+		await connection.query(
+			`DELETE FROM ${table}; DELETE FROM sqlite_sequence WHERE name=${table};`,
+		);
+	}
+	config.set('userManagement.isInstanceOwnerSetUp', false);
+};
+
+const setupUserManagement = async () => {
+	const { connection } = Db;
+	await connection.query('INSERT INTO role (name, scope) VALUES ("owner", "global");');
+	const instanceOwnerRole = (await connection.query(
+		'SELECT last_insert_rowid() as insertId',
+	)) as Array<{ insertId: number }>;
+
+	const roles: Array<[Role['name'], Role['scope']]> = [
+		['member', 'global'],
+		['owner', 'workflow'],
+		['owner', 'credential'],
+		['user', 'credential'],
+		['editor', 'workflow'],
+	];
+
+	await Promise.all(
+		roles.map(async ([name, scope]) =>
+			connection.query(`INSERT INTO role (name, scope) VALUES ("${name}", "${scope}");`),
+		),
+	);
+	await connection.query(
+		`INSERT INTO user (id, globalRoleId) values ("${uuid()}", ${instanceOwnerRole[0].insertId})`,
+	);
+	await connection.query(
+		`INSERT INTO "settings" (key, value, loadOnStartup) values ('userManagement.isInstanceOwnerSetUp', 'false', true), ('userManagement.skipInstanceOwnerSetup', 'false', true)`,
+	);
+};
+
+export const e2eController = Router();
+
+e2eController.post('/db/reset', async (req, res) => {
+	await truncateAll();
+	await setupUserManagement();
+
+	res.writeHead(204).end();
+});
+
+e2eController.post('/db/setup-owner', bodyParser.json(), async (req, res) => {
+	if (config.get('userManagement.isInstanceOwnerSetUp')) {
+		res.writeHead(500).send({ error: 'Owner already setup' });
+		return;
+	}
+
+	const globalRole = await Db.collections.Role.findOneOrFail({
+		name: 'owner',
+		scope: 'global',
+	});
+
+	const owner = await Db.collections.User.findOneOrFail({ globalRole });
+
+	await Db.collections.User.update(owner.id, {
+		email: req.body.email,
+		password: await hashPassword(req.body.password),
+		firstName: req.body.firstName,
+		lastName: req.body.lastName,
+	});
+
+	await Db.collections.Settings.update(
+		{ key: 'userManagement.isInstanceOwnerSetUp' },
+		{ value: 'true' },
+	);
+
+	config.set('userManagement.isInstanceOwnerSetUp', true);
+
+	res.writeHead(204).end();
+});

packages/cli/src/config/index.ts

@@ -13,6 +13,7 @@ const inE2ETests = process.env.E2E_TESTS === 'true';
 if (inE2ETests) {
 	// Skip loading config from env variables in end-to-end tests
 	process.env = {
+		E2E_TESTS: 'true',
 		N8N_USER_FOLDER: mkdtempSync(join(tmpdir(), 'n8n-e2e-')),
 		N8N_DIAGNOSTICS_ENABLED: 'false',
 		N8N_PUBLIC_API_DISABLED: 'true',