def test_unserialize_task(self): task = Task(add_func, [1, 2]) task_data = task.serialize() task2 = Task.unserialize_task(task_data) self.assertEqual(task._fn, task2._fn) self.assertEqual(task._args, task2._args) self.assertEqual(task._kwargs, task2._kwargs)
def test_run(self): task = Task(add_func, [1, 2]) result = task.run() self.assertEqual(result, 3, msg="Result is not correct for the arguments provided") task2 = Task(data={"key": 3}) with self.assertRaises(Exception): task2.run()
def test_serialize(self): task = Task(add_func, [1, 2]) task_data = task.serialize() self.assertTrue( "[1, 2]" in task_data and "tests.utils.add_func" in task_data, msg="Error in task serialization" ) task2 = Task(data={"key": 3}) task2_data = task2.serialize() self.assertEqual(task2_data, '{"key": 3}', msg="Data couldn't be serialized")
def test_enqueue_task(self): """ Here we're actually testing both enqueue and dequeue methods from the queue. """ task = Task(add_func, [1, 2]) self.queue.enqueue_task(task) messages = self.queue.dequeue() self.assertEqual(messages[0][1]._args, [1, 2]) task2 = Task(data={"key": 3}) self.queue.enqueue_task(task2) messages2 = self.queue.dequeue() self.assertEqual(messages2[0][1].data, {"key": 3})
def test_run(self): task = Task(add_func, [1, 2]) result = task.run() self.assertEqual(result, 3, msg="Result is not correct for the arguments provided") task2 = Task(data={ "key": 3 }) with self.assertRaises(Exception): task2.run()
def test_serialize(self): task = Task(add_func, [1, 2]) task_data = task.serialize() self.assertTrue('[1, 2]' in task_data and "tests.utils.add_func" in task_data, msg="Error in task serialization") task2 = Task(data={ "key": 3 }) task2_data = task2.serialize() self.assertEqual(task2_data, '{"key": 3}', msg="Data couldn't be serialized")