def nack(self): """Decline to acknowldge the given message. This will cause the message to be re-delivered to the subscription. """ self._request_queue.put( requests.NackRequest(ack_id=self._ack_id, byte_size=self.size))
def test_nack(): msg = create_message(b"foo", ack_id="bogus_ack_id") with mock.patch.object(msg._request_queue, "put") as put: msg.nack() put.assert_called_once_with( requests.NackRequest(ack_id="bogus_ack_id", byte_size=30)) check_call_types(put, requests.NackRequest)
def test_nack(): msg = create_message(b'foo', ack_id='bogus_ack_id') with mock.patch.object(msg._request_queue, 'put') as put: msg.nack() put.assert_called_once_with( requests.NackRequest( ack_id='bogus_ack_id', byte_size=30, )) check_call_types(put, requests.NackRequest)
def test_nack(): manager = mock.create_autospec(streaming_pull_manager.StreamingPullManager, instance=True) dispatcher_ = dispatcher.Dispatcher(manager, mock.sentinel.queue) items = [requests.NackRequest(ack_id="ack_id_string", byte_size=10)] dispatcher_.nack(items) manager.send.assert_called_once_with( types.StreamingPullRequest(modify_deadline_ack_ids=["ack_id_string"], modify_deadline_seconds=[0]))
from google.cloud.pubsub_v1.subscriber._protocol import requests from google.cloud.pubsub_v1.subscriber._protocol import streaming_pull_manager from google.pubsub_v1 import types as gapic_types import mock import pytest @pytest.mark.parametrize( "item,method_name", [ (requests.AckRequest(0, 0, 0, ""), "ack"), (requests.DropRequest(0, 0, ""), "drop"), (requests.LeaseRequest(0, 0, ""), "lease"), (requests.ModAckRequest(0, 0), "modify_ack_deadline"), (requests.NackRequest(0, 0, ""), "nack"), ], ) def test_dispatch_callback(item, method_name): manager = mock.create_autospec(streaming_pull_manager.StreamingPullManager, instance=True) dispatcher_ = dispatcher.Dispatcher(manager, mock.sentinel.queue) items = [item] with mock.patch.object(dispatcher_, method_name) as method: dispatcher_.dispatch_callback(items) method.assert_called_once_with([item])
from google.cloud.pubsub_v1.subscriber._protocol import requests from google.cloud.pubsub_v1.subscriber._protocol import streaming_pull_manager from google.pubsub_v1 import types as gapic_types import mock import pytest @pytest.mark.parametrize( "item,method_name", [ (requests.AckRequest("0", 0, 0, ""), "ack"), (requests.DropRequest("0", 0, ""), "drop"), (requests.LeaseRequest("0", 0, ""), "lease"), (requests.ModAckRequest("0", 0), "modify_ack_deadline"), (requests.NackRequest("0", 0, ""), "nack"), ], ) def test_dispatch_callback_active_manager(item, method_name): manager = mock.create_autospec( streaming_pull_manager.StreamingPullManager, instance=True ) dispatcher_ = dispatcher.Dispatcher(manager, mock.sentinel.queue) items = [item] with mock.patch.object(dispatcher_, method_name) as method: dispatcher_.dispatch_callback(items) method.assert_called_once_with([item])
from google.cloud.pubsub_v1.subscriber._protocol import dispatcher from google.cloud.pubsub_v1.subscriber._protocol import helper_threads from google.cloud.pubsub_v1.subscriber._protocol import requests from google.cloud.pubsub_v1.subscriber._protocol import streaming_pull_manager import mock from six.moves import queue import pytest @pytest.mark.parametrize('item,method_name', [ (requests.AckRequest(0, 0, 0), 'ack'), (requests.DropRequest(0, 0), 'drop'), (requests.LeaseRequest(0, 0), 'lease'), (requests.ModAckRequest(0, 0), 'modify_ack_deadline'), (requests.NackRequest(0, 0), 'nack') ]) def test_dispatch_callback(item, method_name): manager = mock.create_autospec( streaming_pull_manager.StreamingPullManager, instance=True) dispatcher_ = dispatcher.Dispatcher(manager, mock.sentinel.queue) items = [item] with mock.patch.object(dispatcher_, method_name) as method: dispatcher_.dispatch_callback(items) method.assert_called_once_with([item]) def test_dispatch_callback_inactive():