@Test
public void testReadRowsAsync() throws Exception {

  Query query = Query.create(TABLE_ID);
  FlatRow flatRow = FlatRow.newBuilder().withRowKey(ByteString.copyFromUtf8("key")).build();
  List<FlatRow> listFlatRows = Arrays.asList(flatRow);
  when(client.readFlatRowsAsync(query.toProto(REQUEST_CONTEXT)))
      .thenReturn(Futures.immediateFuture(listFlatRows));
  List<com.google.cloud.bigtable.data.v2.models.Row> actualResult =
      clientWrapper.readRowsAsync(query).get();
  assertEquals(Arrays.asList(FlatRowConverter.convertToModelRow(flatRow)), actualResult);
  verify(client).readFlatRowsAsync(query.toProto(REQUEST_CONTEXT));
}