@Test
public void testReadFlatRowsList() {

  Query query = Query.create(TABLE_ID);
  FlatRow flatRow = FlatRow.newBuilder().withRowKey(ByteString.copyFromUtf8("key")).build();
  List<FlatRow> listFlatRows = Arrays.asList(flatRow);
  when(client.readFlatRowsList(query.toProto(REQUEST_CONTEXT))).thenReturn(listFlatRows);
  clientWrapper.readFlatRowsList(query);
  verify(client).readFlatRowsList(query.toProto(REQUEST_CONTEXT));
}