Replaced some full sorts with simple linear search.

Refactored old code to nicer collection methods.

lib/webrat/core/form.rb

@@ -29,27 +29,17 @@ module Webrat
     end
 
     def find_button(value = nil)
-      return fields_by_type([ButtonField]).first if value.nil?
+      return fields_by_type([ButtonField]).first if value.nil?      
-      
+      possible_buttons = fields_by_type([ButtonField])      
-      possible_buttons = fields_by_type([ButtonField])
+      possible_buttons.detect { |possible_button| possible_button.matches_value?(value) }
-      
-      possible_buttons.each do |possible_button|
-        return possible_button if possible_button.matches_value?(value)
-      end
-      
-      nil
     end
 
     def fields
       return @fields if @fields
       
-      @fields = []
+      @fields = (@element / "button, input, textarea, select").collect do |field_element|
-      
+        Field.class_for_element(field_element).new(self, field_element)
-      (@element / "button, input, textarea, select").each do |field_element|
-        @fields << Field.class_for_element(field_element).new(self, field_element)
       end
-      
-      @fields
     end
     
     def submit
@@ -59,29 +49,18 @@ module Webrat
   protected
   
     def find_field_by_id(possible_fields, id)
-      possible_fields.each do |possible_field|
+      possible_fields.detect { |possible_field| possible_field.matches_id?(id) }
-        return possible_field if possible_field.matches_id?(id)
-      end
-      
-      nil
     end
     
     def find_field_by_name(possible_fields, name)
-      possible_fields.each do |possible_field|
+      possible_fields.detect { |possible_field| possible_field.matches_name?(name) }
-        return possible_field if possible_field.matches_name?(name)
-      end
-      
-      nil
     end
     
     def find_field_by_label(possible_fields, label)      
-      matching_fields = []
+      matching_fields = possible_fields.select do |possible_field|
-      
+        possible_field.matches_label?(label)
-      possible_fields.each do |possible_field|
+      end      
-        matching_fields << possible_field if possible_field.matches_label?(label)
+      matching_fields.min { |a, b| a.label_text.length <=> b.label_text.length }
-      end
-      
-      matching_fields.sort_by { |f| f.label_text.length }.first
     end
   
     def fields_by_type(field_types)

lib/webrat/core/scope.rb

@@ -202,14 +202,12 @@ module Webrat
     end
     
     def find_link(text, selector = nil)
-      matching_links = []
+      matching_links = links_within(selector).select do |possible_link|
-      
+        possible_link.matches_text?(text)
-      links_within(selector).each do |possible_link|
-        matching_links << possible_link if possible_link.matches_text?(text)
       end
       
       if matching_links.any?
-        matching_links.sort_by { |l| l.text.length }.first
+        matching_links.min { |a, b| a.text.length <=> b.text.length }
       else
         flunk("Could not find link with text #{text.inspect}")
       end