Namespace locators in Webrat::Locators

lib/webrat/core/locators/area_locator.rb

@@ -1,24 +1,30 @@
 require "webrat/core/locators/locator"
 
-class AreaLocator < Locator
+module Webrat
+  module Locators
     
-  def locate
+    class AreaLocator < Locator
-    @scope.area_by_element(area_element)
+  
-  end
+      def locate
+        @scope.area_by_element(area_element)
+      end
+  
+      def area_element
+        area_elements.detect do |area_element|
+          Webrat::XML.attribute(area_element, "title") =~ matcher ||
+          Webrat::XML.attribute(area_element, "id") =~ matcher
+        end
+      end
+  
+      def matcher
+        /#{Regexp.escape(@value.to_s)}/i
+      end
+  
+      def area_elements
+        Webrat::XML.css_search(@scope.dom, "area")
+      end
   
-  def area_element
-    area_elements.detect do |area_element|
-      Webrat::XML.attribute(area_element, "title") =~ matcher ||
-      Webrat::XML.attribute(area_element, "id") =~ matcher
     end
-  end
     
-  def matcher
-    /#{Regexp.escape(@value.to_s)}/i
   end
-  
-  def area_elements
-    Webrat::XML.css_search(@scope.dom, "area")
-  end
-  
 end

lib/webrat/core/locators/button_locator.rb

@@ -1,40 +1,46 @@
 require "webrat/core/locators/locator"
 
-class ButtonLocator < Locator
+module Webrat
+  module Locators
     
-  def locate
+    class ButtonLocator < Locator
-    @scope.field_by_element(button_element)
+  
-  end
+      def locate
+        @scope.field_by_element(button_element)
+      end
+  
+      def button_element
+        button_elements.detect do |element|
+          @value.nil?             ||
+          matches_id?(element)    ||
+          matches_value?(element) ||
+          matches_html?(element)  ||
+          matches_alt?(element)
+        end
+      end
+  
+      def matches_id?(element)
+        (@value.is_a?(Regexp) && Webrat::XML.attribute(element, "id") =~ @value) ||
+        (!@value.is_a?(Regexp) && Webrat::XML.attribute(element, "id") == @value.to_s)
+      end
+  
+      def matches_value?(element)
+        Webrat::XML.attribute(element, "value") =~ /^\W*#{Regexp.escape(@value.to_s)}/i
+      end
+  
+      def matches_html?(element)
+        Webrat::XML.inner_html(element) =~ /#{Regexp.escape(@value.to_s)}/i
+      end
+  
+      def matches_alt?(element)
+        Webrat::XML.attribute(element, "alt") =~ /^\W*#{Regexp.escape(@value.to_s)}/i
+      end
+  
+      def button_elements
+        Webrat::XML.xpath_search(@scope.dom, *ButtonField.xpath_search)
+      end
   
-  def button_element
-    button_elements.detect do |element|
-      @value.nil?             ||
-      matches_id?(element)    ||
-      matches_value?(element) ||
-      matches_html?(element)  ||
-      matches_alt?(element)
     end
-  end
     
-  def matches_id?(element)
-    (@value.is_a?(Regexp) && Webrat::XML.attribute(element, "id") =~ @value) ||
-    (!@value.is_a?(Regexp) && Webrat::XML.attribute(element, "id") == @value.to_s)
   end
-  
-  def matches_value?(element)
-    Webrat::XML.attribute(element, "value") =~ /^\W*#{Regexp.escape(@value.to_s)}/i
-  end
-  
-  def matches_html?(element)
-    Webrat::XML.inner_html(element) =~ /#{Regexp.escape(@value.to_s)}/i
-  end
-  
-  def matches_alt?(element)
-    Webrat::XML.attribute(element, "alt") =~ /^\W*#{Regexp.escape(@value.to_s)}/i
-  end
-  
-  def button_elements
-    Webrat::XML.xpath_search(@scope.dom, *Webrat::ButtonField.xpath_search)
-  end
-  
 end

lib/webrat/core/locators/field_by_id_locator.rb

@@ -1,23 +1,29 @@
 require "webrat/core/locators/locator"
 
-class FieldByIdLocator < Locator
+module Webrat
+  module Locators
     
-  def locate
+    class FieldByIdLocator < Locator
-    @scope.field_by_element(field_element)
-  end
   
-  def field_element
+      def locate
-    field_elements.detect do |field_element|
+        @scope.field_by_element(field_element)
-      if @value.is_a?(Regexp)
-        Webrat::XML.attribute(field_element, "id") =~ @value
-      else
-        Webrat::XML.attribute(field_element, "id") == @value.to_s
       end
+  
+      def field_element
+        field_elements.detect do |field_element|
+          if @value.is_a?(Regexp)
+            Webrat::XML.attribute(field_element, "id") =~ @value
+          else
+            Webrat::XML.attribute(field_element, "id") == @value.to_s
+          end
+        end
+      end
+  
+      def field_elements
+        Webrat::XML.xpath_search(@scope.dom, *Field.xpath_search)
+      end
+  
     end
-  end
     
-  def field_elements
-    Webrat::XML.xpath_search(@scope.dom, *Webrat::Field.xpath_search)
   end
