def setUp(self):
self.gerrit = gerrit_api.Gerrit('chromium-review.googlesource.com',
gerrit_api.Credentials(auth=MOCK_AUTH))
self.gerrit_read_only = gerrit_api.Gerrit(
'chromium-review.googlesource.com',
gerrit_api.Credentials(auth=MOCK_AUTH),
read_only=True)
def test_request_throttled(self, mock_method, time_mock_method,
sleep_mock):
gerrit_throttled = gerrit_api.Gerrit(
'chromium-review.googlesource.com',
gerrit_api.Credentials(auth=MOCK_AUTH), 0.1)
mock_method.return_value = _create_mock_return(None, 404)
time_mock_method.return_value = 100
gerrit_throttled._request(method='GET', request_path='/accounts/self')
# Call it twice to test the throttling.
gerrit_throttled._request(method='GET', request_path='/accounts/self')
sleep_mock.assert_called_once_with(0)
time_mock_method.return_value = 101
# Call it again after exceeding the throttle to cover the other branch.
gerrit_throttled._request(method='GET', request_path='/accounts/self')
def test_with_timeout_and_retries(self, mock_request):
mock_request.return_value = _create_mock_return(
'%s[]' % GERRIT_JSON_HEADER, 200)
g = gerrit_api.Gerrit(
'chromium-review.googlesource.com',
gerrit_api.Credentials(auth=MOCK_AUTH),
retry_config=requests.packages.urllib3.util.Retry(
total=1, status_forcelist=[500, 503]),
timeout=(1, 2))
g._request(method='GET', request_path='/self')
mock_request.assert_called_once_with(
data=None,
method='GET',
params=None,
url='https://chromium-review.googlesource.com/a/self',
headers=HEADERS,
hooks=g._instrumentation_hooks,
timeout=(1, 2))
def test_getitem_no_credentials(self):
creds = gerrit_api.Credentials(
netrc_path=self.netrc_file.name,
gitcookies_path=self.gitcookie_file.name)
self.assertRaises(KeyError, creds.__getitem__, 'some.other.host.com')
def test_getitem_with_gitcookies(self):
creds = gerrit_api.Credentials(
gitcookies_path=self.gitcookie_file.name)
auth = creds['chromium.googlesource.com']
self.assertEqual(auth, ('*****@*****.**', 'secret'))
def test_mock_credentials(self):
creds = gerrit_api.Credentials(auth=('*****@*****.**',
'secret'))
self.assertEqual(creds['whatever.com'],
('*****@*****.**', 'secret'))
