def test_delete_snapshot(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = subscriber_client.SubscriberClient() # Mock request snapshot = client.snapshot_path('[PROJECT]', '[SNAPSHOT]') client.delete_snapshot(snapshot) grpc_stub.DeleteSnapshot.assert_called_once() args, kwargs = grpc_stub.DeleteSnapshot.call_args self.assertEqual(len(args), 2) self.assertEqual(len(kwargs), 1) self.assertIn('metadata', kwargs) actual_request = args[0] expected_request = pubsub_pb2.DeleteSnapshotRequest(snapshot=snapshot) self.assertEqual(expected_request, actual_request)
def test_delete_subscription(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = subscriber_client.SubscriberClient() # Mock request subscription = client.subscription_path('[PROJECT]', '[SUBSCRIPTION]') client.delete_subscription(subscription) grpc_stub.DeleteSubscription.assert_called_once() args, kwargs = grpc_stub.DeleteSubscription.call_args self.assertEqual(len(args), 2) self.assertEqual(len(kwargs), 1) self.assertIn('metadata', kwargs) actual_request = args[0] expected_request = pubsub_pb2.DeleteSubscriptionRequest( subscription=subscription) self.assertEqual(expected_request, actual_request)
def test_test_iam_permissions(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock(spec=iam_policy_pb2.IAMPolicyStub) mock_create_stub.return_value = grpc_stub client = subscriber_client.SubscriberClient() # Mock request resource = client.subscription_path('[PROJECT]', '[SUBSCRIPTION]') permissions = [] # Mock response expected_response = iam_policy_pb2.TestIamPermissionsResponse() grpc_stub.TestIamPermissions.return_value = expected_response response = client.test_iam_permissions(resource, permissions) self.assertEqual(expected_response, response) grpc_stub.TestIamPermissions.assert_called_once() request = grpc_stub.TestIamPermissions.call_args[0] self.assertEqual(resource, request.resource) self.assertEqual(permissions, request.permissions)
def test_get_iam_policy(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock(spec=iam_policy_pb2.IAMPolicyStub) mock_create_stub.return_value = grpc_stub client = subscriber_client.SubscriberClient() # Mock request resource = client.subscription_path('[PROJECT]', '[SUBSCRIPTION]') # Mock response version = 351608024 etag = b'21' expected_response = policy_pb2.Policy(version, etag) grpc_stub.GetIamPolicy.return_value = expected_response response = client.get_iam_policy(resource) self.assertEqual(expected_response, response) grpc_stub.GetIamPolicy.assert_called_once() request = grpc_stub.GetIamPolicy.call_args[0] self.assertEqual(resource, request.resource)
def test_pull(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock(spec=pubsub_pb2.SubscriberStub) mock_create_stub.return_value = grpc_stub client = subscriber_client.SubscriberClient() # Mock request subscription = client.subscription_path('[PROJECT]', '[SUBSCRIPTION]') max_messages = 496131527 # Mock response expected_response = pubsub_pb2.PullResponse() grpc_stub.Pull.return_value = expected_response response = client.pull(subscription, max_messages) self.assertEqual(expected_response, response) grpc_stub.Pull.assert_called_once() request = grpc_stub.Pull.call_args[0] self.assertEqual(subscription, request.subscription) self.assertEqual(max_messages, request.max_messages)
def test_modify_push_config(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = subscriber_client.SubscriberClient() # Mock request subscription = client.subscription_path('[PROJECT]', '[SUBSCRIPTION]') push_config = pubsub_pb2.PushConfig() client.modify_push_config(subscription, push_config) grpc_stub.ModifyPushConfig.assert_called_once() args, kwargs = grpc_stub.ModifyPushConfig.call_args self.assertEqual(len(args), 2) self.assertEqual(len(kwargs), 1) self.assertIn('metadata', kwargs) actual_request = args[0] expected_request = pubsub_pb2.ModifyPushConfigRequest( subscription=subscription, push_config=push_config) self.assertEqual(expected_request, actual_request)
def test_update_subscription(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = subscriber_client.SubscriberClient() # Mock request subscription = pubsub_pb2.Subscription() update_mask = field_mask_pb2.FieldMask() # Mock response name = 'name3373707' topic = 'topic110546223' ack_deadline_seconds = 2135351438 retain_acked_messages = False expected_response = pubsub_pb2.Subscription( name=name, topic=topic, ack_deadline_seconds=ack_deadline_seconds, retain_acked_messages=retain_acked_messages) grpc_stub.UpdateSubscription.return_value = expected_response response = client.update_subscription(subscription, update_mask) self.assertEqual(expected_response, response) grpc_stub.UpdateSubscription.assert_called_once() args, kwargs = grpc_stub.UpdateSubscription.call_args self.assertEqual(len(args), 2) self.assertEqual(len(kwargs), 1) self.assertIn('metadata', kwargs) actual_request = args[0] expected_request = pubsub_pb2.UpdateSubscriptionRequest( subscription=subscription, update_mask=update_mask) self.assertEqual(expected_request, actual_request)