|
| 1 | +import { cleanup, render } from 'ink-testing-library'; |
| 2 | +import { EventEmitter } from 'node:events'; |
| 3 | +import React from 'react'; |
| 4 | +import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'; |
| 5 | + |
| 6 | +import { RenderStatus, store } from '../store/index.js'; |
| 7 | +import { DefaultComponent } from './default-component.js'; |
| 8 | + |
| 9 | +// Mock dependent components |
| 10 | +vi.mock('./progress/progress-display', () => ({ |
| 11 | + ProgressDisplay: () => <div>Mock Progress Display</div> |
| 12 | +})); |
| 13 | +vi.mock('./import/index', () => ({ |
| 14 | + ImportParametersForm: () => <div>Mock Import Parameters Form</div> |
| 15 | +})); |
| 16 | +vi.mock('./plan/plan', () => ({ |
| 17 | + PlanComponent: () => <div>Mock Plan Component</div> |
| 18 | +})); |
| 19 | +vi.mock('./import/import-result', () => ({ |
| 20 | + ImportResultComponent: () => <div>Mock Import Result Component</div> |
| 21 | +})); |
| 22 | + |
| 23 | +describe('DefaultComponent', () => { |
| 24 | + let emitter: EventEmitter; |
| 25 | + |
| 26 | + beforeEach(() => { |
| 27 | + emitter = new EventEmitter(); |
| 28 | + vi.resetAllMocks(); |
| 29 | + }); |
| 30 | + |
| 31 | + afterEach(() => { |
| 32 | + cleanup(); |
| 33 | + emitter.removeAllListeners(); |
| 34 | + }); |
| 35 | + |
| 36 | + it('renders progress display when renderStatus is PROGRESS', () => { |
| 37 | + store.set(store.renderState, { status: RenderStatus.PROGRESS }); |
| 38 | + const { lastFrame } = render(<DefaultComponent emitter={emitter} />); |
| 39 | + |
| 40 | + expect(lastFrame()).toContain('Mock Progress Display'); |
| 41 | + }); |
| 42 | + |
| 43 | + it('renders the plan when renderStatus is DISPLAY_PLAN', () => { |
| 44 | + store.set(store.renderState, { status: RenderStatus.DISPLAY_PLAN, data: {} }); |
| 45 | + const { lastFrame } = render(<DefaultComponent emitter={emitter} />); |
| 46 | + |
| 47 | + expect(lastFrame()).toContain('Mock Plan Component'); |
| 48 | + }); |
| 49 | + |
| 50 | + it('handles SUDO_PROMPT event and submits password', () => { |
| 51 | + store.set(store.renderState, { status: RenderStatus.SUDO_PROMPT, data: 'message' }); |
| 52 | + const { lastFrame } = render(<DefaultComponent emitter={emitter} />); |
| 53 | + |
| 54 | + expect(lastFrame()).toContain('Password:'); |
| 55 | + }); |
| 56 | + |
| 57 | + it('renders import parameter form when renderStatus is IMPORT_PROMPT', () => { |
| 58 | + store.set(store.renderState, { status: RenderStatus.IMPORT_PROMPT, data: new Map() }); |
| 59 | + const { lastFrame } = render(<DefaultComponent emitter={emitter} />); |
| 60 | + |
| 61 | + expect(lastFrame()).toContain('Mock Import Parameters Form'); |
| 62 | + }); |
| 63 | +}); |
0 commit comments