Function _mock_urlopen

test/test_azure_helpers.py:36–45

Context manager that patches ``urllib.request.urlopen`` with a fake response.

(status: int, body: str)

@contextmanager
def _mock_urlopen(status: int, body: str):
    """Context manager that patches ``urllib.request.urlopen`` with a fake response."""
    mock_response = MagicMock()
    mock_response.__enter__ = lambda s: s
    mock_response.__exit__ = MagicMock(return_value=False)
    mock_response.status = status
    mock_response.read.return_value = body.encode("utf8")

    with patch("urllib.request.urlopen", return_value=mock_response) as mock_open:
        yield mock_open


class TestGetAzureResponse(unittest.TestCase):