Function patch_urlopen

scripts/pr_quality/tests/test_check_pr.py:115–133

(json_responses=(), status_code=200)

def patch_urlopen(json_responses=(), status_code=200):
    side_effects = []
    if status_code == 200:
        for data in json_responses:
            mock_response = mock.MagicMock()
            mock_response.read.return_value = json.dumps(data).encode()
            mock_cm = mock.MagicMock()
            mock_cm.__enter__.return_value = mock_response
            side_effects.append(mock_cm)
    else:
        error = urllib.error.HTTPError(
            url="https://example.com",
            code=status_code,
            msg="Error",
            hdrs=None,
            fp=None,
        )
        side_effects.append(error)
    return mock.patch("urllib.request.urlopen", side_effect=side_effects)


class BaseTestCase(unittest.TestCase):