def test_task_serializer(mock_seralization):
"""
Test task_serializer helper function.
"""
mock_seralization.dispatch.task_status.return_value = {'status': 'mock'}
mock_seralization.dispatch.spawned_tasks.return_value = {
'spawned_task': 'mock'
}
mock_seralization.dispatch.task_result_href.return_value = {
'_href': '/mock/path/'
}
task = {'id': 'mock_task'}
serialized_task = task_serializer(task)
mock_seralization.dispatch.task_status.assert_called_once()
mock_seralization.dispatch.spawned_tasks.assert_called_once()
mock_seralization.dispatch.task_result_href.assert_called_once()
expected_task = {
'status': 'mock',
'spawned_task': 'mock',
'_href': '/mock/path/'
}
if serialized_task != expected_task:
raise AssertionError(
"Task serializer did not generate expected task. \n" +
"Task: {}, \nExpected Task: {}".format(serialized_task,
expected_task))
def data(self):
"""
Generate a data report for this event.
@return: dictionary of this event's fields
@rtype: dict
"""
d = {'event_type': self.event_type,
'payload': self.payload,
'call_report': task_serializer(self.call_report)}
return d
def data(self):
"""
Generate a data report for this event.
@return: dictionary of this event's fields
@rtype: dict
"""
d = {
'event_type': self.event_type,
'payload': self.payload,
'call_report': task_serializer(self.call_report)
}
return d
def test_task_serializer(mock_seralization):
"""
Test task_serializer helper function.
"""
mock_seralization.task_status.return_value = {'status': 'mock'}
mock_seralization.spawned_tasks.return_value = {'spawned_task': 'mock'}
mock_seralization.task_result_href.return_value = {'_href': '/mock/path/'}
task = {'task_id': 'mock_task'}
serialized_task = task_serializer(task)
mock_seralization.task_status.assert_called_once()
mock_seralization.spawned_tasks.assert_called_once()
mock_seralization.task_result_href.assert_called_once()
expected_task = {'status': 'mock', 'spawned_task': 'mock', '_href': '/mock/path/'}
if serialized_task != expected_task:
raise AssertionError("Task serializer did not generate expected task. \n" +
"Task: {}, \nExpected Task: {}".format(serialized_task, expected_task))
