def test_get_empty_task_args(self): """Ensure get_task_args returns {} if no task_args.""" from furious.batcher import Message message = Message() self.assertEqual({}, message.get_task_args())
def test_get_task_args(self): """Ensure get_task_args returns the message task_args.""" from furious.batcher import Message task_args = {'other': 'zzz', 'nested': 1} options = {'task_args': task_args} message = Message(**options) self.assertEqual(task_args, message.get_task_args())