@Test
public void testReadRowsAsync() throws Exception {

  Query query = Query.create(TABLE_ID);
  FlatRow anotherFlatRow = FlatRow.newBuilder().withRowKey(ROW_KEY).build();
  List<FlatRow> listFlatRows = ImmutableList.of(SAMPLE_FLAT_ROW, anotherFlatRow);
  when(delegate.readFlatRowsAsync(query.toProto(REQUEST_CONTEXT)))
      .thenReturn(Futures.immediateFuture(listFlatRows));
  List<Result> actualResult = dataClientWrapper.readRowsAsync(query).get();
  assertEquals(listFlatRows.size(), actualResult.size());
  assertArrayEquals(
      FLAT_ROW_ADAPTER.adaptResponse(SAMPLE_FLAT_ROW).rawCells(), actualResult.get(0).rawCells());
  assertArrayEquals(
      FLAT_ROW_ADAPTER.adaptResponse(anotherFlatRow).rawCells(), actualResult.get(1).rawCells());
  verify(delegate).readFlatRowsAsync(query.toProto(REQUEST_CONTEXT));
}