def test_cleanup(self):
# Broadcast queue should delete its own subscription, as it's not
# shared.
pubsub = Mock()
pubsub.topic.return_value = Mock()
q = BroadcastQueue(pubsub)
q.subscription = Mock()
q.cleanup()
assert q.subscription.delete.called
# test without subscription
q.subscription = None
q.cleanup()
def test_cleanup(self):
# Broadcast queue should delete its own subscription, as it's not
# shared.
pubsub = Mock()
pubsub.topic.return_value = Mock()
q = BroadcastQueue(pubsub)
q.subscription = Mock()
q.cleanup()
assert q.subscription.delete.called
# test without subscription
q.subscription = None
q.cleanup()
