mirror of
https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git
synced 2025-12-17 01:56:46 +00:00
fix(core): Prevent calling internal hook email event if emailing is disabled (#8462)
Co-authored-by: कारतोफ्फेलस्क्रिप्ट™ <aditya@netroy.in>
This commit is contained in:
@@ -0,0 +1,29 @@
|
||||
import validator from 'validator';
|
||||
import type { User } from '@/databases/entities/User';
|
||||
import type { UserInvitationResult } from '../../shared/utils/users';
|
||||
|
||||
export function assertReturnedUserProps(user: User) {
|
||||
expect(validator.isUUID(user.id)).toBe(true);
|
||||
expect(user.email).toBeDefined();
|
||||
expect(user.personalizationAnswers).toBeNull();
|
||||
expect(user.password).toBeUndefined();
|
||||
expect(user.isPending).toBe(false);
|
||||
expect(user.apiKey).not.toBeDefined();
|
||||
expect(user.globalScopes).toBeDefined();
|
||||
expect(user.globalScopes).not.toHaveLength(0);
|
||||
}
|
||||
|
||||
export const assertStoredUserProps = (user: User) => {
|
||||
expect(user.firstName).toBeNull();
|
||||
expect(user.lastName).toBeNull();
|
||||
expect(user.personalizationAnswers).toBeNull();
|
||||
expect(user.password).toBeNull();
|
||||
expect(user.isPending).toBe(true);
|
||||
};
|
||||
|
||||
export const assertUserInviteResult = (data: UserInvitationResult) => {
|
||||
expect(validator.isUUID(data.user.id)).toBe(true);
|
||||
expect(data.user.inviteAcceptUrl).toBeUndefined();
|
||||
expect(data.user.email).toBeDefined();
|
||||
expect(data.user.emailSent).toBe(true);
|
||||
};
|
||||
@@ -0,0 +1,412 @@
|
||||
import { mocked } from 'jest-mock';
|
||||
import Container from 'typedi';
|
||||
import { Not } from 'typeorm';
|
||||
|
||||
import { InternalHooks } from '@/InternalHooks';
|
||||
import { ExternalHooks } from '@/ExternalHooks';
|
||||
import { UserManagementMailer } from '@/UserManagement/email';
|
||||
import { UserRepository } from '@/databases/repositories/user.repository';
|
||||
import { PasswordUtility } from '@/services/password.utility';
|
||||
|
||||
import {
|
||||
randomEmail,
|
||||
randomInvalidPassword,
|
||||
randomName,
|
||||
randomValidPassword,
|
||||
} from '../../shared/random';
|
||||
import { createMember, createOwner, createUserShell } from '../../shared/db/users';
|
||||
import { mockInstance } from '../../../shared/mocking';
|
||||
import * as utils from '../../shared/utils';
|
||||
|
||||
import {
|
||||
assertReturnedUserProps,
|
||||
assertStoredUserProps,
|
||||
assertUserInviteResult,
|
||||
} from './assertions';
|
||||
|
||||
import type { User } from '@/databases/entities/User';
|
||||
import type { UserInvitationResult } from '../../shared/utils/users';
|
||||
|
||||
describe('InvitationController', () => {
|
||||
const mailer = mockInstance(UserManagementMailer);
|
||||
const externalHooks = mockInstance(ExternalHooks);
|
||||
const internalHooks = mockInstance(InternalHooks);
|
||||
|
||||
const testServer = utils.setupTestServer({ endpointGroups: ['invitations'] });
|
||||
|
||||
let instanceOwner: User;
|
||||
let userRepository: UserRepository;
|
||||
|
||||
beforeAll(async () => {
|
||||
userRepository = Container.get(UserRepository);
|
||||
instanceOwner = await createOwner();
|
||||
});
|
||||
|
||||
beforeEach(async () => {
|
||||
jest.clearAllMocks();
|
||||
await userRepository.delete({ role: Not('global:owner') });
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
jest.restoreAllMocks();
|
||||
});
|
||||
|
||||
describe('POST /invitations/:id/accept', () => {
|
||||
test('should fill out a member shell', async () => {
|
||||
const memberShell = await createUserShell('global:member');
|
||||
|
||||
const memberProps = {
|
||||
inviterId: instanceOwner.id,
|
||||
firstName: randomName(),
|
||||
lastName: randomName(),
|
||||
password: randomValidPassword(),
|
||||
};
|
||||
|
||||
const response = await testServer.authlessAgent
|
||||
.post(`/invitations/${memberShell.id}/accept`)
|
||||
.send(memberProps)
|
||||
.expect(200);
|
||||
|
||||
const { data: returnedMember } = response.body;
|
||||
|
||||
assertReturnedUserProps(returnedMember);
|
||||
|
||||
expect(returnedMember.firstName).toBe(memberProps.firstName);
|
||||
expect(returnedMember.lastName).toBe(memberProps.lastName);
|
||||
expect(returnedMember.role).toBe('global:member');
|
||||
expect(utils.getAuthToken(response)).toBeDefined();
|
||||
|
||||
const storedMember = await userRepository.findOneByOrFail({ id: returnedMember.id });
|
||||
|
||||
expect(storedMember.firstName).toBe(memberProps.firstName);
|
||||
expect(storedMember.lastName).toBe(memberProps.lastName);
|
||||
expect(storedMember.password).not.toBe(memberProps.password);
|
||||
});
|
||||
|
||||
test('should fill out an admin shell', async () => {
|
||||
const adminShell = await createUserShell('global:admin');
|
||||
|
||||
const memberProps = {
|
||||
inviterId: instanceOwner.id,
|
||||
firstName: randomName(),
|
||||
lastName: randomName(),
|
||||
password: randomValidPassword(),
|
||||
};
|
||||
|
||||
const response = await testServer.authlessAgent
|
||||
.post(`/invitations/${adminShell.id}/accept`)
|
||||
.send(memberProps)
|
||||
.expect(200);
|
||||
|
||||
const { data: returnedAdmin } = response.body;
|
||||
|
||||
assertReturnedUserProps(returnedAdmin);
|
||||
|
||||
expect(returnedAdmin.firstName).toBe(memberProps.firstName);
|
||||
expect(returnedAdmin.lastName).toBe(memberProps.lastName);
|
||||
expect(returnedAdmin.role).toBe('global:admin');
|
||||
expect(utils.getAuthToken(response)).toBeDefined();
|
||||
|
||||
const storedAdmin = await userRepository.findOneByOrFail({ id: returnedAdmin.id });
|
||||
|
||||
expect(storedAdmin.firstName).toBe(memberProps.firstName);
|
||||
expect(storedAdmin.lastName).toBe(memberProps.lastName);
|
||||
expect(storedAdmin.password).not.toBe(memberProps.password);
|
||||
});
|
||||
|
||||
test('should fail with invalid payloads', async () => {
|
||||
const memberShell = await userRepository.save({
|
||||
email: randomEmail(),
|
||||
role: 'global:member',
|
||||
});
|
||||
|
||||
const invalidPaylods = [
|
||||
{
|
||||
firstName: randomName(),
|
||||
lastName: randomName(),
|
||||
password: randomValidPassword(),
|
||||
},
|
||||
{
|
||||
inviterId: instanceOwner.id,
|
||||
firstName: randomName(),
|
||||
password: randomValidPassword(),
|
||||
},
|
||||
{
|
||||
inviterId: instanceOwner.id,
|
||||
firstName: randomName(),
|
||||
password: randomValidPassword(),
|
||||
},
|
||||
{
|
||||
inviterId: instanceOwner.id,
|
||||
firstName: randomName(),
|
||||
lastName: randomName(),
|
||||
},
|
||||
{
|
||||
inviterId: instanceOwner.id,
|
||||
firstName: randomName(),
|
||||
lastName: randomName(),
|
||||
password: randomInvalidPassword(),
|
||||
},
|
||||
];
|
||||
|
||||
for (const payload of invalidPaylods) {
|
||||
await testServer.authlessAgent
|
||||
.post(`/invitations/${memberShell.id}/accept`)
|
||||
.send(payload)
|
||||
.expect(400);
|
||||
|
||||
const storedMemberShell = await userRepository.findOneByOrFail({
|
||||
email: memberShell.email,
|
||||
});
|
||||
|
||||
expect(storedMemberShell.firstName).toBeNull();
|
||||
expect(storedMemberShell.lastName).toBeNull();
|
||||
expect(storedMemberShell.password).toBeNull();
|
||||
}
|
||||
});
|
||||
|
||||
test('should fail with already accepted invite', async () => {
|
||||
const member = await createMember();
|
||||
|
||||
const memberProps = {
|
||||
inviterId: instanceOwner.id,
|
||||
firstName: randomName(),
|
||||
lastName: randomName(),
|
||||
password: randomValidPassword(),
|
||||
};
|
||||
|
||||
await testServer.authlessAgent
|
||||
.post(`/invitations/${member.id}/accept`)
|
||||
.send(memberProps)
|
||||
.expect(400);
|
||||
|
||||
const storedMember = await userRepository.findOneByOrFail({
|
||||
email: member.email,
|
||||
});
|
||||
|
||||
expect(storedMember.firstName).not.toBe(memberProps.firstName);
|
||||
expect(storedMember.lastName).not.toBe(memberProps.lastName);
|
||||
expect(storedMember.password).not.toBe(memberProps.password);
|
||||
|
||||
const comparisonResult = await Container.get(PasswordUtility).compare(
|
||||
member.password,
|
||||
storedMember.password,
|
||||
);
|
||||
|
||||
expect(comparisonResult).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
describe('POST /invitations', () => {
|
||||
type InvitationResponse = { body: { data: UserInvitationResult[] } };
|
||||
|
||||
test('should fail with invalid payloads', async () => {
|
||||
const invalidPayloads = [
|
||||
randomEmail(),
|
||||
[randomEmail()],
|
||||
{},
|
||||
[{ name: randomName() }],
|
||||
[{ email: randomName() }],
|
||||
];
|
||||
|
||||
for (const invalidPayload of invalidPayloads) {
|
||||
await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send(invalidPayload)
|
||||
.expect(400);
|
||||
|
||||
await expect(userRepository.count()).resolves.toBe(2); // DB unaffected
|
||||
}
|
||||
});
|
||||
|
||||
test('should return 200 on empty payload', async () => {
|
||||
const response = await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([])
|
||||
.expect(200);
|
||||
|
||||
expect(response.body.data).toStrictEqual([]);
|
||||
|
||||
await expect(userRepository.count()).resolves.toBe(2); // DB unaffected
|
||||
});
|
||||
|
||||
test('should return 200 if emailing is not set up', async () => {
|
||||
mailer.invite.mockResolvedValue({ emailSent: false });
|
||||
|
||||
const response = await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([{ email: randomEmail() }]);
|
||||
|
||||
expect(response.body.data).toBeInstanceOf(Array);
|
||||
expect(response.body.data.length).toBe(1);
|
||||
|
||||
const { user } = response.body.data[0];
|
||||
|
||||
expect(user.inviteAcceptUrl).toBeDefined();
|
||||
|
||||
const inviteUrl = new URL(user.inviteAcceptUrl);
|
||||
|
||||
expect(inviteUrl.searchParams.get('inviterId')).toBe(instanceOwner.id);
|
||||
expect(inviteUrl.searchParams.get('inviteeId')).toBe(user.id);
|
||||
});
|
||||
|
||||
test('should create member shell', async () => {
|
||||
mailer.invite.mockResolvedValue({ emailSent: false });
|
||||
|
||||
const response: InvitationResponse = await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([{ email: randomEmail() }])
|
||||
.expect(200);
|
||||
|
||||
const [result] = response.body.data;
|
||||
|
||||
const storedUser = await userRepository.findOneByOrFail({
|
||||
id: result.user.id,
|
||||
});
|
||||
|
||||
assertStoredUserProps(storedUser);
|
||||
});
|
||||
|
||||
test('should create admin shell when advanced permissions is licensed', async () => {
|
||||
testServer.license.enable('feat:advancedPermissions');
|
||||
|
||||
mailer.invite.mockResolvedValue({ emailSent: false });
|
||||
|
||||
const response: InvitationResponse = await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([{ email: randomEmail(), role: 'global:admin' }])
|
||||
.expect(200);
|
||||
|
||||
const [result] = response.body.data;
|
||||
|
||||
const storedUser = await userRepository.findOneByOrFail({
|
||||
id: result.user.id,
|
||||
});
|
||||
|
||||
assertStoredUserProps(storedUser);
|
||||
});
|
||||
|
||||
test('should reinvite member when sharing is licensed', async () => {
|
||||
testServer.license.enable('feat:sharing');
|
||||
|
||||
mailer.invite.mockResolvedValue({ emailSent: false });
|
||||
|
||||
await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([{ email: randomEmail(), role: 'global:member' }]);
|
||||
|
||||
await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([{ email: randomEmail(), role: 'global:member' }])
|
||||
.expect(200);
|
||||
});
|
||||
|
||||
test('should reinvite admin when advanced permissions is licensed', async () => {
|
||||
testServer.license.enable('feat:advancedPermissions');
|
||||
|
||||
mailer.invite.mockResolvedValue({ emailSent: false });
|
||||
|
||||
await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([{ email: randomEmail(), role: 'global:admin' }]);
|
||||
|
||||
await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([{ email: randomEmail(), role: 'global:admin' }])
|
||||
.expect(200);
|
||||
});
|
||||
|
||||
test('should return 403 on creating admin shell when advanced permissions is unlicensed', async () => {
|
||||
testServer.license.disable('feat:advancedPermissions');
|
||||
|
||||
mailer.invite.mockResolvedValue({ emailSent: false });
|
||||
|
||||
await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([{ email: randomEmail(), role: 'global:admin' }])
|
||||
.expect(403);
|
||||
});
|
||||
|
||||
test('should email invites and create user shells, without inviting existing users', async () => {
|
||||
mailer.invite.mockResolvedValue({ emailSent: true });
|
||||
|
||||
const member = await createMember();
|
||||
const memberShell = await createUserShell('global:member');
|
||||
const newUserEmail = randomEmail();
|
||||
|
||||
const existingUserEmails = [member.email];
|
||||
const inviteeUserEmails = [memberShell.email, newUserEmail];
|
||||
const payload = inviteeUserEmails.concat(existingUserEmails).map((email) => ({ email }));
|
||||
|
||||
const response: InvitationResponse = await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send(payload)
|
||||
.expect(200);
|
||||
|
||||
// invite results
|
||||
|
||||
const { data: results } = response.body;
|
||||
|
||||
for (const result of results) {
|
||||
assertUserInviteResult(result);
|
||||
|
||||
const storedUser = await Container.get(UserRepository).findOneByOrFail({
|
||||
id: result.user.id,
|
||||
});
|
||||
|
||||
assertStoredUserProps(storedUser);
|
||||
}
|
||||
|
||||
// external hooks
|
||||
|
||||
expect(externalHooks.run).toHaveBeenCalledTimes(1);
|
||||
|
||||
const [externalHookName, externalHookArg] = externalHooks.run.mock.calls[0];
|
||||
|
||||
expect(externalHookName).toBe('user.invited');
|
||||
expect(externalHookArg?.[0]).toStrictEqual([newUserEmail]);
|
||||
|
||||
// internal hooks
|
||||
|
||||
const calls = mocked(internalHooks).onUserTransactionalEmail.mock.calls;
|
||||
|
||||
for (const [onUserTransactionalEmailArg] of calls) {
|
||||
expect(onUserTransactionalEmailArg.user_id).toBeDefined();
|
||||
expect(onUserTransactionalEmailArg.message_type).toBe('New user invite');
|
||||
expect(onUserTransactionalEmailArg.public_api).toBe(false);
|
||||
}
|
||||
});
|
||||
|
||||
test('should return 200 and surface error when invite method throws error', async () => {
|
||||
const errorMsg = 'Failed to send email';
|
||||
|
||||
mailer.invite.mockImplementation(async () => {
|
||||
throw new Error(errorMsg);
|
||||
});
|
||||
|
||||
const response: InvitationResponse = await testServer
|
||||
.authAgentFor(instanceOwner)
|
||||
.post('/invitations')
|
||||
.send([{ email: randomEmail() }])
|
||||
.expect(200);
|
||||
|
||||
expect(response.body.data).toBeInstanceOf(Array);
|
||||
expect(response.body.data.length).toBe(1);
|
||||
|
||||
const [result] = response.body.data;
|
||||
|
||||
expect(result.error).toBe(errorMsg);
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user