def test_cleanup_no_subscription(): # test without subscription publisher_client = make_publisher_client() subscriber_client = make_subscriber_client() q = BroadcastQueue( publisher_client, subscriber_client, 'test-project') q.cleanup() subscriber_client.delete_subscription.assert_not_called()
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')
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()