def test_agent_succeeded(self, mock_task_succeeded): dispatch_report = dict(succeeded=True) task_id = 'task_1' consumer_id = 'consumer_1' repo_id = 'repo_1' dist_id = 'dist_1' call_context = { 'task_id': task_id, 'consumer_id': consumer_id, 'repo_id': repo_id, 'distributor_id': dist_id } result = dict(retval=dispatch_report) envelope = Envelope(routing=['A', 'B'], result=result, any=call_context) reply = Succeeded(envelope) handler = ReplyHandler('') handler.succeeded(reply) # validate task updated mock_task_succeeded.assert_called_with(task_id, dispatch_report)
def test_unbind_succeeded_with_error_report(self, mock_task_succeeded, mock_unbind_failed): task_id = 'task_1' consumer_id = 'consumer_1' repo_id = 'repo_1' dist_id = 'dist_1' call_context = { 'action': 'unbind', 'task_id': task_id, 'consumer_id': consumer_id, 'repo_id': repo_id, 'distributor_id': dist_id } dispatch_report = dict(succeeded=False) result = Envelope(retval=dispatch_report) envelope = Envelope(routing=['A', 'B'], result=result, any=call_context) reply = Succeeded(envelope) handler = ReplyHandler('') handler.succeeded(reply) # validate task updated mock_task_succeeded.assert_called_with(task_id, dispatch_report) # validate bind action updated mock_unbind_failed.assert_called_with(task_id, call_context)