def test_pipeline_success(self):
httpretty.register_uri(httpretty.POST, "http://sup.com/callback")
pipeline = Pipeline().map(lowercase)
pipeline_task.apply(args=(pipeline, ['AA', 'BB'], 'http://sup.com/callback')).get()
resp_json = json.loads(httpretty.last_request().body.decode('utf-8'))
self.assertEquals({
'results': ['aa', 'bb'],
'error': None
}, resp_json)
def test_pipeline_failure(self):
httpretty.register_uri(httpretty.POST, "http://sup.com/callback")
pipeline = Pipeline().map(exception)
pipeline_task.apply(args=(pipeline, ['AA', 'BB'], 'http://sup.com/callback')).get()
resp_json = json.loads(httpretty.last_request().body.decode('utf-8'))
# for easier comparison
self.assertTrue(resp_json['error']['traceback'])
resp_json['error']['traceback'] = []
self.assertEquals({
'results': None,
'error': {
'name': 'Exception',
'message': 'exc message',
'traceback': []
}
}, resp_json)
