@Test
public void testReadRows() throws Exception {

  Query query = Query.create(TABLE_ID);
  FlatRow row = FlatRow.newBuilder().withRowKey(ByteString.copyFromUtf8("key")).build();
  when(client.readFlatRows(query.toProto(REQUEST_CONTEXT))).thenReturn(mockFlatRow);
  when(mockFlatRow.next()).thenReturn(row);
  ResultScanner<com.google.cloud.bigtable.data.v2.models.Row> actualResult =
      clientWrapper.readRows(query);
  assertEquals(FlatRowConverter.convertToModelRow(row), actualResult.next());
  verify(mockFlatRow).next();
  verify(client).readFlatRows(query.toProto(REQUEST_CONTEXT));
}