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())