コード例 #1
0
ファイル: test_worker.py プロジェクト: shejianmin/celery
 def test_start_connection_error(self):
     c = self.NoopConsumer(send_events=False, pool=BasePool())
     c.loop.on_nth_call_do_raise(KeyError('foo'), SyntaxError('bar'))
     c.connection_errors = (KeyError, )
     try:
         with self.assertRaises(SyntaxError):
             c.start()
     finally:
         c.timer and c.timer.stop()
コード例 #2
0
ファイル: test_worker.py プロジェクト: ppageon/celery
 def test_start_channel_error(self):
     c = self.NoopConsumer(task_events=False, pool=BasePool())
     c.loop.on_nth_call_do_raise(KeyError('foo'), SyntaxError('bar'))
     c.channel_errors = (KeyError, )
     try:
         with pytest.raises(KeyError):
             c.start()
     finally:
         c.timer and c.timer.stop()
コード例 #3
0
    def __init__(self):
        self.ready_queue = FastQueue()
        self.timer = Timer()
        self.app = current_app
        self.event_dispatcher = Mock()
        self.controller = WorkController()
        self.task_consumer = Mock()

        from celery.concurrency.base import BasePool
        self.pool = BasePool(10)
コード例 #4
0
    def __init__(self):
        self.ready_queue = FastQueue()
        self.ready_queue.put(
            TaskRequest(mytask.name, uuid(), args=(2, 2), kwargs={}))
        self.eta_schedule = Timer()
        self.app = current_app
        self.event_dispatcher = Mock()
        self.controller = WorkController()

        from celery.concurrency.base import BasePool
        self.pool = BasePool(10)
コード例 #5
0
 def test_start_connection_error(self):
     def loop_side_effect():
         yield KeyError('foo')
         yield SyntaxError('bar')
     c = self.NoopConsumer(task_events=False, pool=BasePool())
     c.loop.side_effect = loop_side_effect()
     c.connection_errors = (KeyError,)
     try:
         with pytest.raises(SyntaxError):
             c.start()
     finally:
         c.timer and c.timer.stop()
コード例 #6
0
ファイル: test_control.py プロジェクト: tsailiming/celery
    def __init__(self, app):
        self.app = app
        self.buffer = FastQueue()
        self.handle_task = self.buffer.put
        self.timer = Timer()
        self.event_dispatcher = Mock()
        self.controller = WorkController()
        self.task_consumer = Mock()

        from celery.concurrency.base import BasePool
        self.pool = BasePool(10)
        self.task_buckets = defaultdict(lambda: None)
コード例 #7
0
    def __init__(self):
        self.ready_queue = FastQueue()
        self.ready_queue.put(
            TaskRequest(task_name=mytask.name,
                        task_id=gen_unique_id(),
                        args=(2, 2),
                        kwargs={}))
        self.eta_schedule = Timer()
        self.app = app_or_default()
        self.event_dispatcher = Mock()

        from celery.concurrency.base import BasePool
        self.pool = BasePool(10)
コード例 #8
0
    def __init__(self, app):
        self.app = app
        self.buffer = FastQueue()
        self.timer = Timer()
        self.event_dispatcher = Mock()
        self.controller = WorkController()
        self.task_consumer = Mock()
        self.prefetch_multiplier = 1
        self.initial_prefetch_count = 1

        from celery.concurrency.base import BasePool
        self.pool = BasePool(10)
        self.task_buckets = defaultdict(lambda: None)
        self.hub = None
コード例 #9
0
ファイル: test_worker.py プロジェクト: WoLpH/celery
    def test_start_connection_error(self):

        class MockConsumer(MainConsumer):
            iterations = 0

            def consume_messages(self):
                if not self.iterations:
                    self.iterations = 1
                    raise KeyError("foo")
                raise SyntaxError("bar")

        l = MockConsumer(self.ready_queue, self.eta_schedule, self.logger,
                             send_events=False, pool=BasePool())
        l.connection_errors = (KeyError, )
        self.assertRaises(SyntaxError, l.start)
        l.heart.stop()
        l.priority_timer.stop()
コード例 #10
0
ファイル: test_worker.py プロジェクト: stavxyz/celery
    def test_start_connection_error(self):

        class MockConsumer(Consumer):
            iterations = 0

            def loop(self, *args, **kwargs):
                if not self.iterations:
                    self.iterations = 1
                    raise KeyError('foo')
                raise SyntaxError('bar')

        l = MockConsumer(self.buffer.put, timer=self.timer,
                         send_events=False, pool=BasePool(), app=self.app)

        l.connection_errors = (KeyError, )
        self.assertRaises(SyntaxError, l.start)
        l.timer.stop()
コード例 #11
0
ファイル: test_worker.py プロジェクト: jsheffie/celery
    def test_start_channel_error(self):

        class MockConsumer(Consumer):
            iterations = 0

            def loop(self, *args, **kwargs):
                if not self.iterations:
                    self.iterations = 1
                    raise KeyError('foo')
                raise SyntaxError('bar')

        l = MockConsumer(self.ready_queue, timer=self.timer,
                             send_events=False, pool=BasePool())
        l.channel_errors = (KeyError, )
        with self.assertRaises(KeyError):
            l.start()
        l.timer.stop()
コード例 #12
0
    def test_start_connection_error(self):

        class MockConsumer(BlockingConsumer):
            iterations = 0

            def consume_messages(self):
                if not self.iterations:
                    self.iterations = 1
                    raise KeyError('foo')
                raise SyntaxError('bar')

        l = MockConsumer(self.ready_queue, timer=self.timer,
                         send_events=False, pool=BasePool())
        l.connection_errors = (KeyError, )
        with self.assertRaises(SyntaxError):
            l.start()
        l.heart.stop()
        l.timer.stop()
