upstream-rocketchat/apps/meteor/client/views/room/hooks/useE2EEResetRoomKey.spec.ts

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

import { useE2EEResetRoomKey } from './useE2EEResetRoomKey';
import { e2e } from '../../../lib/e2ee';

jest.mock('../../../lib/e2ee', () => ({
	e2e: {
		getInstanceByRoomId: jest.fn(),
	},
}));

describe('useE2EEResetRoomKey', () => {
	const e2eResetRoomKeyMock = jest.fn().mockResolvedValue({
		e2eKeyId: 'E2E_KEY_ID',
		e2eKey: 'E2E_KEY',
	});
	const resetRoomKeyMock = jest.fn();
	const roomId = 'ROOM_ID';

	afterEach(() => {
		jest.clearAllMocks();
	});

	beforeEach(() => {
		(e2e.getInstanceByRoomId as jest.Mock).mockImplementation(() => ({
			resetRoomKey: e2eResetRoomKeyMock,
		}));
	});

	it('should call resetRoomKey endpoint with correct params', async () => {
		const { result } = renderHook(() => useE2EEResetRoomKey(), {
			wrapper: mockAppRoot().withEndpoint('POST', '/v1/e2e.resetRoomKey', resetRoomKeyMock).build(),
		});

		await waitFor(() => result.current.mutate({ roomId }));

		expect(e2e.getInstanceByRoomId).toHaveBeenCalledTimes(1);
		expect(e2e.getInstanceByRoomId).toHaveBeenCalledWith('ROOM_ID');
		expect(e2eResetRoomKeyMock).toHaveBeenCalledTimes(1);

		expect(resetRoomKeyMock).toHaveBeenCalledWith({
			rid: roomId,
			e2eKeyId: 'E2E_KEY_ID',
			e2eKey: 'E2E_KEY',
		});

		await waitFor(() => expect(result.current.status).toBe('success'));
	});

	it('should return an errror if e2e.getInstanceByRoomId() does not return correct params', async () => {
		(e2e.getInstanceByRoomId as jest.Mock).mockReturnValue(null);

		const { result } = renderHook(() => useE2EEResetRoomKey(), {
			wrapper: mockAppRoot().withEndpoint('POST', '/v1/e2e.resetRoomKey', resetRoomKeyMock).build(),
		});

		await waitFor(() => result.current.mutate({ roomId }));

		expect(e2e.getInstanceByRoomId).toHaveBeenCalledTimes(1);
		expect(e2e.getInstanceByRoomId).toHaveBeenCalledWith('ROOM_ID');
		expect(e2eResetRoomKeyMock).toHaveBeenCalledTimes(0);

		await waitFor(() => expect(result.current.status).toBe('error'));
	});

	it('should return an errror if e2e.resetRoomKey() does not return correct params', async () => {
		const e2eResetRoomKeyMock = jest.fn().mockResolvedValue(null);
		const roomId = 'ROOM_ID';

		(e2e.getInstanceByRoomId as jest.Mock).mockImplementation(() => ({
			resetRoomKey: e2eResetRoomKeyMock,
		}));

		const { result } = renderHook(() => useE2EEResetRoomKey(), {
			wrapper: mockAppRoot().withEndpoint('POST', '/v1/e2e.resetRoomKey', resetRoomKeyMock).build(),
		});

		await waitFor(() => result.current.mutate({ roomId }));

		expect(e2e.getInstanceByRoomId).toHaveBeenCalledTimes(1);
		expect(e2e.getInstanceByRoomId).toHaveBeenCalledWith('ROOM_ID');
		expect(e2eResetRoomKeyMock).toHaveBeenCalledTimes(1);

		expect(resetRoomKeyMock).toHaveBeenCalledTimes(0);

		await waitFor(() => expect(result.current.status).toBe('error'));
	});

	it('should return an error if resetRoomKey does not resolve', async () => {
		resetRoomKeyMock.mockRejectedValue(new Error('error-e2e-key-reset-in-progress'));
		const { result } = renderHook(() => useE2EEResetRoomKey(), {
			wrapper: mockAppRoot().withEndpoint('POST', '/v1/e2e.resetRoomKey', resetRoomKeyMock).build(),
		});

		await waitFor(() => result.current.mutate({ roomId }));

		expect(e2e.getInstanceByRoomId).toHaveBeenCalledTimes(1);
		expect(e2e.getInstanceByRoomId).toHaveBeenCalledWith('ROOM_ID');
		expect(e2eResetRoomKeyMock).toHaveBeenCalledTimes(1);

		expect(resetRoomKeyMock).toHaveBeenCalledWith({
			rid: roomId,
			e2eKeyId: 'E2E_KEY_ID',
			e2eKey: 'E2E_KEY',
		});

		await waitFor(() => expect(result.current.status).toBe('error'));
	});
});