def reload(self): """Reload the metadata for this instance.""" request_pb = messages_v2_pb2.GetInstanceRequest(name=self.name) # We expect `data_v2_pb2.Instance`. instance_pb = self._client._instance_stub.GetInstance(request_pb) # NOTE: _update_from_pb does not check that the project and # instance ID on the response match the request. self._update_from_pb(instance_pb)
def test_reload(self): from google.cloud.bigtable._generated import ( instance_pb2 as data_v2_pb2) from google.cloud.bigtable._generated import ( bigtable_instance_admin_pb2 as messages_v2_pb) from unit_tests._testing import _FakeStub client = _Client(self.PROJECT) instance = self._make_one(self.INSTANCE_ID, client, self.LOCATION_ID) # Create request_pb request_pb = messages_v2_pb.GetInstanceRequest( name=self.INSTANCE_NAME) # Create response_pb DISPLAY_NAME = u'hey-hi-hello' response_pb = data_v2_pb2.Instance( display_name=DISPLAY_NAME, ) # Patch the stub used by the API method. client._instance_stub = stub = _FakeStub(response_pb) # Create expected_result. expected_result = None # reload() has no return value. # Check Instance optional config values before. self.assertEqual(instance.display_name, self.INSTANCE_ID) # Perform the method and check the result. result = instance.reload() self.assertEqual(result, expected_result) self.assertEqual(stub.method_calls, [( 'GetInstance', (request_pb,), {}, )]) # Check Instance optional config values before. self.assertEqual(instance.display_name, DISPLAY_NAME)