Function test_lifespan_startup_failure

tests/api_utils/test_app.py:185–215

Test application startup failure handling.

()

@pytest.mark.asyncio
async def test_lifespan_startup_failure():
    """Test application startup failure handling."""
    app_mock = MagicMock()

    mock_logger = MagicMock()
    state.logger = mock_logger

    with (
        patch("api_utils.app._setup_logging") as mock_setup_logging,
        patch("api_utils.app._initialize_globals"),
        patch("api_utils.app._initialize_proxy_settings"),
        patch("api_utils.app.load_excluded_models"),
        patch(
            "api_utils.app._start_stream_proxy", side_effect=Exception("Startup failed")
        ),
        patch(
            "api_utils.app._shutdown_resources", new_callable=AsyncMock
        ) as mock_shutdown,
        patch("api_utils.app.restore_original_streams"),
    ):
        mock_setup_logging.return_value = (MagicMock(), MagicMock())

        from api_utils.app import lifespan

        with pytest.raises(RuntimeError, match="Application startup failed"):
            async with lifespan(app_mock):
                pass

        # Verify shutdown was called even after failure
        mock_shutdown.assert_called()
        mock_logger.critical.assert_called()


# --- New Tests for Helper Functions ---