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