upstream-rocketchat/apps/meteor/client/lib/videoRecorder.spec.ts

import { VideoRecorder } from './videoRecorder';
import { createDeferredPromise } from '../../tests/mocks/utils/createDeferredMockFn';

describe('VideoRecorder', () => {
	let mockStream: MediaStream;
	let mockVideoTrack: MediaStreamTrack;
	let mockAudioTrack: MediaStreamTrack;
	let mockVideoElement: HTMLVideoElement;
	let getUserMediaMock: jest.Mock;

	const createMockStream = (videoTrack?: MediaStreamTrack, audioTrack?: MediaStreamTrack): MediaStream => {
		return {
			getVideoTracks: jest.fn(() => [videoTrack || ({ stop: jest.fn() } as unknown as MediaStreamTrack)]),
			getAudioTracks: jest.fn(() => [audioTrack || ({ stop: jest.fn() } as unknown as MediaStreamTrack)]),
		} as unknown as MediaStream;
	};

	beforeEach(() => {
		jest.useFakeTimers();

		mockVideoTrack = {
			stop: jest.fn(),
		} as unknown as MediaStreamTrack;

		mockAudioTrack = {
			stop: jest.fn(),
		} as unknown as MediaStreamTrack;

		mockStream = createMockStream(mockVideoTrack, mockAudioTrack);

		mockVideoElement = document.createElement('video');
		mockVideoElement.load = jest.fn();
		mockVideoElement.play = jest.fn().mockResolvedValue(undefined);
		mockVideoElement.pause = jest.fn();

		getUserMediaMock = jest.fn();

		Object.defineProperty(global.navigator, 'mediaDevices', {
			writable: true,
			value: {
				getUserMedia: getUserMediaMock,
			},
		});

		global.MediaRecorder = {
			isTypeSupported: jest.fn((type: string) => type === 'video/webm'),
		} as any;
	});

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

	describe('Asynchronous start and stop handling', () => {
		it('should stop camera tracks when stop is called before getUserMedia resolves', async () => {
			const streamDeferred = createDeferredPromise<MediaStream>();

			getUserMediaMock.mockReturnValue(streamDeferred.promise);

			const callback = jest.fn();
			VideoRecorder.start(mockVideoElement, callback);
			VideoRecorder.stop();

			streamDeferred.resolve(mockStream);
			await jest.runAllTimersAsync();

			expect(mockVideoTrack.stop).toHaveBeenCalled();
			expect(mockAudioTrack.stop).toHaveBeenCalled();
			expect(callback).not.toHaveBeenCalledWith(true);
		});

		it('should not initialize camera when stopped early', async () => {
			const streamDeferred = createDeferredPromise<MediaStream>();

			getUserMediaMock.mockReturnValue(streamDeferred.promise);

			VideoRecorder.start(mockVideoElement, jest.fn());
			VideoRecorder.stop();

			streamDeferred.resolve(mockStream);
			await jest.runAllTimersAsync();

			expect(VideoRecorder.cameraStarted).toBe(false);
		});

		it('should handle multiple start/stop cycles', async () => {
			const stream1 = createMockStream();
			const stream2 = createMockStream(mockVideoTrack, mockAudioTrack);

			getUserMediaMock.mockReturnValueOnce(Promise.resolve(stream1));

			VideoRecorder.start(mockVideoElement, jest.fn());
			VideoRecorder.stop();

			const stream2Deferred = createDeferredPromise<MediaStream>();
			getUserMediaMock.mockReturnValueOnce(stream2Deferred.promise);

			const cb = jest.fn();
			VideoRecorder.start(mockVideoElement, cb);

			stream2Deferred.resolve(stream2);
			await jest.runAllTimersAsync();

			expect(cb).toHaveBeenCalledWith(true);
			expect(VideoRecorder.cameraStarted).toBe(true);
		});

		it('should invalidate pending callbacks from previous start when new start is called', async () => {
			const firstStream = createMockStream();
			const secondStream = createMockStream(mockVideoTrack, mockAudioTrack);

			const firstDeferred = createDeferredPromise<MediaStream>();
			const secondDeferred = createDeferredPromise<MediaStream>();

			getUserMediaMock.mockReturnValueOnce(firstDeferred.promise).mockReturnValueOnce(secondDeferred.promise);

			const cb1 = jest.fn();
			const cb2 = jest.fn();

			VideoRecorder.start(mockVideoElement, cb1);
			VideoRecorder.start(mockVideoElement, cb2);

			secondDeferred.resolve(secondStream);
			await jest.runAllTimersAsync();
			firstDeferred.resolve(firstStream);
			await jest.runAllTimersAsync();

			expect(firstStream.getVideoTracks).toHaveBeenCalled();
			expect(firstStream.getAudioTracks).toHaveBeenCalled();
			expect(cb2).toHaveBeenCalledWith(true);
			expect(cb1).not.toHaveBeenCalledWith(true);
		});
	});

	describe('Normal operation', () => {
		it('should initialize camera', async () => {
			getUserMediaMock.mockResolvedValue(mockStream);

			const cb = jest.fn();
			VideoRecorder.start(mockVideoElement, cb);

			await jest.runAllTimersAsync();

			expect(cb).toHaveBeenCalledWith(true);
			expect(VideoRecorder.cameraStarted).toBe(true);
		});

		it('should stop camera tracks', () => {
			(VideoRecorder as any).stream = mockStream;
			(VideoRecorder as any).started = true;
			(VideoRecorder as any)._cameraStarted = true;

			VideoRecorder.stop();

			expect(mockVideoTrack.stop).toHaveBeenCalled();
			expect(mockAudioTrack.stop).toHaveBeenCalled();
			expect(VideoRecorder.cameraStarted).toBe(false);
		});

		it('should return supported mime types', () => {
			expect(VideoRecorder.getSupportedMimeTypes()).toBe('video/webm; codecs=vp8,opus');
		});

		it('should handle permission errors', async () => {
			getUserMediaMock.mockRejectedValue(new Error('Permission denied'));

			const cb = jest.fn();
			VideoRecorder.start(mockVideoElement, cb);

			await jest.runAllTimersAsync();

			expect(cb).toHaveBeenCalledWith(false);
		});
	});
});