/** * @license * Copyright 2025 Google LLC * SPDX-License-Identifier: Apache-2.0 */ import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'; import { renderHook, act } from '@testing-library/react'; import { usePhraseCycler, WITTY_LOADING_PHRASES, PHRASE_CHANGE_INTERVAL_MS, } from './usePhraseCycler.js'; describe('usePhraseCycler', () => { beforeEach(() => { vi.useFakeTimers(); }); afterEach(() => { vi.restoreAllMocks(); }); it('should initialize with the first witty phrase when not active and not waiting', () => { const { result } = renderHook(() => usePhraseCycler(false, false)); expect(result.current).toBe(WITTY_LOADING_PHRASES[0]); }); it('should show "Waiting for user confirmation..." when isWaiting is true', () => { const { result, rerender } = renderHook( ({ isActive, isWaiting }) => usePhraseCycler(isActive, isWaiting), { initialProps: { isActive: true, isWaiting: false } }, ); rerender({ isActive: true, isWaiting: true }); expect(result.current).toBe('Waiting for user confirmation...'); }); it('should not cycle phrases if isActive is false and not waiting', () => { const { result } = renderHook(() => usePhraseCycler(false, false)); act(() => { vi.advanceTimersByTime(PHRASE_CHANGE_INTERVAL_MS * 2); }); expect(result.current).toBe(WITTY_LOADING_PHRASES[0]); }); it('should cycle through witty phrases when isActive is true and not waiting', () => { const { result } = renderHook(() => usePhraseCycler(true, false)); // Initial phrase should be one of the witty phrases expect(WITTY_LOADING_PHRASES).toContain(result.current); const _initialPhrase = result.current; act(() => { vi.advanceTimersByTime(PHRASE_CHANGE_INTERVAL_MS); }); // Phrase should change and be one of the witty phrases expect(WITTY_LOADING_PHRASES).toContain(result.current); const _secondPhrase = result.current; act(() => { vi.advanceTimersByTime(PHRASE_CHANGE_INTERVAL_MS); }); expect(WITTY_LOADING_PHRASES).toContain(result.current); }); it('should reset to a witty phrase when isActive becomes true after being false (and not waiting)', () => { // Ensure there are at least two phrases for this test to be meaningful. if (WITTY_LOADING_PHRASES.length < 2) { return; } // Mock Math.random to make the test deterministic. let callCount = 0; vi.spyOn(Math, 'random').mockImplementation(() => { // Cycle through 0, 1, 0, 1, ... const val = callCount % 2; callCount++; return val / WITTY_LOADING_PHRASES.length; }); const { result, rerender } = renderHook( ({ isActive, isWaiting }) => usePhraseCycler(isActive, isWaiting), { initialProps: { isActive: false, isWaiting: false } }, ); // Activate rerender({ isActive: true, isWaiting: false }); const firstActivePhrase = result.current; expect(WITTY_LOADING_PHRASES).toContain(firstActivePhrase); // With our mock, this should be the first phrase. expect(firstActivePhrase).toBe(WITTY_LOADING_PHRASES[0]); act(() => { vi.advanceTimersByTime(PHRASE_CHANGE_INTERVAL_MS); }); // Phrase should change to the second phrase. expect(result.current).not.toBe(firstActivePhrase); expect(result.current).toBe(WITTY_LOADING_PHRASES[1]); // Set to inactive - should reset to the default initial phrase rerender({ isActive: false, isWaiting: false }); expect(result.current).toBe(WITTY_LOADING_PHRASES[0]); // Set back to active - should pick a random witty phrase (which our mock controls) act(() => { rerender({ isActive: true, isWaiting: false }); }); // The random mock will now return 0, so it should be the first phrase again. expect(result.current).toBe(WITTY_LOADING_PHRASES[0]); }); it('should clear phrase interval on unmount when active', () => { const { unmount } = renderHook(() => usePhraseCycler(true, false)); const clearIntervalSpy = vi.spyOn(global, 'clearInterval'); unmount(); expect(clearIntervalSpy).toHaveBeenCalledOnce(); }); it('should reset to a witty phrase when transitioning from waiting to active', () => { const { result, rerender } = renderHook( ({ isActive, isWaiting }) => usePhraseCycler(isActive, isWaiting), { initialProps: { isActive: true, isWaiting: false } }, ); const _initialPhrase = result.current; expect(WITTY_LOADING_PHRASES).toContain(_initialPhrase); // Cycle to a different phrase (potentially) act(() => { vi.advanceTimersByTime(PHRASE_CHANGE_INTERVAL_MS); }); if (WITTY_LOADING_PHRASES.length > 1) { // This check is probabilistic with random selection } expect(WITTY_LOADING_PHRASES).toContain(result.current); // Go to waiting state rerender({ isActive: false, isWaiting: true }); expect(result.current).toBe('Waiting for user confirmation...'); // Go back to active cycling - should pick a random witty phrase rerender({ isActive: true, isWaiting: false }); expect(WITTY_LOADING_PHRASES).toContain(result.current); }); });