Python BaseBackend.on_chord_apply 예제들

예제 #1

파일: test_base.py 프로젝트: Birdbird/celery

class test_BaseBackend_interface(AppCase):

    def setup(self):
        self.b = BaseBackend(self.app)

    def test__forget(self):
        with self.assertRaises(NotImplementedError):
            self.b._forget('SOMExx-N0Nex1stant-IDxx-')

    def test_forget(self):
        with self.assertRaises(NotImplementedError):
            self.b.forget('SOMExx-N0nex1stant-IDxx-')

    def test_on_chord_part_return(self):
        self.b.on_chord_part_return(None)

    def test_on_chord_apply(self, unlock='celery.chord_unlock'):
        p, self.app.tasks[unlock] = self.app.tasks.get(unlock), Mock()
        try:
            self.b.on_chord_apply(
                'dakj221', 'sdokqweok',
                result=[self.app.AsyncResult(x) for x in [1, 2, 3]],
            )
            self.assertTrue(self.app.tasks[unlock].apply_async.call_count)
        finally:
            self.app.tasks[unlock] = p

예제 #2

class test_BaseBackend_interface(AppCase):
    def setup(self):
        self.b = BaseBackend(self.app)

    def test__forget(self):
        with self.assertRaises(NotImplementedError):
            self.b._forget('SOMExx-N0Nex1stant-IDxx-')

    def test_forget(self):
        with self.assertRaises(NotImplementedError):
            self.b.forget('SOMExx-N0nex1stant-IDxx-')

    def test_on_chord_part_return(self):
        self.b.on_chord_part_return(None)

    def test_on_chord_apply(self, unlock='celery.chord_unlock'):
        p, self.app.tasks[unlock] = self.app.tasks.get(unlock), Mock()
        try:
            self.b.on_chord_apply(
                'dakj221',
                'sdokqweok',
                result=[self.app.AsyncResult(x) for x in [1, 2, 3]],
            )
            self.assertTrue(self.app.tasks[unlock].apply_async.call_count)
        finally:
            self.app.tasks[unlock] = p