def test_list_database_operations_defaults(self): from google.api_core.operation import Operation from google.cloud.spanner_admin_database_v1 import CreateDatabaseMetadata from google.cloud.spanner_admin_database_v1 import DatabaseAdminClient from google.cloud.spanner_admin_database_v1 import ListDatabaseOperationsRequest from google.cloud.spanner_admin_database_v1 import ( ListDatabaseOperationsResponse, ) from google.cloud.spanner_admin_database_v1 import ( OptimizeRestoredDatabaseMetadata, ) from google.longrunning import operations_pb2 from google.protobuf.any_pb2 import Any api = DatabaseAdminClient(credentials=mock.Mock()) client = _Client(self.PROJECT) client.database_admin_api = api instance = self._make_one(self.INSTANCE_ID, client) create_database_metadata = Any() create_database_metadata.Pack( CreateDatabaseMetadata.pb(CreateDatabaseMetadata(database="database")) ) optimize_database_metadata = Any() optimize_database_metadata.Pack( OptimizeRestoredDatabaseMetadata.pb( OptimizeRestoredDatabaseMetadata(name="database") ) ) databases_pb = ListDatabaseOperationsResponse( operations=[ operations_pb2.Operation(name="op1", metadata=create_database_metadata), operations_pb2.Operation( name="op2", metadata=optimize_database_metadata ), ] ) ldo_api = api._transport._wrapped_methods[ api._transport.list_database_operations ] = mock.Mock(return_value=databases_pb) ops = instance.list_database_operations() expected_metadata = ( ("google-cloud-resource-prefix", instance.name), ("x-goog-request-params", "parent={}".format(instance.name)), ) ldo_api.assert_called_once_with( ListDatabaseOperationsRequest(parent=self.INSTANCE_NAME), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, ) self.assertTrue(all([type(op) == Operation for op in ops]))
def test_list_database_operations_w_options(self): from google.api_core.operation import Operation from google.cloud.spanner_admin_database_v1 import DatabaseAdminClient from google.cloud.spanner_admin_database_v1 import ListDatabaseOperationsRequest from google.cloud.spanner_admin_database_v1 import ( ListDatabaseOperationsResponse, ) from google.cloud.spanner_admin_database_v1 import RestoreDatabaseMetadata from google.cloud.spanner_admin_database_v1 import RestoreSourceType from google.cloud.spanner_admin_database_v1 import UpdateDatabaseDdlMetadata from google.longrunning import operations_pb2 from google.protobuf.any_pb2 import Any api = DatabaseAdminClient(credentials=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( RestoreDatabaseMetadata.pb( RestoreDatabaseMetadata(name="database", source_type=RestoreSourceType.BACKUP))) update_database_metadata = Any() update_database_metadata.Pack( UpdateDatabaseDdlMetadata.pb( UpdateDatabaseDdlMetadata(database="database", statements=["statements"]))) databases_pb = ListDatabaseOperationsResponse(operations=[ operations_pb2.Operation(name="op1", metadata=restore_database_metadata), operations_pb2.Operation(name="op2", metadata=update_database_metadata), ]) ldo_api = api._transport._wrapped_methods[ api._transport.list_database_operations] = mock.Mock( return_value=databases_pb) ops = instance.list_database_operations(filter_="filter", page_size=10) expected_metadata = ( ("google-cloud-resource-prefix", instance.name), ("x-goog-request-params", "parent={}".format(instance.name)), ) ldo_api.assert_called_once_with( ListDatabaseOperationsRequest(parent=self.INSTANCE_NAME, filter="filter", page_size=10), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, ) self.assertTrue(all([type(op) == Operation for op in ops]))