Add query API test