New test class for query