def test_list_databases(self): from google.cloud.spanner_admin_database_v1.gapic import ( database_admin_client) from google.cloud.spanner_admin_database_v1.proto import ( spanner_database_admin_pb2) from google.cloud.spanner_v1.database import Database api = database_admin_client.DatabaseAdminClient(mock.Mock()) client = _Client(self.PROJECT) client.database_admin_api = api instance = self._make_one(self.INSTANCE_ID, client) databases_pb = spanner_database_admin_pb2.ListDatabasesResponse( databases=[ spanner_database_admin_pb2.Database( name='{}/databases/aa'.format(self.INSTANCE_NAME)), spanner_database_admin_pb2.Database( name='{}/databases/bb'.format(self.INSTANCE_NAME)) ]) api._list_databases = mock.Mock(return_value=databases_pb) response = instance.list_databases() databases = list(response) self.assertIsInstance(databases[0], Database) self.assertTrue(databases[0].name.endswith('/aa')) self.assertTrue(databases[1].name.endswith('/bb')) api._list_databases.assert_called_once_with( spanner_database_admin_pb2.ListDatabasesRequest( parent=self.INSTANCE_NAME), metadata=[('google-cloud-resource-prefix', instance.name)], retry=mock.ANY, timeout=mock.ANY)
def test_list_databases_w_options(self): from google.cloud.spanner_admin_database_v1.gapic import ( database_admin_client) from google.cloud.spanner_admin_database_v1.proto import ( spanner_database_admin_pb2) api = database_admin_client.DatabaseAdminClient(mock.Mock()) client = _Client(self.PROJECT) client.database_admin_api = api instance = self._make_one(self.INSTANCE_ID, client) databases_pb = spanner_database_admin_pb2.ListDatabasesResponse( databases=[]) ld_api = api._inner_api_calls['list_databases'] = mock.Mock( return_value=databases_pb) page_size = 42 page_token = 'token' response = instance.list_databases(page_size=page_size, page_token=page_token) databases = list(response) self.assertEqual(databases, []) ld_api.assert_called_once_with( spanner_database_admin_pb2.ListDatabasesRequest( parent=self.INSTANCE_NAME, page_size=page_size, page_token=page_token), metadata=[('google-cloud-resource-prefix', instance.name)], retry=mock.ANY, timeout=mock.ANY)
def test_list_database_operations_w_options(self): from google.api_core.operation import Operation from google.cloud.spanner_admin_database_v1.gapic import database_admin_client from google.cloud.spanner_admin_database_v1.proto import ( spanner_database_admin_pb2, ) from google.longrunning import operations_pb2 from google.protobuf.any_pb2 import Any api = database_admin_client.DatabaseAdminClient(mock.Mock()) client = _Client(self.PROJECT) client.database_admin_api = api instance = self._make_one(self.INSTANCE_ID, client) restore_database_metadata = Any() restore_database_metadata.Pack( spanner_database_admin_pb2.RestoreDatabaseMetadata() ) update_database_metadata = Any() update_database_metadata.Pack( spanner_database_admin_pb2.UpdateDatabaseDdlMetadata() ) databases_pb = spanner_database_admin_pb2.ListDatabaseOperationsResponse( operations=[ operations_pb2.Operation( name="op1", metadata=restore_database_metadata ), operations_pb2.Operation(name="op2", metadata=update_database_metadata), ] ) ldo_api = api._inner_api_calls["list_database_operations"] = mock.Mock( return_value=databases_pb ) operations = instance.list_database_operations(filter_="filter", page_size=10) for op in operations: self.assertIsInstance(op, Operation) expected_metadata = [ ("google-cloud-resource-prefix", instance.name), ("x-goog-request-params", "parent={}".format(instance.name)), ] ldo_api.assert_called_once_with( spanner_database_admin_pb2.ListDatabaseOperationsRequest( parent=self.INSTANCE_NAME, filter="filter", page_size=10 ), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, )
def test_list_databases(self): from google.cloud.spanner_admin_database_v1.gapic import database_admin_client from google.cloud.spanner_admin_database_v1.proto import ( spanner_database_admin_pb2, ) from google.cloud.spanner_v1.database import Database api = database_admin_client.DatabaseAdminClient(mock.Mock()) client = _Client(self.PROJECT) client.database_admin_api = api instance = self._make_one(self.INSTANCE_ID, client) databases_pb = spanner_database_admin_pb2.ListDatabasesResponse( databases=[ spanner_database_admin_pb2.Database( name="{}/databases/aa".format(self.INSTANCE_NAME) ), spanner_database_admin_pb2.Database( name="{}/databases/bb".format(self.INSTANCE_NAME) ), ] ) ld_api = api._inner_api_calls["list_databases"] = mock.Mock( return_value=databases_pb ) response = instance.list_databases() databases = list(response) self.assertIsInstance(databases[0], Database) self.assertTrue(databases[0].name.endswith("/aa")) self.assertTrue(databases[1].name.endswith("/bb")) expected_metadata = [ ("google-cloud-resource-prefix", instance.name), ("x-goog-request-params", "parent={}".format(instance.name)), ] ld_api.assert_called_once_with( spanner_database_admin_pb2.ListDatabasesRequest(parent=self.INSTANCE_NAME), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, )
def test_list_backup_operations_defaults(self): from google.api_core.operation import Operation from google.cloud.spanner_admin_database_v1.gapic import database_admin_client from google.cloud.spanner_admin_database_v1.proto import backup_pb2 from google.longrunning import operations_pb2 from google.protobuf.any_pb2 import Any api = database_admin_client.DatabaseAdminClient(mock.Mock()) client = _Client(self.PROJECT) client.database_admin_api = api instance = self._make_one(self.INSTANCE_ID, client) create_backup_metadata = Any() create_backup_metadata.Pack(backup_pb2.CreateBackupMetadata()) operations_pb = backup_pb2.ListBackupOperationsResponse( operations=[ operations_pb2.Operation(name="op1", metadata=create_backup_metadata) ] ) ldo_api = api._inner_api_calls["list_backup_operations"] = mock.Mock( return_value=operations_pb ) operations = instance.list_backup_operations() for op in operations: self.assertIsInstance(op, Operation) expected_metadata = [ ("google-cloud-resource-prefix", instance.name), ("x-goog-request-params", "parent={}".format(instance.name)), ] ldo_api.assert_called_once_with( backup_pb2.ListBackupOperationsRequest(parent=self.INSTANCE_NAME), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, )
def test_list_backups_w_options(self): from google.cloud.spanner_admin_database_v1.gapic import database_admin_client from google.cloud.spanner_admin_database_v1.proto import backup_pb2 from google.cloud.spanner_v1.backup import Backup api = database_admin_client.DatabaseAdminClient(mock.Mock()) client = _Client(self.PROJECT) client.database_admin_api = api instance = self._make_one(self.INSTANCE_ID, client) backups_pb = backup_pb2.ListBackupsResponse( backups=[ backup_pb2.Backup(name=instance.name + "/backups/op1"), backup_pb2.Backup(name=instance.name + "/backups/op2"), backup_pb2.Backup(name=instance.name + "/backups/op3"), ] ) ldo_api = api._inner_api_calls["list_backups"] = mock.Mock( return_value=backups_pb ) backups = instance.list_backups(filter_="filter", page_size=10) for backup in backups: self.assertIsInstance(backup, Backup) expected_metadata = [ ("google-cloud-resource-prefix", instance.name), ("x-goog-request-params", "parent={}".format(instance.name)), ] ldo_api.assert_called_once_with( backup_pb2.ListBackupsRequest( parent=self.INSTANCE_NAME, filter="filter", page_size=10 ), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, )
def test_list_databases_w_options(self): from google.cloud.spanner_admin_database_v1.gapic import database_admin_client from google.cloud.spanner_admin_database_v1.proto import ( spanner_database_admin_pb2, ) api = database_admin_client.DatabaseAdminClient(mock.Mock()) client = _Client(self.PROJECT) client.database_admin_api = api instance = self._make_one(self.INSTANCE_ID, client) databases_pb = spanner_database_admin_pb2.ListDatabasesResponse( databases=[]) ld_api = api._inner_api_calls["list_databases"] = mock.Mock( return_value=databases_pb) page_size = 42 page_token = "token" response = instance.list_databases(page_size=page_size, page_token=page_token) databases = list(response) self.assertEqual(databases, []) expected_metadata = [ ("google-cloud-resource-prefix", instance.name), ("x-goog-request-params", "parent={}".format(instance.name)), ] ld_api.assert_called_once_with( spanner_database_admin_pb2.ListDatabasesRequest( parent=self.INSTANCE_NAME, page_size=page_size, page_token=page_token), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, )