def test_create_database(self, mock_create_stub):
# Mock gRPC layer
grpc_stub = mock.Mock()
mock_create_stub.return_value = grpc_stub
client = database_admin_client.DatabaseAdminClient()
# Mock request
parent = client.instance_path('[PROJECT]', '[INSTANCE]')
create_statement = 'createStatement552974828'
# Mock response
name = 'name3373707'
expected_response = spanner_database_admin_pb2.Database(name=name)
operation = operations_pb2.Operation(
name='operations/test_create_database', done=True)
operation.response.Pack(expected_response)
grpc_stub.CreateDatabase.return_value = operation
response = client.create_database(parent, create_statement)
self.assertEqual(expected_response, response.result())
grpc_stub.CreateDatabase.assert_called_once()
args, kwargs = grpc_stub.CreateDatabase.call_args
self.assertEqual(len(args), 2)
self.assertEqual(len(kwargs), 1)
self.assertIn('metadata', kwargs)
actual_request = args[0]
expected_request = spanner_database_admin_pb2.CreateDatabaseRequest(
parent=parent, create_statement=create_statement)
self.assertEqual(expected_request, actual_request)
def test_get_database(self, mock_create_stub):
# Mock gRPC layer
grpc_stub = mock.Mock()
mock_create_stub.return_value = grpc_stub
client = database_admin_client.DatabaseAdminClient()
# Mock request
name = client.database_path('[PROJECT]', '[INSTANCE]', '[DATABASE]')
# Mock response
name_2 = 'name2-1052831874'
expected_response = spanner_database_admin_pb2.Database(name=name_2)
grpc_stub.GetDatabase.return_value = expected_response
response = client.get_database(name)
self.assertEqual(expected_response, response)
grpc_stub.GetDatabase.assert_called_once()
args, kwargs = grpc_stub.GetDatabase.call_args
self.assertEqual(len(args), 2)
self.assertEqual(len(kwargs), 1)
self.assertIn('metadata', kwargs)
actual_request = args[0]
expected_request = spanner_database_admin_pb2.GetDatabaseRequest(
name=name)
self.assertEqual(expected_request, actual_request)
def test_list_databases(self, mock_create_stub):
# Mock gRPC layer
grpc_stub = mock.Mock()
mock_create_stub.return_value = grpc_stub
client = database_admin_client.DatabaseAdminClient()
# Mock request
parent = client.instance_path('[PROJECT]', '[INSTANCE]')
# Mock response
next_page_token = ''
databases_element = spanner_database_admin_pb2.Database()
databases = [databases_element]
expected_response = spanner_database_admin_pb2.ListDatabasesResponse(
next_page_token=next_page_token, databases=databases)
grpc_stub.ListDatabases.return_value = expected_response
paged_list_response = client.list_databases(parent)
resources = list(paged_list_response)
self.assertEqual(1, len(resources))
self.assertEqual(expected_response.databases[0], resources[0])
grpc_stub.ListDatabases.assert_called_once()
args, kwargs = grpc_stub.ListDatabases.call_args
self.assertEqual(len(args), 2)
self.assertEqual(len(kwargs), 1)
self.assertIn('metadata', kwargs)
actual_request = args[0]
expected_request = spanner_database_admin_pb2.ListDatabasesRequest(
parent=parent)
self.assertEqual(expected_request, actual_request)
def test_from_pb_bad_database_name(self):
from google.cloud.proto.spanner.admin.database.v1 import (
spanner_database_admin_pb2 as admin_v1_pb2)
database_name = 'INCORRECT_FORMAT'
database_pb = admin_v1_pb2.Database(name=database_name)
klass = self._getTargetClass()
with self.assertRaises(ValueError):
klass.from_pb(database_pb, None)
def test_from_pb_project_mistmatch(self):
from google.cloud.proto.spanner.admin.database.v1 import (
spanner_database_admin_pb2 as admin_v1_pb2)
ALT_PROJECT = 'ALT_PROJECT'
client = _Client(project=ALT_PROJECT)
instance = _Instance(self.INSTANCE_NAME, client)
database_pb = admin_v1_pb2.Database(name=self.DATABASE_NAME)
klass = self._getTargetClass()
with self.assertRaises(ValueError):
klass.from_pb(database_pb, instance)
def test_from_pb_instance_mistmatch(self):
from google.cloud.proto.spanner.admin.database.v1 import (
spanner_database_admin_pb2 as admin_v1_pb2)
ALT_INSTANCE = '/projects/%s/instances/ALT-INSTANCE' % (
self.PROJECT_ID,)
client = _Client()
instance = _Instance(ALT_INSTANCE, client)
database_pb = admin_v1_pb2.Database(name=self.DATABASE_NAME)
klass = self._getTargetClass()
with self.assertRaises(ValueError):
klass.from_pb(database_pb, instance)
def test_from_pb_success_w_explicit_pool(self):
from google.cloud.proto.spanner.admin.database.v1 import (
spanner_database_admin_pb2 as admin_v1_pb2)
client = _Client()
instance = _Instance(self.INSTANCE_NAME, client)
database_pb = admin_v1_pb2.Database(name=self.DATABASE_NAME)
klass = self._getTargetClass()
pool = _Pool()
database = klass.from_pb(database_pb, instance, pool=pool)
self.assertTrue(isinstance(database, klass))
self.assertEqual(database._instance, instance)
self.assertEqual(database.database_id, self.DATABASE_ID)
self.assertIs(database._pool, pool)
def test_from_pb_success_w_hyphen_w_default_pool(self):
from google.cloud.proto.spanner.admin.database.v1 import (
spanner_database_admin_pb2 as admin_v1_pb2)
from google.cloud.spanner.pool import BurstyPool
DATABASE_ID_HYPHEN = 'database-id'
DATABASE_NAME_HYPHEN = (
self.INSTANCE_NAME + '/databases/' + DATABASE_ID_HYPHEN)
client = _Client()
instance = _Instance(self.INSTANCE_NAME, client)
database_pb = admin_v1_pb2.Database(name=DATABASE_NAME_HYPHEN)
klass = self._getTargetClass()
database = klass.from_pb(database_pb, instance)
self.assertTrue(isinstance(database, klass))
self.assertEqual(database._instance, instance)
self.assertEqual(database.database_id, DATABASE_ID_HYPHEN)
self.assertIsInstance(database._pool, BurstyPool)
# BurstyPool does not create sessions during 'bind()'.
self.assertTrue(database._pool._sessions.empty())
