def test_worker_knows_how_to_request(self):
responses.add(responses.GET,
"http://foo.com/",
status=200)
task = Task(url="http://foo.com")
worker = Worker(task)
worker.fetch()
self.assertEquals(1, len(responses.calls))
def test_worker_gets_a_404(self):
responses.add(responses.GET,
"http://www.idonotexist.com",
body='{"error": "not found"}',
content_type='application/json',
status=404)
task = Task(url="http://www.idonotexist.com")
w = Worker(task)
bad_result = w.fetch()
self.assertEquals(1, len(w.get_errors()))
self.assertEquals(404, w.get_errors()[0].get('code'))
self.assertIsNone(bad_result)
def test_worker_invalid_response(self):
responses.add(responses.GET,
"http://www.bigo.com",
body='i wish i was a json',
content_type='application/json',
status=200)
t = Task(url='http://www.bigo.com')
worker = Worker(t)
bad_result = worker.fetch()
self.assertIsNone(bad_result)
errors = worker.get_errors()
self.assertEquals(CUSTOM_ERRORS.invalid_response, errors[0].get('code'))
