Function test_shutdown_resources

tests/api_utils/test_app.py:479–513

Test _shutdown_resources cleans up everything.

()

@pytest.mark.asyncio
async def test_shutdown_resources():
    """Test _shutdown_resources cleans up everything."""

    # Create a dummy task in the current loop
    async def dummy_coro():
        await asyncio.sleep(0.1)

    real_task = asyncio.create_task(dummy_coro())

    # Set up state with mocked resources
    mock_stream_process = MagicMock()
    state.STREAM_PROCESS = mock_stream_process
    state.worker_task = real_task
    state.page_instance = MagicMock()

    mock_browser = MagicMock()
    mock_browser.is_connected.return_value = True
    mock_browser.close = AsyncMock()
    state.browser_instance = mock_browser

    mock_pw = MagicMock()
    mock_pw.stop = AsyncMock()
    state.playwright_manager = mock_pw

    with patch(
        "api_utils.app._close_page_logic", new_callable=AsyncMock
    ) as mock_close_page:
        await _shutdown_resources()

        mock_stream_process.terminate.assert_called_once()
        # The task should be cancelled
        assert real_task.cancelled() or real_task.done()
        mock_close_page.assert_called_once()
        mock_browser.close.assert_called_once()
        mock_pw.stop.assert_called_once()


@pytest.mark.asyncio