def test_cleanup():
# 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():
# 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():
# Broadcast queue should delete its own subscription, as it's not
# shared.
publisher_client = make_publisher_client()
subscriber_client = make_subscriber_client()
q = BroadcastQueue(
publisher_client, subscriber_client, 'test-project')
q.subscription = 'test-subscription'
q.cleanup()
subscriber_client.delete_subscription.assert_called_once_with(
'test-subscription')
