Exemplo n.º 1
0
    def test_execute_using_pool(self):
        tid = uuid()
        tw = TaskRequest(mytask.name, tid, [4], {"f": "x"})

        class MockPool(BasePool):
            target = None
            args = None
            kwargs = None

            def __init__(self, *args, **kwargs):
                pass

            def apply_async(self, target, args=None, kwargs=None,
                    *margs, **mkwargs):
                self.target = target
                self.args = args
                self.kwargs = kwargs

        p = MockPool()
        tw.execute_using_pool(p)
        self.assertTrue(p.target)
        self.assertEqual(p.args[0], mytask.name)
        self.assertEqual(p.args[1], tid)
        self.assertEqual(p.args[2], [4])
        self.assertIn("f", p.args[3])
        self.assertIn([4], p.args)
Exemplo n.º 2
0
    def test_execute_using_pool(self):
        tid = uuid()
        tw = TaskRequest(mytask.name, tid, [4], {"f": "x"})

        class MockPool(BasePool):
            target = None
            args = None
            kwargs = None

            def __init__(self, *args, **kwargs):
                pass

            def apply_async(self,
                            target,
                            args=None,
                            kwargs=None,
                            *margs,
                            **mkwargs):
                self.target = target
                self.args = args
                self.kwargs = kwargs

        p = MockPool()
        tw.execute_using_pool(p)
        self.assertTrue(p.target)
        self.assertEqual(p.args[0], mytask.name)
        self.assertEqual(p.args[1], tid)
        self.assertEqual(p.args[2], [4])
        self.assertIn("f", p.args[3])
        self.assertIn([4], p.args)
Exemplo n.º 3
0
    def test_execute_using_pool(self):
        tid = uuid()
        job = TaskRequest(self.mytask.name, tid, [4], {'f': 'x'}, app=self.app)

        class MockPool(BasePool):
            target = None
            args = None
            kwargs = None

            def __init__(self, *args, **kwargs):
                pass

            def apply_async(self, target, args=None, kwargs=None,
                            *margs, **mkwargs):
                self.target = target
                self.args = args
                self.kwargs = kwargs

        p = MockPool()
        job.execute_using_pool(p)
        self.assertTrue(p.target)
        self.assertEqual(p.args[0], self.mytask.name)
        self.assertEqual(p.args[1], tid)
        self.assertEqual(p.args[2], [4])
        self.assertIn('f', p.args[3])
        self.assertIn([4], p.args)

        job.task.accept_magic_kwargs = False
        job.execute_using_pool(p)
Exemplo n.º 4
0
 def test_execute_using_pool_does_not_execute_revoked(self):
     job = TaskRequest(
         self.mytask.name, uuid(), [1], {'f': 'x'}, app=self.app,
     )
     revoked.add(job.id)
     with self.assertRaises(TaskRevokedError):
         job.execute_using_pool(None)
Exemplo n.º 5
0
 def test_execute_using_pool_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"})
     revoked.add(tw.id)
     tw.execute_using_pool(None)
Exemplo n.º 6
0
 def test_execute_using_pool_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"})
     revoked.add(tw.task_id)
     tw.execute_using_pool(None)
Exemplo n.º 7
0
 def test_execute_using_pool_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'}, app=self.app)
     revoked.add(tw.id)
     with self.assertRaises(TaskRevokedError):
         tw.execute_using_pool(None)
Exemplo n.º 8
0
 def test_execute_using_pool_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'})
     revoked.add(tw.id)
     with self.assertRaises(TaskRevokedError):
         tw.execute_using_pool(None)