Python Mock.task_consumerの例

コード例 #1

ファイル: test_consumer.py プロジェクト: jdufresne/celery

    def test_stop(self):
        c = Mock()
        tasks = Tasks(c)
        assert c.task_consumer is None
        assert c.qos is None

        c.task_consumer = Mock()
        tasks.stop(c)

コード例 #2

ファイル: test_control.py プロジェクト: Scalr/celery

    def test_active_queues(self):
        import kombu

        x = kombu.Consumer(self.app.connection_for_read(),
                           [kombu.Queue('foo', kombu.Exchange('foo'), 'foo'),
                            kombu.Queue('bar', kombu.Exchange('bar'), 'bar')],
                           auto_declare=False)
        consumer = Mock()
        consumer.task_consumer = x
        panel = self.create_panel(consumer=consumer)
        r = panel.handle('active_queues')
        assert list(sorted(q['name'] for q in r)) == ['bar', 'foo']

コード例 #3

ファイル: test_control.py プロジェクト: celery/celery

    def test_active_queues(self):
        import kombu

        x = kombu.Consumer(
            self.app.connection_for_read(),
            [kombu.Queue("foo", kombu.Exchange("foo"), "foo"), kombu.Queue("bar", kombu.Exchange("bar"), "bar")],
            auto_declare=False,
        )
        consumer = Mock()
        consumer.task_consumer = x
        panel = self.create_panel(consumer=consumer)
        r = panel.handle("active_queues")
        assert list(sorted(q["name"] for q in r)) == ["bar", "foo"]

コード例 #4

ファイル: test_control.py プロジェクト: Scalr/celery

 def test_active_queues__empty(self):
     consumer = Mock(name='consumer')
     panel = self.create_panel(consumer=consumer)
     consumer.task_consumer = None
     assert not panel.handle('active_queues')