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
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