@Test
public void testReadFlatRowsAsyncWithStream() {

  StreamObserver<FlatRow> streamOb = mock(StreamObserver.class);
  when(dataClientV2.readRowsCallable(any(FlatRowAdapter.class))).thenReturn(serverStreaming);
  doNothing().when(serverStreaming).call(any(Query.class), any(ResponseObserver.class));
  dataGCJClient.readFlatRowsAsync(request, streamOb);
  streamOb.onNext(flatRows.get(0));
  verify(dataClientV2).readRowsCallable(any(FlatRowAdapter.class));
  verify(serverStreaming).call(any(Query.class), any(ResponseObserver.class));
}