def test_create_subscription(self): # Setup Expected Response name_2 = "name2-1052831874" topic_2 = "topic2-1139259102" ack_deadline_seconds = 2135351438 retain_acked_messages = False expected_response = { "name": name_2, "topic": topic_2, "ack_deadline_seconds": ack_deadline_seconds, "retain_acked_messages": retain_acked_messages, } expected_response = pubsub_pb2.Subscription(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = subscriber_client.SubscriberClient() # Setup Request name = client.subscription_path("[PROJECT]", "[SUBSCRIPTION]") topic = client.topic_path("[PROJECT]", "[TOPIC]") response = client.create_subscription(name, topic) assert expected_response == response assert len(channel.requests) == 1 expected_request = pubsub_pb2.Subscription(name=name, topic=topic) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_create_subscription(self): # Setup Expected Response name_2 = 'name2-1052831874' topic_2 = 'topic2-1139259102' ack_deadline_seconds = 2135351438 retain_acked_messages = False expected_response = { 'name': name_2, 'topic': topic_2, 'ack_deadline_seconds': ack_deadline_seconds, 'retain_acked_messages': retain_acked_messages } expected_response = pubsub_pb2.Subscription(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = subscriber_client.SubscriberClient(channel=channel) # Setup Request name = client.subscription_path('[PROJECT]', '[SUBSCRIPTION]') topic = client.topic_path('[PROJECT]', '[TOPIC]') response = client.create_subscription(name, topic) assert expected_response == response assert len(channel.requests) == 1 expected_request = pubsub_pb2.Subscription(name=name, topic=topic) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_update_subscription(self): # Setup Expected Response name = 'name3373707' topic = 'topic110546223' ack_deadline_seconds = 2135351438 retain_acked_messages = False expected_response = { 'name': name, 'topic': topic, 'ack_deadline_seconds': ack_deadline_seconds, 'retain_acked_messages': retain_acked_messages } expected_response = pubsub_pb2.Subscription(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = subscriber_client.SubscriberClient(channel=channel) # Setup Request subscription = {} update_mask = {} response = client.update_subscription(subscription, update_mask) assert expected_response == response assert len(channel.requests) == 1 expected_request = pubsub_pb2.UpdateSubscriptionRequest( subscription=subscription, update_mask=update_mask) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_get_subscription(self): # Setup Expected Response name = 'name3373707' topic = 'topic110546223' ack_deadline_seconds = 2135351438 retain_acked_messages = False expected_response = { 'name': name, 'topic': topic, 'ack_deadline_seconds': ack_deadline_seconds, 'retain_acked_messages': retain_acked_messages } expected_response = pubsub_pb2.Subscription(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = subscriber_client.SubscriberClient(channel=channel) # Setup Request subscription = client.subscription_path('[PROJECT]', '[SUBSCRIPTION]') response = client.get_subscription(subscription) assert expected_response == response assert len(channel.requests) == 1 expected_request = pubsub_pb2.GetSubscriptionRequest( subscription=subscription) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_get_subscription(self): # Setup Expected Response name = "name3373707" topic = "topic110546223" ack_deadline_seconds = 2135351438 retain_acked_messages = False expected_response = { "name": name, "topic": topic, "ack_deadline_seconds": ack_deadline_seconds, "retain_acked_messages": retain_acked_messages, } expected_response = pubsub_pb2.Subscription(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = subscriber_client.SubscriberClient() # Setup Request subscription = client.subscription_path("[PROJECT]", "[SUBSCRIPTION]") response = client.get_subscription(subscription) assert expected_response == response assert len(channel.requests) == 1 expected_request = pubsub_pb2.GetSubscriptionRequest( subscription=subscription) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_update_subscription(self): # Setup Expected Response name = "name3373707" topic = "topic110546223" ack_deadline_seconds_2 = 921632575 retain_acked_messages = False enable_message_ordering = True filter_ = "filter-1274492040" detached = True expected_response = { "name": name, "topic": topic, "ack_deadline_seconds": ack_deadline_seconds_2, "retain_acked_messages": retain_acked_messages, "enable_message_ordering": enable_message_ordering, "filter": filter_, "detached": detached, } expected_response = pubsub_pb2.Subscription(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = subscriber_client.SubscriberClient() # Setup Request ack_deadline_seconds = 42 subscription = {"ack_deadline_seconds": ack_deadline_seconds} paths_element = "ack_deadline_seconds" paths = [paths_element] update_mask = {"paths": paths} response = client.update_subscription(subscription, update_mask) assert expected_response == response assert len(channel.requests) == 1 expected_request = pubsub_pb2.UpdateSubscriptionRequest( subscription=subscription, update_mask=update_mask) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_update_subscription(self): # Setup Expected Response name = 'name3373707' topic = 'topic110546223' ack_deadline_seconds_2 = 921632575 retain_acked_messages = False expected_response = { 'name': name, 'topic': topic, 'ack_deadline_seconds': ack_deadline_seconds_2, 'retain_acked_messages': retain_acked_messages } expected_response = pubsub_pb2.Subscription(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch('google.api_core.grpc_helpers.create_channel') with patch as create_channel: create_channel.return_value = channel client = subscriber_client.SubscriberClient() # Setup Request ack_deadline_seconds = 42 subscription = {'ack_deadline_seconds': ack_deadline_seconds} paths_element = 'ack_deadline_seconds' paths = [paths_element] update_mask = {'paths': paths} response = client.update_subscription(subscription, update_mask) assert expected_response == response assert len(channel.requests) == 1 expected_request = pubsub_pb2.UpdateSubscriptionRequest( subscription=subscription, update_mask=update_mask) actual_request = channel.requests[0][1] assert expected_request == actual_request