Test coverage in MockQueryHelper