Initialize Labels with a session refrence

brynary · Nov 30, 2008 · 40e0272 · 40e0272
1 parent 9e743ea
commit 40e0272
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/lib/webrat/core/field.rb b/lib/webrat/core/field.rb
@@ -96,7 +96,7 @@ def escaped_value
     end
 
     def labels
-      @labels ||= label_elements.map { |element| Label.new(self, element) }
+      @labels ||= label_elements.map { |element| Label.new(@session, element) }
     end
 
     def label_elements

diff --git a/lib/webrat/core/form.rb b/lib/webrat/core/form.rb
@@ -48,7 +48,7 @@ def fields
 
     def labels
       @labels ||= Webrat::XML.css_search(element, "label").map do |element|
-        label = Label.new(nil, element)
+        label = Label.new(@session, element)
         @session.elements[Webrat::XML.xpath_to(element)] = label
         label
       end

diff --git a/lib/webrat/core/label.rb b/lib/webrat/core/label.rb
@@ -7,8 +7,8 @@ def self.xpath_search
       ".//label"
     end
 
-    def initialize(field, element)
-      @field    = field
+    def initialize(session, element)
+      @session  = session
       @element  = element
     end