def test_retry_on_server_errors(self):
"""Test if the client retries when some HTTP errors are found"""
reqs = []
body = read_file('data/phabricator/phabricator_tasks_empty.json', 'rb')
responses = {
PHABRICATOR_TASKS_URL: [(502, 'error'), (503, 'error'),
(200, body)],
PHABRICATOR_USERS_URL: [(503, 'error'), (503, 'error'),
(503, 'error'), (503, 'error')],
PHABRICATOR_PHIDS_URL: [(404, 'not found')]
}
def request_callback(method, uri, headers):
reqs.append(httpretty.last_request())
resp = responses[uri].pop(0)
return (resp[0], headers, resp[1])
httpretty.register_uri(httpretty.POST,
PHABRICATOR_TASKS_URL,
responses=[
httpretty.Response(body=request_callback)
for _ in range(3)
])
httpretty.register_uri(httpretty.POST,
PHABRICATOR_USERS_URL,
responses=[
httpretty.Response(body=request_callback)
for _ in range(4)
])
httpretty.register_uri(httpretty.POST,
PHABRICATOR_PHIDS_URL,
responses=[
httpretty.Response(body=request_callback)
for _ in range(1)
])
# These tests are based on the maximum number of retries,
# set by default to 3. The client only retries 502 and 503
# HTTP errors.
client = ConduitClient(PHABRICATOR_URL, 'aaaa')
# After 3 tries (request + 2 retries) it gets the result
reqs = []
_ = [r for r in client.tasks()]
self.assertEqual(len(reqs), 3)
# After 4 tries (request + 3 retries) it fails
reqs = []
with self.assertRaises(requests.exceptions.RetryError):
_ = client.users("PHID-USER-2uk52xorcqb6sjvp467y")
self.assertEqual(len(reqs), 4)
# After 1 try if fails
reqs = []
with self.assertRaises(requests.exceptions.HTTPError):
_ = client.phids("PHID-APPS-PhabricatorHeraldApplication")
self.assertEqual(len(reqs), 1)
def test_tasks(self):
"""Test if a set of tasks is returned"""
http_requests = setup_http_server()
client = ConduitClient(PHABRICATOR_URL, 'aaaa')
dt = datetime.datetime(2016, 5, 3, 0, 0, 0)
result = client.tasks(from_date=dt)
result = [r for r in result]
self.assertEqual(len(result), 2)
expected = [{
'__conduit__': ['True'],
'output': ['json'],
'params': {
'__conduit__': {
'token': 'aaaa'
},
'attachments': {
'projects': True
},
'constraints': {
'modifiedStart': 1462233600
},
'order': 'outdated'
}
}, {
'__conduit__': ['True'],
'output': ['json'],
'params': {
'__conduit__': {
'token': 'aaaa'
},
'after': '335',
'attachments': {
'projects': True
},
'constraints': {
'modifiedStart': 1462233600
},
'order': 'outdated'
}
}]
self.assertEqual(len(http_requests), len(expected))
for i in range(len(expected)):
rparams = http_requests[i].parsed_body
rparams['params'] = json.loads(rparams['params'][0])
self.assertDictEqual(rparams, expected[i])
