def test_list_instances(self): # Setup Expected Response next_page_token = "nextPageToken-1530815211" expected_response = {"next_page_token": next_page_token} expected_response = bigtable_instance_admin_pb2.ListInstancesResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = bigtable_admin_v2.BigtableInstanceAdminClient() # Setup Request parent = client.project_path("[PROJECT]") response = client.list_instances(parent) assert expected_response == response assert len(channel.requests) == 1 expected_request = bigtable_instance_admin_pb2.ListInstancesRequest( parent=parent) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_list_instances(self): from google.cloud.bigtable_admin_v2.proto import (instance_pb2 as data_v2_pb2) from google.cloud.bigtable_admin_v2.proto import ( bigtable_instance_admin_pb2 as messages_v2_pb2) from google.cloud.bigtable_admin_v2.gapic import \ bigtable_instance_admin_client from google.cloud.bigtable.instance import Instance FAILED_LOCATION = 'FAILED' INSTANCE_ID1 = 'instance-id1' INSTANCE_ID2 = 'instance-id2' INSTANCE_NAME1 = ('projects/' + self.PROJECT + '/instances/' + INSTANCE_ID1) INSTANCE_NAME2 = ('projects/' + self.PROJECT + '/instances/' + INSTANCE_ID2) credentials = _make_credentials() api = bigtable_instance_admin_client.BigtableInstanceAdminClient( mock.Mock()) client = self._make_one(project=self.PROJECT, credentials=credentials, admin=True) # Create response_pb response_pb = messages_v2_pb2.ListInstancesResponse( failed_locations=[ FAILED_LOCATION, ], instances=[ data_v2_pb2.Instance( name=INSTANCE_NAME1, display_name=INSTANCE_NAME1, ), data_v2_pb2.Instance( name=INSTANCE_NAME2, display_name=INSTANCE_NAME2, ), ], ) # Patch the stub used by the API method. client._instance_admin_client = api bigtable_instance_stub = (client.instance_admin_client.transport) bigtable_instance_stub.list_instances.side_effect = [response_pb] # Perform the method and check the result. instances, failed_locations = client.list_instances() instance_1, instance_2 = instances self.assertIsInstance(instance_1, Instance) self.assertEqual(instance_1.name, INSTANCE_NAME1) self.assertTrue(instance_1._client is client) self.assertIsInstance(instance_2, Instance) self.assertEqual(instance_2.name, INSTANCE_NAME2) self.assertTrue(instance_2._client is client) self.assertEqual(failed_locations, [FAILED_LOCATION])
def test_list_instances(self): from google.cloud.bigtable_admin_v2.proto import (instance_pb2 as data_v2_pb2) from google.cloud.bigtable_admin_v2.proto import ( bigtable_instance_admin_pb2 as messages_v2_pb2) FAILED_LOCATION = 'FAILED' INSTANCE_ID1 = 'instance-id1' INSTANCE_ID2 = 'instance-id2' INSTANCE_NAME1 = ('projects/' + self.PROJECT + '/instances/' + INSTANCE_ID1) INSTANCE_NAME2 = ('projects/' + self.PROJECT + '/instances/' + INSTANCE_ID2) channel = _make_channel() client = self._make_one(project=self.PROJECT, channel=channel, admin=True) # Create response_pb response_pb = messages_v2_pb2.ListInstancesResponse( failed_locations=[ FAILED_LOCATION, ], instances=[ data_v2_pb2.Instance( name=INSTANCE_NAME1, display_name=INSTANCE_NAME1, ), data_v2_pb2.Instance( name=INSTANCE_NAME2, display_name=INSTANCE_NAME2, ), ], ) # Patch the stub used by the API method. bigtable_instance_stub = ( client._instance_admin_client.bigtable_instance_admin_stub) bigtable_instance_stub.ListInstances.side_effect = [response_pb] expected_result = response_pb # Perform the method and check the result. result = client.list_instances() self.assertEqual(result, expected_result)
def test_list_instances(self): # Setup Expected Response next_page_token = 'nextPageToken-1530815211' expected_response = {'next_page_token': next_page_token} expected_response = bigtable_instance_admin_pb2.ListInstancesResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = bigtable_admin_v2.BigtableInstanceAdminClient(channel=channel) # Setup Request parent = client.project_path('[PROJECT]') response = client.list_instances(parent) assert expected_response == response assert len(channel.requests) == 1 expected_request = bigtable_instance_admin_pb2.ListInstancesRequest( parent=parent) actual_request = channel.requests[0][1] assert expected_request == actual_request