@Test
public void testReadFlatRows() {

  Query query = Query.create(TABLE_ID).range("start", "end");
  when(client.readFlatRows(query.toProto(REQUEST_CONTEXT))).thenReturn(mockFlatRow);
  clientWrapper.readFlatRows(query);
  verify(client).readFlatRows(query.toProto(REQUEST_CONTEXT));
}