def test_url_http_status_invalid(self):
with self.assertRaises(Exception):
url_http_status_task_handler(input_params={'url': 'wtf.derp'})
def test_url_http_status_valid(self):
responses.add(responses.GET, 'http://test.url.whatever', body="")
responses.add(responses.GET, 'http://test.url.something_else', status=404, body="")
assert(url_http_status_task_handler(input_params={'url': 'http://test.url.whatever'})['status'] == 200)
assert(url_http_status_task_handler(input_params={'url': 'http://test.url.something_else'})['status'] == 404)
