def test_fetch_task(self): responses.add( responses.GET, 'http://example.com', body=json.dumps({'job': {'id': 1}}), content_type='application/json' ) task = fetch_task('http://example.com', None) self.assertEqual(task['id'], 1)
def test_fetch_task_should_sleep_when_requests_fails(self, mock_send, mock_sleep): fetch_task('http://example.com', None) mock_sleep.assert_called_once_with(20) self.assertTrue(mock_send.called)