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
