Exemplos de Queue.extra_context em Python

Exemplo n.º 1

Arquivo: queue_test.py Projeto: harsha31/psq

def test_context():
    # Test queue-local context.
    pubsub = Mock()
    pubsub.topic.return_value = Mock()
    q = Queue(pubsub)

    with q.queue_context():
        assert current_queue == q

    # Test additional context manager.
    spy = Mock()

    @contextmanager
    def extra_context():
        spy()
        yield

    q.extra_context = extra_context

    with q.queue_context():
        assert spy.called

Exemplo n.º 2

Arquivo: queue_test.py Projeto: BroetchenEK/psq

    def test_context(self):
        # Test queue-local context.
        pubsub = Mock()
        pubsub.topic.return_value = Mock()
        q = Queue(pubsub)

        with q.queue_context():
            assert current_queue == q

        # Test additional context manager.
        spy = Mock()

        @contextmanager
        def extra_context():
            spy()
            yield

        q.extra_context = extra_context

        with q.queue_context():
            assert spy.called