-  
 end

lib/webrat/core/locators/field_named_locator.rb

@@ -1,27 +1,33 @@
 require "webrat/core/locators/locator"
 
-class FieldNamedLocator < Locator
+module Webrat
+  module Locators
     
-  def locate
+    class FieldNamedLocator < Locator
-    @scope.field_by_element(field_element)
+  
-  end
+      def locate
+        @scope.field_by_element(field_element)
+      end
+  
+      def field_element
+        field_elements.detect do |field_element|
+          Webrat::XML.attribute(field_element, "name") == @value.to_s
+        end
+      end
+  
+      def field_elements
+        Webrat::XML.xpath_search(@scope.dom, *xpath_searches)
+      end
+  
+      def xpath_searches
+        if @field_types.any?
+          @field_types.map { |field_type| field_type.xpath_search }.flatten
+        else
+          Array(Field.xpath_search)
+        end
+      end
   
-  def field_element
-    field_elements.detect do |field_element|
-      Webrat::XML.attribute(field_element, "name") == @value.to_s
     end
-  end
     
-  def field_elements
-    Webrat::XML.xpath_search(@scope.dom, *xpath_searches)
   end
-  
-  def xpath_searches
-    if @field_types.any?
-      @field_types.map { |field_type| field_type.xpath_search }.flatten
-    else
-      Array(Field.xpath_search)
-    end
-  end
-  
 end

lib/webrat/core/locators/link_locator.rb

@@ -1,52 +1,58 @@
 require "webrat/core/locators/locator"
 
-class LinkLocator < Locator
+module Webrat
+  module Locators
     
-  def locate
+    class LinkLocator < Locator
-    @scope.link_by_element(link_element)
-  end
   
-  def link_element
+      def locate
-    matching_links.min { |a, b| Webrat::XML.inner_text(a).length <=> Webrat::XML.inner_text(b).length }
+        @scope.link_by_element(link_element)
-  end
+      end
   
-  def matching_links
+      def link_element
-    @matching_links ||= link_elements.select do |link_element|
+        matching_links.min { |a, b| Webrat::XML.inner_text(a).length <=> Webrat::XML.inner_text(b).length }
-      matches_text?(link_element) ||
+      end
-      matches_id?(link_element)
+  
-    end
+      def matching_links
-  end
+        @matching_links ||= link_elements.select do |link_element|
+          matches_text?(link_element) ||
+          matches_id?(link_element)
+        end
+      end
+  
+      def matches_text?(link)
+        if @value.is_a?(Regexp)
+          matcher = @value
+        else
+          matcher = /#{Regexp.escape(@value.to_s)}/i
+        end
+
+        replace_nbsp(Webrat::XML.inner_text(link)) =~ matcher ||
+        replace_nbsp_ref(Webrat::XML.inner_html(link)) =~ matcher ||
+        Webrat::XML.attribute(link, "title")=~ matcher
+      end
+
+      def matches_id?(link)
+        if @value.is_a?(Regexp)
+          (Webrat::XML.attribute(link, "id") =~ @value) ? true : false
+        else
+          (Webrat::XML.attribute(link, "id") == @value) ? true : false
+        end
+      end
+  
+      def link_elements
+        Webrat::XML.css_search(@scope.dom, *Link.css_search)
+      end
+  
+      def replace_nbsp(str)
+        str.gsub([0xA0].pack('U'), ' ')
+      end
+
+      def replace_nbsp_ref(str)
+        str.gsub('&#xA0;',' ').gsub('&nbsp;', ' ')
+      end
   
-  def matches_text?(link)
-    if @value.is_a?(Regexp)
-      matcher = @value
-    else
-      matcher = /#{Regexp.escape(@value.to_s)}/i
     end
     
-    replace_nbsp(Webrat::XML.inner_text(link)) =~ matcher ||
-    replace_nbsp_ref(Webrat::XML.inner_html(link)) =~ matcher ||
-    Webrat::XML.attribute(link, "title")=~ matcher
   end
-
-  def matches_id?(link)
-    if @value.is_a?(Regexp)
-      (Webrat::XML.attribute(link, "id") =~ @value) ? true : false
-    else
-      (Webrat::XML.attribute(link, "id") == @value) ? true : false
-    end
-  end
-  
-  def link_elements
-    Webrat::XML.css_search(@scope.dom, *Webrat::Link.css_search)
-  end
-  
-  def replace_nbsp(str)
-    str.gsub([0xA0].pack('U'), ' ')
-  end
-
-  def replace_nbsp_ref(str)
-    str.gsub('&#xA0;',' ').gsub('&nbsp;', ' ')
-  end
-  
 end

lib/webrat/core/locators/locator.rb

@@ -1,9 +1,15 @@
-class Locator
+module Webrat
+  module Locators
+  
+    class Locator
+  
+      def initialize(scope, value, *field_types)
+        @scope = scope
+        @value = value
+        @field_types = field_types
+      end
+  
+    end
     
-  def initialize(scope, value, *field_types)
-    @scope = scope
-    @value = value
-    @field_types = field_types
   end
-  
 end