upstream-rocketchat/packages/ui-voip/src/context/usePeekMediaSessionPeerInfo.spec.tsx

import { Emitter } from '@rocket.chat/emitter';
import type { CallContact } from '@rocket.chat/media-signaling';
import { renderHook, act } from '@testing-library/react';
import type { ReactNode } from 'react';

import { MediaCallInstanceContext, defaultContextValue } from './MediaCallInstanceContext';
import { usePeekMediaSessionPeerInfo } from './usePeekMediaSessionPeerInfo';

type MockInstance = {
	getState: () => { confirmed: true; remoteParticipant: { contact: CallContact } } | null;
	on: (event: 'sessionStateChange', onStoreChange: () => void) => () => void;
};

const createWrapper = (instance: MockInstance | undefined) => {
	const wrapper = ({ children }: { children?: ReactNode }) => (
		<MediaCallInstanceContext.Provider
			value={{
				...defaultContextValue,
				instance: instance as any,
			}}
		>
			{children}
		</MediaCallInstanceContext.Provider>
	);
	return wrapper;
};

describe('usePeekMediaSessionPeerInfo', () => {
	it('returns undefined when instance is undefined', () => {
		const { result } = renderHook(() => usePeekMediaSessionPeerInfo(), {
			wrapper: createWrapper(undefined),
		});

		expect(result.current).toBeUndefined();
	});

	it('returns undefined when instance has no main call', () => {
		const instance: MockInstance = {
			getState: () => null,
			on: () => () => undefined,
		};

		const { result } = renderHook(() => usePeekMediaSessionPeerInfo(), {
			wrapper: createWrapper(instance),
		});

		expect(result.current).toBeUndefined();
	});

	describe('when main call has a contact', () => {
		it('returns external peer info for SIP contact', () => {
			const instance: MockInstance = {
				getState: () => ({
					confirmed: true,
					remoteParticipant: {
						contact: {
							type: 'sip',
							id: '+5511999999999',
						},
					},
				}),
				on: () => () => undefined,
			};

			const { result } = renderHook(() => usePeekMediaSessionPeerInfo(), {
				wrapper: createWrapper(instance),
			});

			expect(result.current).toEqual({ number: '+5511999999999' });
		});

		it('returns internal peer info for user contact', () => {
			const instance: MockInstance = {
				getState: () => ({
					confirmed: true,
					remoteParticipant: {
						contact: {
							type: 'user',
							id: 'userId123',
							displayName: 'John Doe',
							username: 'johndoe',
							sipExtension: '1001',
						},
					},
				}),
				on: () => () => undefined,
			};

			const { result } = renderHook(() => usePeekMediaSessionPeerInfo(), {
				wrapper: createWrapper(instance),
			});

			expect(result.current).toEqual({
				displayName: 'John Doe',
				userId: 'userId123',
				username: 'johndoe',
				callerId: '1001',
			});
		});
	});

	describe('sessionStateChange subscription', () => {
		it('updates peer info when sessionStateChange is emitted', () => {
			const emitter = new Emitter<{ sessionStateChange: void }>();

			const defaultInstanceState = {
				confirmed: true as const,
				remoteParticipant: {
					contact: {
						type: 'sip' as const,
						id: '+5511999999999',
					} as CallContact,
				},
			};

			let instanceState: typeof defaultInstanceState | null = defaultInstanceState;

			const instance: MockInstance = {
				getState: () => instanceState,
				on: (event, onStoreChange) => emitter.on(event, onStoreChange),
			};

			const { result } = renderHook(() => usePeekMediaSessionPeerInfo(), {
				wrapper: createWrapper(instance),
			});

			expect(result.current).toEqual({ number: '+5511999999999' });

			act(() => {
				instanceState = null;
				emitter.emit('sessionStateChange');
			});

			expect(result.current).toBeUndefined();

			act(() => {
				instanceState = {
					confirmed: true,
					remoteParticipant: {
						contact: {
							type: 'user',
							id: 'userId456',
							displayName: 'Jane Smith',
						},
					},
				};
				emitter.emit('sessionStateChange');
			});

			expect(result.current).toEqual({
				displayName: 'Jane Smith',
				userId: 'userId456',
				username: undefined,
				callerId: undefined,
			});
		});
	});
});