test/lib/puppettest/support/collection.rb in puppet-0.24.9 vs test/lib/puppettest/support/collection.rb in puppet-0.25.0

@@ -16,10 +16,10 @@
                 end
                 parser = mkparser
                 query = nil
 
                 assert_nothing_raised("Could not parse '#{str}'") do
-                    query = parser.parse(code).classes[""].code[0].query
+                    query = parser.parse(code).hostclass("").code[0].query
                 end
 
                 yield str, res, query
             end
         end