Publicize document_root_element.

Author: kaspth

lib/rails/dom/testing/assertions/selector_assertions.rb

@@ -258,14 +258,14 @@ def assert_select_email(&block)
           end
         end
 
+        def document_root_element
+          raise NotImplementedError, 'Implementing document_root_element makes ' \
+            'assert_select work without needing to specify an element to select from.'
+        end
+
         private
           include CountDescripable
 
-          def document_root_element
-            raise NotImplementedError, 'Implementing document_root_element makes ' \
-              'assert_select work without needing to specify an element to select from.'
-          end
-
           # +equals+ must contain :minimum, :maximum and :count keys
           def assert_size_match!(size, equals, css_selector, message = nil)
             min, max, count = equals[:minimum], equals[:maximum], equals[:count]

lib/rails/dom/testing/assertions/selector_assertions/html_selector.rb

@@ -63,7 +63,7 @@ def extract_root(rootable, previous_selection)
     elsif previous_selection
       previous_selection
     else
-      rootable.send :document_root_element
+      rootable.document_root_element
     end
   end
 

test/selector_assertions_test.rb

@@ -284,6 +284,10 @@ def test_body_not_present_in_empty_document
     assert_select 'body', 0
   end
 
+  def document_root_element
+    @html_document
+  end
+
   protected
     def render_html(html)
       fake_render(:html, html)
@@ -300,8 +304,4 @@ def fake_render(content_type, content)
         Nokogiri::HTML::DocumentFragment.parse(content)
       end
     end
-
-    def document_root_element
-      @html_document
-    end
 end