def test_list_instances_w_options(self): from google.cloud.spanner_admin_instance_v1 import InstanceAdminClient from google.cloud.spanner_admin_instance_v1 import ListInstancesRequest from google.cloud.spanner_admin_instance_v1 import ListInstancesResponse api = InstanceAdminClient(credentials=mock.Mock()) credentials = _make_credentials() client = self._make_one(project=self.PROJECT, credentials=credentials) client._instance_admin_api = api instance_pbs = ListInstancesResponse(instances=[]) li_api = api._transport._wrapped_methods[ api._transport.list_instances ] = mock.Mock(return_value=instance_pbs) page_size = 42 filter_ = "name:instance" list(client.list_instances(filter_=filter_, page_size=42)) expected_metadata = ( ("google-cloud-resource-prefix", client.project_name), ("x-goog-request-params", "parent={}".format(client.project_name)), ) li_api.assert_called_once_with( ListInstancesRequest(parent=self.PATH, filter=filter_, page_size=page_size), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, )
def test_list_instance_configs_w_options(self): from google.cloud.spanner_admin_instance_v1 import InstanceAdminClient from google.cloud.spanner_admin_instance_v1 import ( InstanceConfig as InstanceConfigPB, ) from google.cloud.spanner_admin_instance_v1 import ListInstanceConfigsRequest from google.cloud.spanner_admin_instance_v1 import ListInstanceConfigsResponse api = InstanceAdminClient(credentials=mock.Mock()) credentials = _make_credentials() client = self._make_one(project=self.PROJECT, credentials=credentials) client._instance_admin_api = api instance_config_pbs = ListInstanceConfigsResponse(instance_configs=[ InstanceConfigPB(name=self.CONFIGURATION_NAME, display_name=self.DISPLAY_NAME) ]) lic_api = api._transport._wrapped_methods[ api._transport.list_instance_configs] = mock.Mock( return_value=instance_config_pbs) page_size = 42 list(client.list_instance_configs(page_size=42)) expected_metadata = ( ("google-cloud-resource-prefix", client.project_name), ("x-goog-request-params", "parent={}".format(client.project_name)), ) lic_api.assert_called_once_with( ListInstanceConfigsRequest(parent=self.PATH, page_size=page_size), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, )
def instance_admin_api(self): """Helper for session-related API calls.""" if self._instance_admin_api is None: if self._emulator_host is not None: transport = InstanceAdminGrpcTransport( channel=grpc.insecure_channel(target=self._emulator_host)) self._instance_admin_api = InstanceAdminClient( client_info=self._client_info, client_options=self._client_options, transport=transport, ) else: self._instance_admin_api = InstanceAdminClient( credentials=self.credentials, client_info=self._client_info, client_options=self._client_options, ) return self._instance_admin_api
def test_list_instances(self): from google.cloud.spanner_admin_instance_v1 import InstanceAdminClient from google.cloud.spanner_admin_instance_v1 import Instance as InstancePB from google.cloud.spanner_admin_instance_v1 import ListInstancesRequest from google.cloud.spanner_admin_instance_v1 import ListInstancesResponse api = InstanceAdminClient(credentials=mock.Mock()) credentials = _make_credentials() client = self._make_one(project=self.PROJECT, credentials=credentials) client._instance_admin_api = api instance_pbs = ListInstancesResponse( instances=[ InstancePB( name=self.INSTANCE_NAME, config=self.CONFIGURATION_NAME, display_name=self.DISPLAY_NAME, node_count=self.NODE_COUNT, processing_units=self.PROCESSING_UNITS, ) ] ) li_api = api._transport._wrapped_methods[ api._transport.list_instances ] = mock.Mock(return_value=instance_pbs) response = client.list_instances() instances = list(response) instance = instances[0] self.assertIsInstance(instance, InstancePB) self.assertEqual(instance.name, self.INSTANCE_NAME) self.assertEqual(instance.config, self.CONFIGURATION_NAME) self.assertEqual(instance.display_name, self.DISPLAY_NAME) self.assertEqual(instance.node_count, self.NODE_COUNT) self.assertEqual(instance.processing_units, self.PROCESSING_UNITS) expected_metadata = ( ("google-cloud-resource-prefix", client.project_name), ("x-goog-request-params", "parent={}".format(client.project_name)), ) li_api.assert_called_once_with( ListInstancesRequest(parent=self.PATH), metadata=expected_metadata, retry=mock.ANY, timeout=mock.ANY, )