def test_update_instance(self): # Setup Expected Response name = 'name3373707' config = 'config-1354792126' display_name = 'displayName1615086568' node_count = 1539922066 expected_response = { 'name': name, 'config': config, 'display_name': display_name, 'node_count': node_count } expected_response = spanner_instance_admin_pb2.Instance( **expected_response) operation = operations_pb2.Operation( name='operations/test_update_instance', done=True) operation.response.Pack(expected_response) # Mock the API response channel = ChannelStub(responses=[operation]) client = spanner_admin_instance_v1.InstanceAdminClient(channel=channel) # Setup Request instance = {} field_mask = {} response = client.update_instance(instance, field_mask) result = response.result() assert expected_response == result assert len(channel.requests) == 1 expected_request = spanner_instance_admin_pb2.UpdateInstanceRequest( instance=instance, field_mask=field_mask) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_list_instance_configs(self): # Setup Expected Response next_page_token = "" instance_configs_element = {} instance_configs = [instance_configs_element] expected_response = { "next_page_token": next_page_token, "instance_configs": instance_configs, } expected_response = spanner_instance_admin_pb2.ListInstanceConfigsResponse( **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 = spanner_admin_instance_v1.InstanceAdminClient() # Setup Request parent = client.project_path("[PROJECT]") paged_list_response = client.list_instance_configs(parent) resources = list(paged_list_response) assert len(resources) == 1 assert expected_response.instance_configs[0] == resources[0] assert len(channel.requests) == 1 expected_request = spanner_instance_admin_pb2.ListInstanceConfigsRequest( parent=parent ) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_get_instance(self): # Setup Expected Response name_2 = 'name2-1052831874' config = 'config-1354792126' display_name = 'displayName1615086568' node_count = 1539922066 expected_response = { 'name': name_2, 'config': config, 'display_name': display_name, 'node_count': node_count } expected_response = spanner_instance_admin_pb2.Instance( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = spanner_admin_instance_v1.InstanceAdminClient(channel=channel) # Setup Request name = client.instance_path('[PROJECT]', '[INSTANCE]') response = client.get_instance(name) assert expected_response == response assert len(channel.requests) == 1 expected_request = spanner_instance_admin_pb2.GetInstanceRequest( name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_get_iam_policy(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = spanner_admin_instance_v1.InstanceAdminClient() # Mock request resource = client.instance_path('[PROJECT]', '[INSTANCE]') # Mock response version = 351608024 etag = b'21' expected_response = {'version': version, 'etag': etag} expected_response = policy_pb2.Policy(**expected_response) grpc_stub.GetIamPolicy.return_value = expected_response response = client.get_iam_policy(resource) self.assertEqual(expected_response, response) grpc_stub.GetIamPolicy.assert_called_once() args, kwargs = grpc_stub.GetIamPolicy.call_args self.assertEqual(len(args), 2) self.assertEqual(len(kwargs), 1) self.assertIn('metadata', kwargs) actual_request = args[0] expected_request = iam_policy_pb2.GetIamPolicyRequest( resource=resource) self.assertEqual(expected_request, actual_request)
def test_test_iam_permissions(self): # Setup Expected Response expected_response = {} expected_response = iam_policy_pb2.TestIamPermissionsResponse( **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 = spanner_admin_instance_v1.InstanceAdminClient() # Setup Request resource = client.instance_path("[PROJECT]", "[INSTANCE]") permissions = [] response = client.test_iam_permissions(resource, permissions) assert expected_response == response assert len(channel.requests) == 1 expected_request = iam_policy_pb2.TestIamPermissionsRequest( resource=resource, permissions=permissions ) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_test_iam_permissions(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = spanner_admin_instance_v1.InstanceAdminClient() # Mock request resource = client.instance_path('[PROJECT]', '[INSTANCE]') permissions = [] # Mock response expected_response = {} expected_response = iam_policy_pb2.TestIamPermissionsResponse( **expected_response) 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() args, kwargs = grpc_stub.TestIamPermissions.call_args self.assertEqual(len(args), 2) self.assertEqual(len(kwargs), 1) self.assertIn('metadata', kwargs) actual_request = args[0] expected_request = iam_policy_pb2.TestIamPermissionsRequest( resource=resource, permissions=permissions) self.assertEqual(expected_request, actual_request)
def test_set_iam_policy(self): # Setup Expected Response version = 351608024 etag = b"21" expected_response = {"version": version, "etag": etag} expected_response = policy_pb2.Policy(**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 = spanner_admin_instance_v1.InstanceAdminClient() # Setup Request resource = client.instance_path("[PROJECT]", "[INSTANCE]") policy = {} response = client.set_iam_policy(resource, policy) assert expected_response == response assert len(channel.requests) == 1 expected_request = iam_policy_pb2.SetIamPolicyRequest( resource=resource, policy=policy ) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_get_instance_config(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = spanner_admin_instance_v1.InstanceAdminClient() # Mock request name = client.instance_config_path('[PROJECT]', '[INSTANCE_CONFIG]') # Mock response name_2 = 'name2-1052831874' display_name = 'displayName1615086568' expected_response = {'name': name_2, 'display_name': display_name} expected_response = spanner_instance_admin_pb2.InstanceConfig( **expected_response) grpc_stub.GetInstanceConfig.return_value = expected_response response = client.get_instance_config(name) self.assertEqual(expected_response, response) grpc_stub.GetInstanceConfig.assert_called_once() args, kwargs = grpc_stub.GetInstanceConfig.call_args self.assertEqual(len(args), 2) self.assertEqual(len(kwargs), 1) self.assertIn('metadata', kwargs) actual_request = args[0] expected_request = spanner_instance_admin_pb2.GetInstanceConfigRequest( name=name) self.assertEqual(expected_request, actual_request)
def test_get_instance(self): # Setup Expected Response name_2 = "name2-1052831874" config = "config-1354792126" display_name = "displayName1615086568" node_count = 1539922066 expected_response = { "name": name_2, "config": config, "display_name": display_name, "node_count": node_count, } expected_response = spanner_instance_admin_pb2.Instance(**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 = spanner_admin_instance_v1.InstanceAdminClient() # Setup Request name = client.instance_path("[PROJECT]", "[INSTANCE]") response = client.get_instance(name) assert expected_response == response assert len(channel.requests) == 1 expected_request = spanner_instance_admin_pb2.GetInstanceRequest(name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_get_iam_policy(self): # Setup Expected Response version = 351608024 etag = b'21' expected_response = {'version': version, 'etag': etag} expected_response = policy_pb2.Policy(**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 = spanner_admin_instance_v1.InstanceAdminClient() # Setup Request resource = client.instance_path('[PROJECT]', '[INSTANCE]') response = client.get_iam_policy(resource) assert expected_response == response assert len(channel.requests) == 1 expected_request = iam_policy_pb2.GetIamPolicyRequest( resource=resource) actual_request = channel.requests[0][1] assert expected_request == actual_request
def sample_create_instance(): # Create a client client = spanner_admin_instance_v1.InstanceAdminClient() # Initialize request argument(s) instance = spanner_admin_instance_v1.Instance() instance.name = "name_value" instance.config = "config_value" instance.display_name = "display_name_value" request = spanner_admin_instance_v1.CreateInstanceRequest( parent="parent_value", instance_id="instance_id_value", instance=instance, ) # Make the request operation = client.create_instance(request=request) print("Waiting for operation to complete...") response = operation.result() # Handle the response print(response)
def test_get_instance_config(self): # Setup Expected Response name_2 = 'name2-1052831874' display_name = 'displayName1615086568' expected_response = {'name': name_2, 'display_name': display_name} expected_response = spanner_instance_admin_pb2.InstanceConfig( **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 = spanner_admin_instance_v1.InstanceAdminClient() # Setup Request name = client.instance_config_path('[PROJECT]', '[INSTANCE_CONFIG]') response = client.get_instance_config(name) assert expected_response == response assert len(channel.requests) == 1 expected_request = spanner_instance_admin_pb2.GetInstanceConfigRequest( name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_list_instance_configs(self): # Setup Expected Response next_page_token = '' instance_configs_element = {} instance_configs = [instance_configs_element] expected_response = { 'next_page_token': next_page_token, 'instance_configs': instance_configs } expected_response = spanner_instance_admin_pb2.ListInstanceConfigsResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = spanner_admin_instance_v1.InstanceAdminClient(channel=channel) # Setup Request parent = client.project_path('[PROJECT]') paged_list_response = client.list_instance_configs(parent) resources = list(paged_list_response) assert len(resources) == 1 assert expected_response.instance_configs[0] == resources[0] assert len(channel.requests) == 1 expected_request = spanner_instance_admin_pb2.ListInstanceConfigsRequest( parent=parent) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_delete_instance_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) client = spanner_admin_instance_v1.InstanceAdminClient(channel=channel) # Setup request name = client.instance_path('[PROJECT]', '[INSTANCE]') with pytest.raises(CustomException): client.delete_instance(name)
def sample_delete_instance(): # Create a client client = spanner_admin_instance_v1.InstanceAdminClient() # Initialize request argument(s) request = spanner_admin_instance_v1.DeleteInstanceRequest( name="name_value", ) # Make the request client.delete_instance(request=request)
def test_list_instance_configs_exception(self): channel = ChannelStub(responses=[CustomException()]) client = spanner_admin_instance_v1.InstanceAdminClient(channel=channel) # Setup request parent = client.project_path('[PROJECT]') paged_list_response = client.list_instance_configs(parent) with pytest.raises(CustomException): list(paged_list_response)
def test_get_iam_policy_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) client = spanner_admin_instance_v1.InstanceAdminClient(channel=channel) # Setup request resource = client.instance_path('[PROJECT]', '[INSTANCE]') with pytest.raises(CustomException): client.get_iam_policy(resource)
def sample_get_instance(): # Create a client client = spanner_admin_instance_v1.InstanceAdminClient() # Initialize request argument(s) request = spanner_admin_instance_v1.GetInstanceRequest(name="name_value", ) # Make the request response = client.get_instance(request=request) # Handle the response print(response)
def sample_set_iam_policy(): # Create a client client = spanner_admin_instance_v1.InstanceAdminClient() # Initialize request argument(s) request = iam_policy_pb2.SetIamPolicyRequest(resource="resource_value", ) # Make the request response = client.set_iam_policy(request=request) # Handle the response print(response)
def test_delete_instance_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch('google.api_core.grpc_helpers.create_channel') with patch as create_channel: create_channel.return_value = channel client = spanner_admin_instance_v1.InstanceAdminClient() # Setup request name = client.instance_path('[PROJECT]', '[INSTANCE]') with pytest.raises(CustomException): client.delete_instance(name)
def test_list_instances_exception(self): channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = spanner_admin_instance_v1.InstanceAdminClient() # Setup request parent = client.project_path("[PROJECT]") paged_list_response = client.list_instances(parent) with pytest.raises(CustomException): list(paged_list_response)
def test_get_iam_policy_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = spanner_admin_instance_v1.InstanceAdminClient() # Setup request resource = client.instance_path("[PROJECT]", "[INSTANCE]") with pytest.raises(CustomException): client.get_iam_policy(resource)
def sample_list_instance_configs(): # Create a client client = spanner_admin_instance_v1.InstanceAdminClient() # Initialize request argument(s) request = spanner_admin_instance_v1.ListInstanceConfigsRequest( parent="parent_value", ) # Make the request page_result = client.list_instance_configs(request=request) # Handle the response for response in page_result: print(response)
def test_get_iam_policy_exception(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = spanner_admin_instance_v1.InstanceAdminClient() # Mock request resource = client.instance_path('[PROJECT]', '[INSTANCE]') # Mock exception response grpc_stub.GetIamPolicy.side_effect = CustomException() self.assertRaises(errors.GaxError, client.get_iam_policy, resource)
def test_delete_instance_exception(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = spanner_admin_instance_v1.InstanceAdminClient() # Mock request name = client.instance_path('[PROJECT]', '[INSTANCE]') # Mock exception response grpc_stub.DeleteInstance.side_effect = CustomException() self.assertRaises(errors.GaxError, client.delete_instance, name)
def test_delete_instance(self): channel = ChannelStub() client = spanner_admin_instance_v1.InstanceAdminClient(channel=channel) # Setup Request name = client.instance_path('[PROJECT]', '[INSTANCE]') client.delete_instance(name) assert len(channel.requests) == 1 expected_request = spanner_instance_admin_pb2.DeleteInstanceRequest( name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_test_iam_permissions_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = spanner_admin_instance_v1.InstanceAdminClient() # Setup request resource = "resource-341064690" permissions = [] with pytest.raises(CustomException): client.test_iam_permissions(resource, permissions)
def test_list_instance_configs_exception(self, mock_create_stub): # Mock gRPC layer grpc_stub = mock.Mock() mock_create_stub.return_value = grpc_stub client = spanner_admin_instance_v1.InstanceAdminClient() # Mock request parent = client.project_path('[PROJECT]') # Mock exception response grpc_stub.ListInstanceConfigs.side_effect = CustomException() paged_list_response = client.list_instance_configs(parent) self.assertRaises(errors.GaxError, list, paged_list_response)
def sample_test_iam_permissions(): # Create a client client = spanner_admin_instance_v1.InstanceAdminClient() # Initialize request argument(s) request = iam_policy_pb2.TestIamPermissionsRequest( resource="resource_value", permissions=['permissions_value_1', 'permissions_value_2'], ) # Make the request response = client.test_iam_permissions(request=request) # Handle the response print(response)
def test_delete_instance(self): channel = ChannelStub() patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = spanner_admin_instance_v1.InstanceAdminClient() # Setup Request name = client.instance_path("[PROJECT]", "[INSTANCE]") client.delete_instance(name) assert len(channel.requests) == 1 expected_request = spanner_instance_admin_pb2.DeleteInstanceRequest(name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request