@Test
public void shouldCallOnErrorIfEntityDatasetIsNotSet() {

  service = getServingServiceGRpcController(false);
  GetOnlineFeaturesRequest missingEntityName =
      GetOnlineFeaturesRequest.newBuilder(validRequest).clearEntityRows().build();
  service.getOnlineFeatures(missingEntityName, mockStreamObserver);
  Mockito.verify(mockStreamObserver).onError(Mockito.any(StatusRuntimeException.class));
}