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 setUp(self): self.gerrit = gerrit_api.Gerrit('chromium-review.googlesource.com', MockCredentials())