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)
