Function create_mock_response

python/ray/_common/tests/test_usage_stats.py:1550–1585

(url: str, provider: str, error_providers: list[str])

def test_get_cloud_from_metadata_requests(monkeypatch):
    def create_mock_response(url: str, provider: str, error_providers: list[str]):
        # Create a mock response based on the URL.
        mock_response = Mock()

        if url == "http://metadata.google.internal/computeMetadata/v1":
            # GCP endpoint
            if "gcp" in error_providers:
                print("raising")
                raise requests.exceptions.ConnectionError()
            mock_response.status_code = 200 if provider == "gcp" else 400
        elif url == "http://169.254.169.254/latest/meta-data/":
            # AWS endpoint
            if "aws" in error_providers:
                raise requests.exceptions.ConnectionError()
            # Azure IMDS returns 400 (not 404) when queried with AWS endpoint format
            # because Azure requires the "Metadata: true" header (not sent in AWS queries).
            # See: https://learn.microsoft.com/en-us/azure/virtual-machines/instance-metadata-service#errors-and-debugging
            if provider == "azure":
                mock_response.status_code = (
                    400  # Bad Request (missing headers/wrong path)
                )
            else:
                mock_response.status_code = 200 if provider == "aws" else 404
        elif url == "http://169.254.169.254/metadata/instance?api-version=2021-12-13":
            # Azure endpoint
            if "azure" in error_providers:
                raise requests.exceptions.ConnectionError()
            # AWS IMDS returns 404 when queried with Azure endpoint format
            # because Azure's URL path doesn't exist on AWS metadata service.
            # See: https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html#instance-metadata-returns
            if provider == "aws":
                mock_response.status_code = 404  # Not Found
            else:
                mock_response.status_code = 200 if provider == "azure" else 400

        return mock_response

    with patch("requests.get") as mock_get:
        mock_get.side_effect = lambda url, **kwargs: create_mock_response(