Ejemplo n.º 1
0
    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)
Ejemplo n.º 2
0
    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)
Ejemplo n.º 3
0
    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()
Ejemplo n.º 4
0
    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")
Ejemplo n.º 5
0
    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})
Ejemplo n.º 6
0
    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()
Ejemplo n.º 7
0
    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")