from __future__ import annotations

import signal
from types import SimpleNamespace
from unittest.mock import Mock

import pytest

from langbot.pkg.core import boot


@pytest.mark.asyncio
async def test_main_signal_handler_handles_sigint_before_app_created(monkeypatch):
    captured_handler = {}

    def fake_signal(sig, handler):
        captured_handler[sig] = handler

    async def fake_make_app(loop):
        captured_handler[signal.SIGINT](signal.SIGINT, None)

    monkeypatch.setattr(signal, 'signal', fake_signal)
    monkeypatch.setattr(boot, 'make_app', fake_make_app)

    await boot.main(SimpleNamespace())


@pytest.mark.asyncio
async def test_main_signal_handler_disposes_created_app(monkeypatch):
    captured_handler = {}
    app_inst = SimpleNamespace(shutdown_called=False)

    def fake_signal(sig, handler):
        captured_handler[sig] = handler

    async def shutdown():
        app_inst.shutdown_called = True

    async def run():
        captured_handler[signal.SIGINT](signal.SIGINT, None)

    async def fake_make_app(loop):
        app_inst.shutdown = shutdown
        app_inst.run = run
        return app_inst

    monkeypatch.setattr(signal, 'signal', fake_signal)
    monkeypatch.setattr(boot, 'make_app', fake_make_app)

    await boot.main(SimpleNamespace())

    assert app_inst.shutdown_called is True


@pytest.mark.asyncio
async def test_main_reports_app_run_failure_and_still_shuts_down(monkeypatch):
    app_inst = SimpleNamespace(shutdown_called=False)

    async def shutdown():
        app_inst.shutdown_called = True

    async def run():
        raise RuntimeError('run failed')

    async def fake_make_app(loop):
        app_inst.shutdown = shutdown
        app_inst.run = run
        return app_inst

    print_exc = Mock()
    monkeypatch.setattr(signal, 'signal', lambda *_args: None)
    monkeypatch.setattr(boot, 'make_app', fake_make_app)
    monkeypatch.setattr(boot.traceback, 'print_exc', print_exc)

    await boot.main(SimpleNamespace())

    print_exc.assert_called_once()
    assert app_inst.shutdown_called is True