Sort nodeset on demand

Author: tompng

lib/rexml/functions.rb

@@ -87,7 +87,7 @@ def get_namespace( node_set = nil )
       else
         if node_set.respond_to? :each
           result = []
-          node_set.each do |node|
+          XPathParser.sort(node_set.to_a).each do |node|
             result << yield(node) if node.respond_to?(:namespace)
           end
           result
@@ -149,7 +149,7 @@ def string( object=@context[:node] )
       else
         case object
         when Array
-          string(object[0])
+          string(XPathParser.sort(object)[0])
         when Float
           if object.nan?
             "NaN"

lib/rexml/xpath_parser.rb

@@ -156,7 +156,7 @@ def match(path_stack, node)
       result = expr(path_stack, nodeset)
       case result
       when Array # nodeset
-        result.uniq
+        XPathParser.sort(result)
       else
         [result]
       end
@@ -199,7 +199,7 @@ def expr( path_stack, nodeset, context=nil )
           nodeset = [nodeset.first.root_node]
         when :self
           nodeset = step(path_stack) do
-            [:iterate_nodesets, [nodeset]]
+            [:iterate_nodesets, [XPathParser.sort(nodeset)]]
           end
         when :child
           nodeset = step(path_stack) do
@@ -323,7 +323,7 @@ def expr( path_stack, nodeset, context=nil )
             # If result is a nodeset, apply following predicates
             path_stack.unshift(:node)
             nodeset = step(path_stack) do
-              [:iterate_nodesets, [result]]
+              [:iterate_nodesets, [XPathParser.sort(result)]]
             end
           else
             return result
@@ -621,7 +621,7 @@ def step(path_stack, any_type: :element)
             nodes << node
           end
         end
-        new_nodeset = sort(nodes.to_a)
+        new_nodeset = nodes.to_a
       ensure
         leave(:step, path_stack, new_nodeset) if @debug
       end
@@ -761,7 +761,7 @@ def leave(tag, *args)
     # in and out of function calls.  If I knew what the index of the nodes was,
     # I wouldn't have to do this.  Maybe add a document IDX for each node?
     # Problems with mutable documents.  Or, rewrite everything.
-    def sort(array_of_nodes)
+    def self.sort(array_of_nodes)
       return array_of_nodes if array_of_nodes.size <= 1
 
       new_arry = []

test/test_jaxen.rb

@@ -87,6 +87,13 @@ def process_value_of(context, variables, namespaces, value_of)
       xpath = value_of.attributes["select"]
       matched = XPath.match(context, xpath, namespaces, variables, strict: true)
 
+      # XPath.match can be a nodeset or a primitive value wrapped in an array.
+      # We need to unwrap primitive value because Functions doesn't accept array which is not a nodeset.
+      unless matched.all? { |node| node.is_a?(REXML::Node) }
+        raise "[BUG] Primitive value should be a single value: #{matched.inspect}" if matched.size != 1
+        matched = matched.first
+      end
+
       message = user_message(context, xpath, matched)
       assert_equal(expected || "",
                    REXML::Functions.string(matched),