mirror of
https://github.com/n8n-io/n8n.git
synced 2024-11-10 22:54:05 -08:00
510 lines
17 KiB
TypeScript
510 lines
17 KiB
TypeScript
import type { SuperAgentTest } from 'supertest';
|
|
import { In } from 'typeorm';
|
|
import { UserSettings } from 'n8n-core';
|
|
import type { IUser } from 'n8n-workflow';
|
|
|
|
import * as Db from '@/Db';
|
|
import { RESPONSE_ERROR_MESSAGES } from '@/constants';
|
|
import type { CredentialWithSharings } from '@/credentials/credentials.types';
|
|
import * as UserManagementHelpers from '@/UserManagement/UserManagementHelper';
|
|
import type { Role } from '@db/entities/Role';
|
|
import type { User } from '@db/entities/User';
|
|
import { randomCredentialPayload } from './shared/random';
|
|
import * as testDb from './shared/testDb';
|
|
import type { AuthAgent, SaveCredentialFunction } from './shared/types';
|
|
import * as utils from './shared/utils';
|
|
|
|
let globalMemberRole: Role;
|
|
let owner: User;
|
|
let member: User;
|
|
let authOwnerAgent: SuperAgentTest;
|
|
let authAgent: AuthAgent;
|
|
let saveCredential: SaveCredentialFunction;
|
|
let sharingSpy: jest.SpyInstance<boolean>;
|
|
|
|
beforeAll(async () => {
|
|
const app = await utils.initTestServer({ endpointGroups: ['credentials'] });
|
|
|
|
await utils.initConfigFile();
|
|
|
|
const globalOwnerRole = await testDb.getGlobalOwnerRole();
|
|
globalMemberRole = await testDb.getGlobalMemberRole();
|
|
const credentialOwnerRole = await testDb.getCredentialOwnerRole();
|
|
|
|
owner = await testDb.createUser({ globalRole: globalOwnerRole });
|
|
member = await testDb.createUser({ globalRole: globalMemberRole });
|
|
|
|
authAgent = utils.createAuthAgent(app);
|
|
authOwnerAgent = authAgent(owner);
|
|
|
|
saveCredential = testDb.affixRoleToSaveCredential(credentialOwnerRole);
|
|
sharingSpy = jest.spyOn(UserManagementHelpers, 'isSharingEnabled').mockReturnValue(true);
|
|
});
|
|
|
|
beforeEach(async () => {
|
|
await testDb.truncate(['SharedCredentials', 'Credentials']);
|
|
});
|
|
|
|
afterAll(async () => {
|
|
await testDb.terminate();
|
|
});
|
|
|
|
// ----------------------------------------
|
|
// dynamic router switching
|
|
// ----------------------------------------
|
|
describe('router should switch based on flag', () => {
|
|
let savedCredentialId: string;
|
|
|
|
beforeEach(async () => {
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
savedCredentialId = savedCredential.id;
|
|
});
|
|
|
|
test('when sharing is disabled', async () => {
|
|
sharingSpy.mockReturnValueOnce(false);
|
|
|
|
await authOwnerAgent
|
|
.put(`/credentials/${savedCredentialId}/share`)
|
|
.send({ shareWithIds: [member.id] })
|
|
.expect(404);
|
|
|
|
await authOwnerAgent.get(`/credentials/${savedCredentialId}`).send().expect(200);
|
|
});
|
|
|
|
test('when sharing is enabled', async () => {
|
|
await authOwnerAgent
|
|
.put(`/credentials/${savedCredentialId}/share`)
|
|
.send({ shareWithIds: [member.id] })
|
|
.expect(200);
|
|
|
|
await authOwnerAgent.get(`/credentials/${savedCredentialId}`).send().expect(200);
|
|
});
|
|
});
|
|
|
|
// ----------------------------------------
|
|
// GET /credentials - fetch all credentials
|
|
// ----------------------------------------
|
|
describe('GET /credentials', () => {
|
|
test('should return all creds for owner', async () => {
|
|
const [member1, member2, member3] = await testDb.createManyUsers(3, {
|
|
globalRole: globalMemberRole,
|
|
});
|
|
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
await saveCredential(randomCredentialPayload(), { user: member1 });
|
|
|
|
const sharedWith = [member1, member2, member3];
|
|
await testDb.shareCredentialWithUsers(savedCredential, sharedWith);
|
|
|
|
const response = await authOwnerAgent.get('/credentials');
|
|
|
|
expect(response.statusCode).toBe(200);
|
|
expect(response.body.data).toHaveLength(2); // owner retrieved owner cred and member cred
|
|
|
|
const [ownerCredential, memberCredential] = response.body.data as CredentialWithSharings[];
|
|
|
|
validateMainCredentialData(ownerCredential);
|
|
expect(ownerCredential.data).toBeUndefined();
|
|
|
|
validateMainCredentialData(memberCredential);
|
|
expect(memberCredential.data).toBeUndefined();
|
|
|
|
expect(ownerCredential.ownedBy).toMatchObject({
|
|
id: owner.id,
|
|
email: owner.email,
|
|
firstName: owner.firstName,
|
|
lastName: owner.lastName,
|
|
});
|
|
|
|
expect(Array.isArray(ownerCredential.sharedWith)).toBe(true);
|
|
expect(ownerCredential.sharedWith).toHaveLength(3);
|
|
|
|
// Fix order issue (MySQL might return items in any order)
|
|
const ownerCredentialsSharedWithOrdered = [...ownerCredential.sharedWith!].sort(
|
|
(a: IUser, b: IUser) => (a.email < b.email ? -1 : 1),
|
|
);
|
|
const orderedSharedWith = [...sharedWith].sort((a, b) => (a.email < b.email ? -1 : 1));
|
|
|
|
ownerCredentialsSharedWithOrdered.forEach((sharee: IUser, idx: number) => {
|
|
expect(sharee).toMatchObject({
|
|
id: orderedSharedWith[idx].id,
|
|
email: orderedSharedWith[idx].email,
|
|
firstName: orderedSharedWith[idx].firstName,
|
|
lastName: orderedSharedWith[idx].lastName,
|
|
});
|
|
});
|
|
|
|
expect(memberCredential.ownedBy).toMatchObject({
|
|
id: member1.id,
|
|
email: member1.email,
|
|
firstName: member1.firstName,
|
|
lastName: member1.lastName,
|
|
});
|
|
|
|
expect(Array.isArray(memberCredential.sharedWith)).toBe(true);
|
|
expect(memberCredential.sharedWith).toHaveLength(0);
|
|
});
|
|
|
|
test('should return only relevant creds for member', async () => {
|
|
const [member1, member2] = await testDb.createManyUsers(2, {
|
|
globalRole: globalMemberRole,
|
|
});
|
|
|
|
await saveCredential(randomCredentialPayload(), { user: member2 });
|
|
const savedMemberCredential = await saveCredential(randomCredentialPayload(), {
|
|
user: member1,
|
|
});
|
|
|
|
await testDb.shareCredentialWithUsers(savedMemberCredential, [member2]);
|
|
|
|
const response = await authAgent(member1).get('/credentials');
|
|
|
|
expect(response.statusCode).toBe(200);
|
|
expect(response.body.data).toHaveLength(1); // member retrieved only member cred
|
|
|
|
const [member1Credential] = response.body.data;
|
|
|
|
validateMainCredentialData(member1Credential);
|
|
expect(member1Credential.data).toBeUndefined();
|
|
|
|
expect(member1Credential.ownedBy).toMatchObject({
|
|
id: member1.id,
|
|
email: member1.email,
|
|
firstName: member1.firstName,
|
|
lastName: member1.lastName,
|
|
});
|
|
|
|
expect(Array.isArray(member1Credential.sharedWith)).toBe(true);
|
|
expect(member1Credential.sharedWith).toHaveLength(1);
|
|
|
|
const [sharee] = member1Credential.sharedWith;
|
|
|
|
expect(sharee).toMatchObject({
|
|
id: member2.id,
|
|
email: member2.email,
|
|
firstName: member2.firstName,
|
|
lastName: member2.lastName,
|
|
});
|
|
});
|
|
});
|
|
|
|
// ----------------------------------------
|
|
// GET /credentials/:id - fetch a certain credential
|
|
// ----------------------------------------
|
|
describe('GET /credentials/:id', () => {
|
|
test('should retrieve owned cred for owner', async () => {
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
|
|
const firstResponse = await authOwnerAgent.get(`/credentials/${savedCredential.id}`);
|
|
|
|
expect(firstResponse.statusCode).toBe(200);
|
|
|
|
const { data: firstCredential } = firstResponse.body;
|
|
validateMainCredentialData(firstCredential);
|
|
expect(firstCredential.data).toBeUndefined();
|
|
expect(firstCredential.ownedBy).toMatchObject({
|
|
id: owner.id,
|
|
email: owner.email,
|
|
firstName: owner.firstName,
|
|
lastName: owner.lastName,
|
|
});
|
|
expect(firstCredential.sharedWith).toHaveLength(0);
|
|
|
|
const secondResponse = await authOwnerAgent
|
|
.get(`/credentials/${savedCredential.id}`)
|
|
.query({ includeData: true });
|
|
|
|
expect(secondResponse.statusCode).toBe(200);
|
|
|
|
const { data: secondCredential } = secondResponse.body;
|
|
validateMainCredentialData(secondCredential);
|
|
expect(secondCredential.data).toBeDefined();
|
|
});
|
|
|
|
test('should retrieve non-owned cred for owner', async () => {
|
|
const [member1, member2] = await testDb.createManyUsers(2, {
|
|
globalRole: globalMemberRole,
|
|
});
|
|
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: member1 });
|
|
await testDb.shareCredentialWithUsers(savedCredential, [member2]);
|
|
|
|
const response1 = await authOwnerAgent.get(`/credentials/${savedCredential.id}`);
|
|
|
|
expect(response1.statusCode).toBe(200);
|
|
|
|
validateMainCredentialData(response1.body.data);
|
|
expect(response1.body.data.data).toBeUndefined();
|
|
expect(response1.body.data.ownedBy).toMatchObject({
|
|
id: member1.id,
|
|
email: member1.email,
|
|
firstName: member1.firstName,
|
|
lastName: member1.lastName,
|
|
});
|
|
expect(response1.body.data.sharedWith).toHaveLength(1);
|
|
expect(response1.body.data.sharedWith[0]).toMatchObject({
|
|
id: member2.id,
|
|
email: member2.email,
|
|
firstName: member2.firstName,
|
|
lastName: member2.lastName,
|
|
});
|
|
|
|
const response2 = await authOwnerAgent
|
|
.get(`/credentials/${savedCredential.id}`)
|
|
.query({ includeData: true });
|
|
|
|
expect(response2.statusCode).toBe(200);
|
|
|
|
validateMainCredentialData(response2.body.data);
|
|
expect(response2.body.data.data).toBeUndefined();
|
|
expect(response2.body.data.sharedWith).toHaveLength(1);
|
|
});
|
|
|
|
test('should retrieve owned cred for member', async () => {
|
|
const [member1, member2, member3] = await testDb.createManyUsers(3, {
|
|
globalRole: globalMemberRole,
|
|
});
|
|
const authMemberAgent = authAgent(member1);
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: member1 });
|
|
await testDb.shareCredentialWithUsers(savedCredential, [member2, member3]);
|
|
|
|
const firstResponse = await authMemberAgent.get(`/credentials/${savedCredential.id}`);
|
|
|
|
expect(firstResponse.statusCode).toBe(200);
|
|
|
|
const { data: firstCredential } = firstResponse.body;
|
|
validateMainCredentialData(firstCredential);
|
|
expect(firstCredential.data).toBeUndefined();
|
|
expect(firstCredential.ownedBy).toMatchObject({
|
|
id: member1.id,
|
|
email: member1.email,
|
|
firstName: member1.firstName,
|
|
lastName: member1.lastName,
|
|
});
|
|
expect(firstCredential.sharedWith).toHaveLength(2);
|
|
firstCredential.sharedWith.forEach((sharee: IUser, idx: number) => {
|
|
expect([member2.id, member3.id]).toContain(sharee.id);
|
|
});
|
|
|
|
const secondResponse = await authMemberAgent
|
|
.get(`/credentials/${savedCredential.id}`)
|
|
.query({ includeData: true });
|
|
|
|
expect(secondResponse.statusCode).toBe(200);
|
|
|
|
const { data: secondCredential } = secondResponse.body;
|
|
validateMainCredentialData(secondCredential);
|
|
expect(secondCredential.data).toBeDefined();
|
|
expect(firstCredential.sharedWith).toHaveLength(2);
|
|
});
|
|
|
|
test('should not retrieve non-owned cred for member', async () => {
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
|
|
const response = await authAgent(member).get(`/credentials/${savedCredential.id}`);
|
|
|
|
expect(response.statusCode).toBe(403);
|
|
expect(response.body.data).toBeUndefined(); // owner's cred not returned
|
|
});
|
|
|
|
test('should fail with missing encryption key', async () => {
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
|
|
const mock = jest.spyOn(UserSettings, 'getEncryptionKey');
|
|
mock.mockRejectedValue(new Error(RESPONSE_ERROR_MESSAGES.NO_ENCRYPTION_KEY));
|
|
|
|
const response = await authOwnerAgent
|
|
.get(`/credentials/${savedCredential.id}`)
|
|
.query({ includeData: true });
|
|
|
|
expect(response.statusCode).toBe(500);
|
|
|
|
mock.mockRestore();
|
|
});
|
|
|
|
test('should return 404 if cred not found', async () => {
|
|
const response = await authOwnerAgent.get('/credentials/789');
|
|
expect(response.statusCode).toBe(404);
|
|
|
|
const responseAbc = await authOwnerAgent.get('/credentials/abc');
|
|
expect(responseAbc.statusCode).toBe(404);
|
|
|
|
// because EE router has precedence, check if forwards this route
|
|
const responseNew = await authOwnerAgent.get('/credentials/new');
|
|
expect(responseNew.statusCode).toBe(200);
|
|
});
|
|
});
|
|
|
|
// ----------------------------------------
|
|
// idempotent share/unshare
|
|
// ----------------------------------------
|
|
describe('PUT /credentials/:id/share', () => {
|
|
test('should share the credential with the provided userIds and unshare it for missing ones', async () => {
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
|
|
const [member1, member2, member3, member4, member5] = await testDb.createManyUsers(5, {
|
|
globalRole: globalMemberRole,
|
|
});
|
|
const shareWithIds = [member1.id, member2.id, member3.id];
|
|
|
|
await testDb.shareCredentialWithUsers(savedCredential, [member4, member5]);
|
|
|
|
const response = await authOwnerAgent
|
|
.put(`/credentials/${savedCredential.id}/share`)
|
|
.send({ shareWithIds });
|
|
|
|
expect(response.statusCode).toBe(200);
|
|
expect(response.body.data).toBeUndefined();
|
|
|
|
const sharedCredentials = await Db.collections.SharedCredentials.find({
|
|
relations: ['role'],
|
|
where: { credentialsId: savedCredential.id },
|
|
});
|
|
|
|
// check that sharings have been removed/added correctly
|
|
expect(sharedCredentials.length).toBe(shareWithIds.length + 1); // +1 for the owner
|
|
|
|
sharedCredentials.forEach((sharedCredential) => {
|
|
if (sharedCredential.userId === owner.id) {
|
|
expect(sharedCredential.role.name).toBe('owner');
|
|
expect(sharedCredential.role.scope).toBe('credential');
|
|
return;
|
|
}
|
|
expect(shareWithIds).toContain(sharedCredential.userId);
|
|
expect(sharedCredential.role.name).toBe('user');
|
|
expect(sharedCredential.role.scope).toBe('credential');
|
|
});
|
|
});
|
|
|
|
test('should share the credential with the provided userIds', async () => {
|
|
const [member1, member2, member3] = await testDb.createManyUsers(3, {
|
|
globalRole: globalMemberRole,
|
|
});
|
|
const memberIds = [member1.id, member2.id, member3.id];
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
|
|
const response = await authOwnerAgent
|
|
.put(`/credentials/${savedCredential.id}/share`)
|
|
.send({ shareWithIds: memberIds });
|
|
|
|
expect(response.statusCode).toBe(200);
|
|
expect(response.body.data).toBeUndefined();
|
|
|
|
// check that sharings got correctly set in DB
|
|
const sharedCredentials = await Db.collections.SharedCredentials.find({
|
|
relations: ['role'],
|
|
where: { credentialsId: savedCredential.id, userId: In([...memberIds]) },
|
|
});
|
|
|
|
expect(sharedCredentials.length).toBe(memberIds.length);
|
|
|
|
sharedCredentials.forEach((sharedCredential) => {
|
|
expect(sharedCredential.role.name).toBe('user');
|
|
expect(sharedCredential.role.scope).toBe('credential');
|
|
});
|
|
|
|
// check that owner still exists
|
|
const ownerSharedCredential = await Db.collections.SharedCredentials.findOneOrFail({
|
|
relations: ['role'],
|
|
where: { credentialsId: savedCredential.id, userId: owner.id },
|
|
});
|
|
|
|
expect(ownerSharedCredential.role.name).toBe('owner');
|
|
expect(ownerSharedCredential.role.scope).toBe('credential');
|
|
});
|
|
|
|
test('should respond 403 for non-existing credentials', async () => {
|
|
const response = await authOwnerAgent
|
|
.put('/credentials/1234567/share')
|
|
.send({ shareWithIds: [member.id] });
|
|
|
|
expect(response.statusCode).toBe(403);
|
|
});
|
|
|
|
test('should respond 403 for non-owned credentials', async () => {
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: member });
|
|
|
|
const response = await authOwnerAgent
|
|
.put(`/credentials/${savedCredential.id}/share`)
|
|
.send({ shareWithIds: [member.id] });
|
|
|
|
expect(response.statusCode).toBe(403);
|
|
});
|
|
|
|
test('should ignore pending sharee', async () => {
|
|
const memberShell = await testDb.createUserShell(globalMemberRole);
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
|
|
const response = await authOwnerAgent
|
|
.put(`/credentials/${savedCredential.id}/share`)
|
|
.send({ shareWithIds: [memberShell.id] });
|
|
|
|
expect(response.statusCode).toBe(200);
|
|
|
|
const sharedCredentials = await Db.collections.SharedCredentials.find({
|
|
where: { credentialsId: savedCredential.id },
|
|
});
|
|
|
|
expect(sharedCredentials).toHaveLength(1);
|
|
expect(sharedCredentials[0].userId).toBe(owner.id);
|
|
});
|
|
|
|
test('should ignore non-existing sharee', async () => {
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
|
|
const response = await authOwnerAgent
|
|
.put(`/credentials/${savedCredential.id}/share`)
|
|
.send({ shareWithIds: ['bce38a11-5e45-4d1c-a9ee-36e4a20ab0fc'] });
|
|
|
|
expect(response.statusCode).toBe(200);
|
|
|
|
const sharedCredentials = await Db.collections.SharedCredentials.find({
|
|
where: { credentialsId: savedCredential.id },
|
|
});
|
|
|
|
expect(sharedCredentials).toHaveLength(1);
|
|
expect(sharedCredentials[0].userId).toBe(owner.id);
|
|
});
|
|
|
|
test('should respond 400 if invalid payload is provided', async () => {
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
|
|
const responses = await Promise.all([
|
|
authOwnerAgent.put(`/credentials/${savedCredential.id}/share`).send(),
|
|
authOwnerAgent.put(`/credentials/${savedCredential.id}/share`).send({ shareWithIds: [1] }),
|
|
]);
|
|
|
|
responses.forEach((response) => expect(response.statusCode).toBe(400));
|
|
});
|
|
test('should unshare the credential', async () => {
|
|
const savedCredential = await saveCredential(randomCredentialPayload(), { user: owner });
|
|
|
|
const [member1, member2] = await testDb.createManyUsers(2, {
|
|
globalRole: globalMemberRole,
|
|
});
|
|
|
|
await testDb.shareCredentialWithUsers(savedCredential, [member1, member2]);
|
|
|
|
const response = await authOwnerAgent
|
|
.put(`/credentials/${savedCredential.id}/share`)
|
|
.send({ shareWithIds: [] });
|
|
|
|
expect(response.statusCode).toBe(200);
|
|
|
|
const sharedCredentials = await Db.collections.SharedCredentials.find({
|
|
where: { credentialsId: savedCredential.id },
|
|
});
|
|
|
|
expect(sharedCredentials).toHaveLength(1);
|
|
expect(sharedCredentials[0].userId).toBe(owner.id);
|
|
});
|
|
});
|
|
|
|
function validateMainCredentialData(credential: CredentialWithSharings) {
|
|
expect(typeof credential.name).toBe('string');
|
|
expect(typeof credential.type).toBe('string');
|
|
expect(typeof credential.nodesAccess[0].nodeType).toBe('string');
|
|
expect(credential.ownedBy).toBeDefined();
|
|
expect(Array.isArray(credential.sharedWith)).toBe(true);
|
|
}
|