upstream-rocketchat/apps/meteor/client/navbar/NavBarSettingsToolbar/UserMenu/hooks/useUserMenu.spec.ts

import { UIActionButtonContext } from '@rocket.chat/apps-engine/definition/ui';
import { mockAppRoot } from '@rocket.chat/mock-providers';
import { renderHook, waitFor } from '@testing-library/react';

import { useUserMenu } from './useUserMenu';

// Mock userStatuses to avoid sdk.stream to call Meteor.connection.subscribe
jest.mock('../../../../lib/userStatuses', () => ({
	userStatuses: {
		invisibleAllowed: true,
		watch: jest.fn(() => () => undefined),
		sync: jest.fn(() => Promise.resolve()),
	},
}));

const fakeUser = {
	_id: 'john.doe',
	username: 'john.doe',
	name: 'John Doe',
	createdAt: new Date(),
	active: true,
	_updatedAt: new Date(),
	roles: ['admin'],
	type: 'user' as const,
};

it('should return one action from the server with no conditions', async () => {
	const { result } = renderHook(() => useUserMenu(fakeUser), {
		wrapper: mockAppRoot()
			.withEndpoint('GET', '/apps/actionButtons', () => [
				{
					appId: 'APP_ID',
					actionId: 'ACTION_ID',
					labelI18n: 'LABEL_I18N',
					context: UIActionButtonContext.USER_DROPDOWN_ACTION,
				},
			])
			.build(),
	});

	await waitFor(() => {
		const appsGroup = result.current.find((group) => group.title === 'Apps');
		expect(appsGroup).toBeDefined();
		expect(appsGroup?.items).toContainEqual(expect.objectContaining({ id: 'APP_ID_ACTION_ID' }));
	});
});

describe('user menu with role conditions', () => {
	it('should return the action if the user has admin role', async () => {
		const { result } = renderHook(() => useUserMenu(fakeUser), {
			wrapper: mockAppRoot()
				.withJohnDoe()
				.withEndpoint('GET', '/apps/actionButtons', () => [
					{
						appId: 'APP_ID',
						actionId: 'ACTION_ID',
						labelI18n: 'LABEL_I18N',
						context: UIActionButtonContext.USER_DROPDOWN_ACTION,
						when: {
							hasOneRole: ['admin'],
						},
					},
				])
				.withRole('admin')
				.build(),
		});

		await waitFor(() => {
			const appsGroup = result.current.find((group) => group.title === 'Apps');
			expect(appsGroup).toBeDefined();
			expect(appsGroup?.items).toContainEqual(expect.objectContaining({ id: 'APP_ID_ACTION_ID' }));
		});
	});

	it('should filter out the action if the user does not have admin role', async () => {
		const { result } = renderHook(() => useUserMenu(fakeUser), {
			wrapper: mockAppRoot()
				.withEndpoint('GET', '/apps/actionButtons', () => [
					{
						appId: 'APP_ID',
						actionId: 'ACTION_ID',
						labelI18n: 'LABEL_I18N',
						context: UIActionButtonContext.USER_DROPDOWN_ACTION,
						when: {
							hasOneRole: ['admin'],
						},
					},
				])
				.build(),
		});

		await waitFor(() => {
			const appsGroup = result.current.find((group) => group.title === 'Apps');
			expect(appsGroup).not.toBeDefined();
		});
	});
});

describe('user menu with permission conditions', () => {
	it('should return the action if the user has manage-apps permission', async () => {
		const { result } = renderHook(() => useUserMenu(fakeUser), {
			wrapper: mockAppRoot()
				.withEndpoint('GET', '/apps/actionButtons', () => [
					{
						appId: 'APP_ID',
						actionId: 'ACTION_ID',
						labelI18n: 'LABEL_I18N',
						context: UIActionButtonContext.USER_DROPDOWN_ACTION,
						when: {
							hasOnePermission: ['manage-apps'],
						},
					},
				])
				.withPermission('manage-apps')
				.build(),
		});

		await waitFor(() => {
			const appsGroup = result.current.find((group) => group.title === 'Apps');
			expect(appsGroup).toBeDefined();
			expect(appsGroup?.items).toContainEqual(expect.objectContaining({ id: 'APP_ID_ACTION_ID' }));
		});
	});

	it('should filter out the action if the user does not have any permission', async () => {
		const { result } = renderHook(() => useUserMenu(fakeUser), {
			wrapper: mockAppRoot()
				.withEndpoint('GET', '/apps/actionButtons', () => [
					{
						appId: 'APP_ID',
						actionId: 'ACTION_ID',
						labelI18n: 'LABEL_I18N',
						context: UIActionButtonContext.USER_DROPDOWN_ACTION,
						when: {
							hasOnePermission: ['any'],
						},
					},
				])
				.build(),
		});

		await waitFor(() => {
			const appsGroup = result.current.find((group) => group.title === 'Apps');
			expect(appsGroup).not.toBeDefined();
		});
	});
});