Python BaseBackend.wait_forの例

コード例 #1

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

    def test_wait_for__on_interval(self):
        self.patch("time.sleep")
        b = BaseBackend(app=self.app)
        b._get_task_meta_for = Mock()
        b._get_task_meta_for.return_value = {"status": states.PENDING}
        callback = Mock(name="callback")
        with self.assertRaises(TimeoutError):
            b.wait_for(task_id="1", on_interval=callback, timeout=1)
        callback.assert_called_with()

        b._get_task_meta_for.return_value = {"status": states.SUCCESS}
        b.wait_for(task_id="1", timeout=None)

コード例 #2

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

    def test_wait_for__on_interval(self):
        self.patching('time.sleep')
        b = BaseBackend(app=self.app)
        b._get_task_meta_for = Mock()
        b._get_task_meta_for.return_value = {'status': states.PENDING}
        callback = Mock(name='callback')
        with pytest.raises(TimeoutError):
            b.wait_for(task_id='1', on_interval=callback, timeout=1)
        callback.assert_called_with()

        b._get_task_meta_for.return_value = {'status': states.SUCCESS}
        b.wait_for(task_id='1', timeout=None)

コード例 #3

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

    def test_wait_for__on_interval(self):
        self.patching('time.sleep')
        b = BaseBackend(app=self.app)
        b._get_task_meta_for = Mock()
        b._get_task_meta_for.return_value = {'status': states.PENDING}
        callback = Mock(name='callback')
        with pytest.raises(TimeoutError):
            b.wait_for(task_id='1', on_interval=callback, timeout=1)
        callback.assert_called_with()

        b._get_task_meta_for.return_value = {'status': states.SUCCESS}
        b.wait_for(task_id='1', timeout=None)