コード例 #13
0
    def test_start_channel_error(self):

        class MockConsumer(Consumer):
            iterations = 0

            def loop(self, *args, **kwargs):
                if not self.iterations:
                    self.iterations = 1
                    raise KeyError('foo')
                raise SyntaxError('bar')

        l = MockConsumer(self.buffer.put, timer=self.timer,
                         send_events=False, pool=BasePool(), app=self.app)
        l.controller = l.app.WorkController()
        l.pool = l.controller.pool = Mock()
        l.channel_errors = (KeyError,)
        with self.assertRaises(KeyError):
            l.start()
        l.timer.stop()
コード例 #14
0
    def test_start_channel_error(self):
        # Regression test for AMQPChannelExceptions that can occur within the
        # consumer. (i.e. 404 errors)

        class MockConsumer(BlockingConsumer):
            iterations = 0

            def consume_messages(self):
                if not self.iterations:
                    self.iterations = 1
                    raise KeyError('foo')
                raise SyntaxError('bar')

        l = MockConsumer(self.ready_queue, timer=self.timer,
                         send_events=False, pool=BasePool())

        l.channel_errors = (KeyError, )
        self.assertRaises(SyntaxError, l.start)
        l.heart.stop()
        l.timer.stop()
コード例 #15
0
    def test_start_channel_error(self):
        # Regression test for AMQPChannelExceptions that can occur within the
        # consumer. (i.e. 404 errors)

        class MockConsumer(MainConsumer):
            iterations = 0

            def consume_messages(self):
                if not self.iterations:
                    self.iterations = 1
                    raise KeyError("foo")
                raise SyntaxError("bar")

        l = MockConsumer(self.ready_queue, self.eta_schedule, self.logger,
                             send_events=False, pool=BasePool())

        l.channel_errors = (KeyError, )
        self.assertRaises(SyntaxError, l.start)
        l.heart.stop()
        l.priority_timer.stop()
コード例 #16
0
ファイル: test_worker.py プロジェクト: zzkristy/celery
 def __init__(self, *args, **kwargs):
     kwargs.setdefault('pool', BasePool(2))
     kwargs.setdefault('controller', Mock())
     super(_MyKombuConsumer, self).__init__(*args, **kwargs)
コード例 #17
0
 def test_interface_terminate_job(self):
     with self.assertRaises(NotImplementedError):
         BasePool(10).terminate_job(101)
コード例 #18
0
 def test_interface_on_terminate(self):
     p = BasePool(10)
     p.on_terminate()
コード例 #19
0
 def test_num_processes(self):
     self.assertEqual(BasePool(7).num_processes, 7)
コード例 #20
0
ファイル: __init__.py プロジェクト: crankycoder/zamboni-lib
 def test_interface_on_apply(self):
     BasePool(10).on_apply()
コード例 #21
0
ファイル: __init__.py プロジェクト: crankycoder/zamboni-lib
 def test_active(self):
     p = BasePool(10)
     self.assertFalse(p.active)
     p._state = p.RUN
     self.assertTrue(p.active)
コード例 #22
0
ファイル: test_concurrency.py プロジェクト: adrianrego/celery
 def test_interface_close(self):
     p = BasePool(10)
     p.on_close = Mock()
     p.close()
     self.assertEqual(p._state, p.CLOSE)
     p.on_close.assert_called_with()
コード例 #23
0
ファイル: __init__.py プロジェクト: crankycoder/zamboni-lib
 def test_interface_on_start(self):
     BasePool(10).on_start()
コード例 #24
0
 def test_active(self):
     p = BasePool(10)
     assert not p.active
     p._state = p.RUN
     assert p.active
コード例 #25
0
 def test_restart(self):
     p = BasePool(10)
     with pytest.raises(NotImplementedError):
         p.restart()
コード例 #26
0
 def test_interface_on_terminate(self):
     p = BasePool(10)
     p.on_terminate()
コード例 #27
0
 def test_does_not_debug(self):
     x = BasePool(10)
     x._does_debug = False
     x.apply_async(object)
コード例 #28
0
 def test_interface_close(self):
     p = BasePool(10)
     p.on_close = Mock()
     p.close()
     assert p._state == p.CLOSE
     p.on_close.assert_called_with()
コード例 #29
0
ファイル: test_worker.py プロジェクト: fxiang21/company_bt
 def __init__(self, *args, **kwargs):
     kwargs.setdefault('pool', BasePool(2))
     super(MyKombuConsumer, self).__init__(*args, **kwargs)
コード例 #30
0
 def test_interface_on_hard_timeout(self):
     self.assertIsNone(BasePool(10).on_hard_timeout(Mock()))
コード例 #31
0
ファイル: __init__.py プロジェクト: crankycoder/zamboni-lib
 def test_interface_on_stop(self):
     BasePool(10).on_stop()
コード例 #32
0
 def test_interface_close(self):
     p = BasePool(10)
     p.on_close = Mock()
     p.close()
     self.assertEqual(p._state, p.CLOSE)
     p.on_close.assert_called_with()
コード例 #33
0
ファイル: __init__.py プロジェクト: crankycoder/zamboni-lib
 def test_interface_info(self):
     self.assertDictEqual(BasePool(10).info, {})
コード例 #34
0
 def test_interface_no_close(self):
     self.assertIsNone(BasePool(10).on_close())
コード例 #35
0
ファイル: __init__.py プロジェクト: crankycoder/zamboni-lib
 def test_restart(self):
     p = BasePool(10)
     with self.assertRaises(NotImplementedError):
         p.restart()
コード例 #36
0
 def test_does_not_debug(self):
     x = BasePool(10)
     x._does_debug = False
     x.apply_async(object)