runnerLifecycle.test.ts

import { describe, it, expect, vi, beforeEach } from 'vitest';
import { createRunnerLifecycle } from './runnerLifecycle';
import type { RunnerLifecycle } from './runnerLifecycle';

// Mock heavy deps
vi.mock('@/ui/logger', () => ({
    logger: {
        debug: vi.fn(),
        getLogPath: vi.fn(() => '/tmp/test.log'),
    },
}));

vi.mock('@/ui/terminalState', () => ({
    restoreTerminalState: vi.fn(),
}));

function createMockApiSession() {
    return {
        updateMetadata: vi.fn(),
        sendSessionDeath: vi.fn(),
        flush: vi.fn(),
        close: vi.fn(),
    } as unknown as Parameters<typeof createRunnerLifecycle>[0]['session'];
}

describe('createRunnerLifecycle', () => {
    let lifecycle: RunnerLifecycle;

    beforeEach(() => {
        vi.clearAllMocks();
        lifecycle = createRunnerLifecycle({
            session: createMockApiSession(),
            logTag: 'test',
        });
    });

    // --- D-9: hasExplicitSessionEndReason ---

    describe('hasExplicitSessionEndReason', () => {
        it('returns false initially', () => {
            expect(lifecycle.hasExplicitSessionEndReason()).toBe(false);
        });

        it('returns true after setSessionEndReason is called', () => {
            lifecycle.setSessionEndReason('completed');
            expect(lifecycle.hasExplicitSessionEndReason()).toBe(true);
        });

        it('returns false after markCrash — markCrash does NOT set explicit flag', () => {
            lifecycle.markCrash(new Error('boom'));
            expect(lifecycle.hasExplicitSessionEndReason()).toBe(false);
        });

        it('stays true once set — subsequent markCrash does not clear it', () => {
            lifecycle.setSessionEndReason('handoff');
            lifecycle.markCrash(new Error('late crash'));
            expect(lifecycle.hasExplicitSessionEndReason()).toBe(true);
        });
    });

    // --- markCrash sets reason to 'error' but not explicit ---

    describe('markCrash', () => {
        it('sets sessionEndReason to error via sendSessionDeath during cleanup', async () => {
            const session = createMockApiSession();
            const lc = createRunnerLifecycle({ session, logTag: 'test' });
            lc.markCrash(new Error('fatal'));

            // cleanup triggers sendSessionDeath — verify 'error' reason
            await lc.cleanup();
            expect(session.sendSessionDeath).toHaveBeenCalledWith('error');
        });
    });

    // --- setSessionEndReason + cleanup propagates correct reason ---

    describe('setSessionEndReason + cleanup', () => {
        it('sends explicit reason via sendSessionDeath during cleanup', async () => {
            const session = createMockApiSession();
            const lc = createRunnerLifecycle({ session, logTag: 'test' });
            lc.setSessionEndReason('completed');

            await lc.cleanup();
            expect(session.sendSessionDeath).toHaveBeenCalledWith('completed');
        });
    });
});