Spaces:
Sleeping
Sleeping
File size: 4,873 Bytes
79ed62f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 | import { renderHook, act } from '@testing-library/react';
import { describe, it, expect, vi, beforeEach } from 'vitest';
import { useTripWebSocket } from '../../../src/hooks/useTripWebSocket';
import { useTripStore } from '../../../src/store/tripStore';
vi.mock('../../../src/api/websocket', () => ({
connect: vi.fn(),
disconnect: vi.fn(),
getSocketId: vi.fn(() => 'mock-socket-id'),
joinTrip: vi.fn(),
leaveTrip: vi.fn(),
addListener: vi.fn(),
removeListener: vi.fn(),
setRefetchCallback: vi.fn(),
setPreReconnectHook: vi.fn(),
}));
// Import the mocked module AFTER vi.mock
const wsMock = await import('../../../src/api/websocket');
describe('useTripWebSocket', () => {
beforeEach(() => {
vi.clearAllMocks();
});
it('FE-HOOK-WS-001: on mount, joinTrip(tripId) is called', () => {
const { unmount } = renderHook(() => useTripWebSocket(42));
expect(wsMock.joinTrip).toHaveBeenCalledWith(42);
unmount();
});
it('FE-HOOK-WS-002: on mount, addListener is called (registers event handlers)', () => {
const { unmount } = renderHook(() => useTripWebSocket(42));
// addListener is called twice: once for handleRemoteEvent, once for collabFileSync
expect(wsMock.addListener).toHaveBeenCalled();
expect((wsMock.addListener as ReturnType<typeof vi.fn>).mock.calls.length).toBeGreaterThanOrEqual(1);
unmount();
});
it('FE-HOOK-WS-003: on unmount, leaveTrip(tripId) is called', () => {
const { unmount } = renderHook(() => useTripWebSocket(42));
unmount();
expect(wsMock.leaveTrip).toHaveBeenCalledWith(42);
});
it('FE-HOOK-WS-004: on unmount, removeListener is called', () => {
const { unmount } = renderHook(() => useTripWebSocket(42));
unmount();
expect(wsMock.removeListener).toHaveBeenCalled();
});
it('FE-HOOK-WS-005: when tripId changes, leaves old trip and joins new one', () => {
const { rerender, unmount } = renderHook(({ id }) => useTripWebSocket(id), {
initialProps: { id: 1 as number | undefined },
});
expect(wsMock.joinTrip).toHaveBeenCalledWith(1);
rerender({ id: 2 });
expect(wsMock.leaveTrip).toHaveBeenCalledWith(1);
expect(wsMock.joinTrip).toHaveBeenCalledWith(2);
unmount();
});
it('FE-HOOK-WS-006: one of the registered listeners is handleRemoteEvent from tripStore', () => {
const handler = useTripStore.getState().handleRemoteEvent;
renderHook(() => useTripWebSocket(42));
const addListenerCalls = (wsMock.addListener as ReturnType<typeof vi.fn>).mock.calls;
const registeredFunctions = addListenerCalls.map((call) => call[0]);
expect(registeredFunctions).toContain(handler);
});
it('FE-HOOK-WS-006b: collab file sync listener is also registered (second addListener call)', () => {
const { unmount } = renderHook(() => useTripWebSocket(42));
// Two listeners registered: handleRemoteEvent + collabFileSync
expect((wsMock.addListener as ReturnType<typeof vi.fn>).mock.calls.length).toBe(2);
unmount();
});
it('FE-HOOK-WS-006c: collab file sync listener reacts to collab:note:deleted events', () => {
const mockLoadFiles = vi.fn();
useTripStore.setState({ loadFiles: mockLoadFiles } as any);
renderHook(() => useTripWebSocket(42));
// The second addListener call is the collabFileSync function
const addListenerCalls = (wsMock.addListener as ReturnType<typeof vi.fn>).mock.calls;
const collabFileSync = addListenerCalls[1]?.[0];
expect(collabFileSync).toBeTypeOf('function');
act(() => {
collabFileSync({ type: 'collab:note:deleted' });
});
expect(mockLoadFiles).toHaveBeenCalledWith(42);
});
it('FE-HOOK-WS-006d: collab file sync listener reacts to collab:note:updated events', () => {
const mockLoadFiles = vi.fn();
useTripStore.setState({ loadFiles: mockLoadFiles } as any);
renderHook(() => useTripWebSocket(42));
const addListenerCalls = (wsMock.addListener as ReturnType<typeof vi.fn>).mock.calls;
const collabFileSync = addListenerCalls[1]?.[0];
act(() => {
collabFileSync({ type: 'collab:note:updated' });
});
expect(mockLoadFiles).toHaveBeenCalledWith(42);
});
it('FE-HOOK-WS-006e: collab file sync listener ignores unrelated event types', () => {
const mockLoadFiles = vi.fn();
useTripStore.setState({ loadFiles: mockLoadFiles } as any);
renderHook(() => useTripWebSocket(42));
const addListenerCalls = (wsMock.addListener as ReturnType<typeof vi.fn>).mock.calls;
const collabFileSync = addListenerCalls[1]?.[0];
act(() => {
collabFileSync({ type: 'place:created' });
});
expect(mockLoadFiles).not.toHaveBeenCalled();
});
it('FE-HOOK-WS-007: no joinTrip call when tripId is undefined', () => {
renderHook(() => useTripWebSocket(undefined));
expect(wsMock.joinTrip).not.toHaveBeenCalled();
});
});
|