// tests/main/services/VoiceModeService.test.ts // 상태 머신 전이 + 이중 조건 플러시 + accidentalPress 테스트 import { describe, it, expect, beforeEach, vi } from 'vitest' import { EventEmitter } from 'events' import { RecognitionState, AudioState } from '@d3ro/core/types' import { TIMING } from '@d3ro/core/constants' // 모든 하위 서비스 모킹 vi.mock('../../../src/main/services/LoggerService', () => ({ getLogger: () => ({ info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() }) })) const mockSTT = { initialize: vi.fn(() => Promise.resolve()), transcribe: vi.fn(() => Promise.resolve({ text: '테스트 전사', segments: [], language: 'ko', duration: 2, processingTime: 500 }) ), getStatus: vi.fn(() => ({ state: 'ready', modelId: 'base', uptime: 0 })), // 프리플라이트 검증(모델 설치 여부)용 — 기본은 설치된 상태로 목킹 getModels: vi.fn(() => [ { id: 'base', name: 'Base', sizeBytes: 0, downloaded: true, languages: [], accuracy: 2, speed: 4 }, ]), on: vi.fn(), off: vi.fn() } vi.mock('../../../src/main/services/LocalSTTService', () => ({ getLocalSTTService: () => mockSTT, resetLocalSTTServiceForTests: () => undefined, })) const audioBus = new EventEmitter() const mockAudio = { start: vi.fn(() => Promise.resolve()), stop: vi.fn(() => Promise.resolve()), on: vi.fn((ev: string, fn: (...args: unknown[]) => void) => { audioBus.on(ev, fn) }), off: vi.fn((ev: string, fn: (...args: unknown[]) => void) => { audioBus.off(ev, fn) }), } vi.mock('../../../src/main/services/AudioCaptureService', () => ({ getAudioCaptureService: () => mockAudio })) const mockHotkey = { on: vi.fn(), off: vi.fn() } vi.mock('../../../src/main/services/HotkeyService', () => ({ getHotkeyService: () => mockHotkey })) vi.mock('../../../src/main/services/ConfigService', () => ({ configGet: vi.fn((key: string) => { const defaults: Record = { sttModelId: 'base', defaultLLMAction: 'refine', ollamaServerUrl: 'http://localhost:11434', llmModelId: 'gemma4:e4b' } return defaults[key] }) })) const mockTextInsert = { insertText: vi.fn(() => Promise.resolve({ success: true, method: 'clipboard', textLength: 10, durationMs: 50 })) } vi.mock('../../../src/main/services/TextInsertService', () => ({ getTextInsertService: () => mockTextInsert })) const mockLLM = { isAvailable: vi.fn(() => false), processText: vi.fn(() => Promise.resolve('다듬어진 텍스트')), on: vi.fn(), off: vi.fn() } vi.mock('../../../src/main/services/LocalLLMService', () => ({ getLocalLLMService: () => mockLLM })) let getVoiceModeService: () => ReturnType beforeEach(async () => { vi.resetModules() vi.clearAllMocks() audioBus.removeAllListeners() mockSTT.initialize.mockResolvedValue(undefined as never) mockSTT.transcribe.mockResolvedValue({ text: '테스트 전사', segments: [], language: 'ko', duration: 2, processingTime: 500, } as never) mockLLM.processText.mockResolvedValue('다듬어진 텍스트' as never) const mod = await import('../../../src/main/services/VoiceModeService') mod.resetVoiceModeServiceForTests() getVoiceModeService = mod.getVoiceModeService }) describe('VoiceModeService', () => { describe('상태 머신', () => { it('초기 상태는 IDLE이다', () => { const svc = getVoiceModeService() const state = svc.getState() expect(state.recognitionState).toBe(RecognitionState.IDLE) expect(state.audioState).toBe(AudioState.IDLE) expect(state.sessionId).toBeNull() }) it('startSession 호출 시 PREPARING으로 전이한다', async () => { const svc = getVoiceModeService() const stateChanges: RecognitionState[] = [] svc.on('recognition-state-changed', (payload: { current: RecognitionState }) => { stateChanges.push(payload.current) }) await svc.startSession('dictation') // PREPARING → CONNECTING → READY 순서 expect(stateChanges[0]).toBe(RecognitionState.PREPARING) expect(stateChanges).toContain(RecognitionState.CONNECTING) }) it('isActive는 세션이 활성일 때 true이다', async () => { const svc = getVoiceModeService() expect(svc.isActive).toBe(false) // startSession은 완전 비동기이므로 await 후 세션 활성 확인 await svc.startSession('dictation') expect(svc.isActive).toBe(true) }) }) describe('accidentalPress', () => { it('700ms 미만 세션은 자동 취소된다', async () => { const svc = getVoiceModeService() let cancelReason: string | null = null svc.on('session-cancelled', (payload: { reason: string }) => { cancelReason = payload.reason }) // 세션 시작 즉시 종료 (700ms 미만) await svc.startSession('dictation') await svc.stopSession() expect(cancelReason).toBe('too-short') }) }) describe('STT 준비 대기', () => { it('전사가 시작된 뒤에는 readiness timeout 으로 세션을 취소하지 않는다', async () => { let resolveInit: (() => void) | undefined mockSTT.initialize.mockImplementation( () => new Promise((resolve) => { resolveInit = resolve }), ) mockSTT.transcribe.mockResolvedValue({ text: '안녕하세요', segments: [], language: 'ko', duration: 2, processingTime: 50, } as never) const svc = getVoiceModeService() let cancelReason: string | null = null let completed = false svc.on('session-cancelled', (payload: { reason: string }) => { cancelReason = payload.reason }) const completedPromise = new Promise((resolve) => { svc.once('session-completed', () => { completed = true resolve() }) }) await svc.startSession('dictation') audioBus.emit('audio-data', { buffer: Buffer.alloc(16000 * 2) }) await new Promise((r) => setTimeout(r, 850)) await svc.stopSession() resolveInit?.() await Promise.race([completedPromise, new Promise((r) => setTimeout(r, 1000))]) expect(cancelReason).not.toBe('timeout') expect(completed).toBe(true) }) }) describe('cancelSession', () => { it('user 취소로 세션을 종료한다', async () => { const svc = getVoiceModeService() let cancelReason: string | null = null svc.on('session-cancelled', (payload: { reason: string }) => { cancelReason = payload.reason }) await svc.startSession('dictation') svc.cancelSession() expect(cancelReason).toBe('user') }) }) describe('getState', () => { it('현재 상태를 VoiceState 형태로 반환한다', () => { const svc = getVoiceModeService() const state = svc.getState() expect(state).toHaveProperty('recognitionState') expect(state).toHaveProperty('audioState') expect(state).toHaveProperty('mode') expect(state).toHaveProperty('sessionId') expect(state).toHaveProperty('recordingStartedAt') }) }) describe('터미널 상태', () => { it('cancelSession 후 _resetToIdle의 200ms 딜레이 후 IDLE로 전이한다', async () => { vi.useFakeTimers() const svc = getVoiceModeService() await svc.startSession('dictation') svc.cancelSession() // 200ms 딜레이로 IDLE 전이 예약됨 vi.advanceTimersByTime(250) const state = svc.getState() expect(state.recognitionState).toBe(RecognitionState.IDLE) vi.useRealTimers() }) }) }) describe('TIMING 상수', () => { it('핵심 타이밍 값이 Speakly 패턴과 일치한다', () => { expect(TIMING.MIN_AUDIO_DURATION).toBe(700) expect(TIMING.DOUBLE_PRESS_DURATION).toBe(300) expect(TIMING.POST_RECORDING_WAIT).toBe(4000) expect(TIMING.POST_RECORDING_WAIT_BUFFERED).toBe(6000) expect(TIMING.ABSOLUTE_MAX_WAIT).toBe(120000) expect(TIMING.AUDIO_LEVEL_INTERVAL).toBe(100) }) })