Function test_lifespan_startup_shutdown

tests/api_utils/test_app.py:136–181

Test application startup and shutdown sequence.

()

@pytest.mark.asyncio
async def test_lifespan_startup_shutdown():
    """Test application startup and shutdown sequence."""
    app_mock = MagicMock()

    # Set state.is_page_ready before the test
    state.is_page_ready = True
    mock_logger = MagicMock()
    state.logger = mock_logger

    # Mock all the dependencies
    with (
        patch("api_utils.app._setup_logging") as mock_setup_logging,
        patch("api_utils.app._initialize_globals") as mock_init_globals,
        patch("api_utils.app._initialize_proxy_settings") as mock_init_proxy,
        patch("api_utils.app.load_excluded_models") as mock_load_models,
        patch(
            "api_utils.app._start_stream_proxy", new_callable=AsyncMock
        ) as mock_start_proxy,
        patch(
            "api_utils.app._initialize_browser_and_page", new_callable=AsyncMock
        ) as mock_init_browser,
        patch(
            "api_utils.app._shutdown_resources", new_callable=AsyncMock
        ) as mock_shutdown,
        patch("api_utils.queue_worker", new_callable=AsyncMock),
        patch("api_utils.app.restore_original_streams") as mock_restore_streams,
    ):
        mock_setup_logging.return_value = (MagicMock(), MagicMock())

        # Get the lifespan context manager
        from api_utils.app import lifespan

        async with lifespan(app_mock):
            # Verify startup actions
            mock_init_globals.assert_called_once()
            mock_init_proxy.assert_called_once()
            mock_load_models.assert_called_once()
            mock_start_proxy.assert_called_once()
            mock_init_browser.assert_called_once()
            # Check actual log messages from the implementation
            mock_logger.info.assert_any_call("Starting AI Studio Proxy Server...")

        # Verify shutdown actions
        mock_shutdown.assert_called_once()
        mock_restore_streams.assert_called()
        mock_logger.info.assert_any_call("Shutting down server...")


@pytest.mark.asyncio