Add a data query test