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_can_verify_based_on_expectations(self):
fetched_result = json.loads(u'{ "user": { "name": "bigo", "items": [ { "foo": "bar", "baz": "bleh" }, { "foo": "poker", "size": "small" } ] } }')
some_assertions = ['user.name', 'something.not[0].there', 'user.items[1]', 'user.items[0].baz']
t = Task(url=u'http://smth.com', expected_fields=some_assertions)
worker = Worker(t)
worker.verify(fetched_result)
errors = worker.get_errors()
self.assertNotEquals([], errors)
self.assertFalse(worker.has_expected_data())
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'))
def test_worker_happy_path(self):
fetched_result = u'{ "user": { "name": "bigo", "items": [ { "foo": "bar", "baz": "bleh" }, { "foo": "poker", "size": "small" } ] } }'
responses.add(responses.GET,
"http://bigodin.es",
body=fetched_result,
content_type='application/json',
status=200)
some_assertions = ['user.name', 'user.items[1]', 'user.items[0].baz']
t = Task(url=u'http://bigodin.es', expected_fields=some_assertions)
worker = Worker(t)
worker.execute()
errors = worker.get_errors()
self.assertEquals(1, len(responses.calls))
self.assertEquals([], errors)
self.assertTrue(worker.has_expected_data())