Function test_start

python/ray/serve/tests/test_haproxy_api.py:1277–1334

Test HAProxy start functionality.

(haproxy_api_cleanup)

@pytest.mark.asyncio
async def test_start(haproxy_api_cleanup):
    """Test HAProxy start functionality."""
    with tempfile.TemporaryDirectory() as temp_dir:
        config_file_path = os.path.join(temp_dir, "haproxy.cfg")
        socket_path = os.path.join(temp_dir, "admin.sock")

        # Create HAProxy config
        config = HAProxyConfig(
            http_options=HTTPOptions(
                host="127.0.0.1",
                port=8000,
                keep_alive_timeout_s=58,
            ),
            stats_port=8404,
            pass_health_checks=True,
            socket_path=socket_path,
            has_received_routes=True,
            has_received_servers=True,
        )

        # Add a backend so routes are populated
        backend = BackendConfig(
            name="test_backend",
            path_prefix="/",
            app_name="test_app",
            servers=[ServerConfig(name="server", host="127.0.0.1", port=9999)],
        )

        api = HAProxyApi(
            cfg=config,
            backend_configs={"test_backend": backend},
            config_file_path=config_file_path,
        )

        haproxy_api_cleanup(api)

        await api.start()

        assert api._proc is not None, "HAProxy process should exist"
        assert api._is_running(), "HAProxy should be running"

        # Verify config file contains expected content
        with open(config_file_path, "r") as f:
            config_content = f.read()
            assert "frontend http_frontend" in config_content
            assert f"bind 127.0.0.1:{config.frontend_port}" in config_content
            assert "acl healthcheck path -i /-/healthz" in config_content

        health_response = requests.get(
            f"http://127.0.0.1:{config.frontend_port}/-/healthz", timeout=5
        )
        assert (
            health_response.status_code == 503
        ), "Health check with no servers up should return 503"

        await api.stop()
        assert api._proc is None
        assert not api._is_running()