def test_retry_failed_requests_attribute_error(self, m_handle_request): m_handle_request.side_effect = AttributeError gh_req = GithubRequest.objects.create( action=u'issues', body=json.dumps({'test': True}), method=u'post', time=timezone.now(), ) retry_failed_requests() m_handle_request.assert_called_once_with(gh_req)
def test_retry_failed_requests(self, m_handle_request): for i in range(5): GithubRequest.objects.create( action=u'issues', body=json.dumps({'test': True}), method=u'post', time=timezone.now(), ) retry_failed_requests() self.assertEqual(m_handle_request.call_count, 5) for gh_req in GithubRequest.objects.all(): m_handle_request.assert_any_call(gh_req)
def handle(self, *args, **kwargs): retry_failed_requests()