Python BroadcastQueue.cleanupの例

コード例 #1

ファイル: broadcast_queue_test.py プロジェクト: squarish-paper/bookshelf

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

コード例 #2

ファイル: broadcast_queue_test.py プロジェクト: squarish-paper/bookshelf

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

コード例 #3

ファイル: broadcast_queue_test.py プロジェクト: ichinaski/psq

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

コード例 #4

ファイル: broadcast_queue_test.py プロジェクト: GoogleCloudPlatform/psq

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