Moved html_document to ActionDispatch::Assertions. Included the Rails…

…::Dom::Testing::Assertions there as well.
kaspth · Jun 16, 2014 · 9efdffe · 9efdffe
1 parent 50347b1
commit 9efdffe
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/actionpack/lib/action_controller/test_case.rb b/actionpack/lib/action_controller/test_case.rb
@@ -684,14 +684,9 @@ def build_response(klass)
       private
 
       def document_root_element
-        @html_document ||= if @response.content_type =~ /xml$/
-          Loofah.xml_document(@response.body)
-        else
-          Loofah.document(@response.body)
-        end.root
+        html_document.root
       end
 
-
       def check_required_ivars
         # Sanity check for required instance variables so we can give an
         # understandable error message.

diff --git a/actionpack/lib/action_dispatch/testing/assertions.rb b/actionpack/lib/action_dispatch/testing/assertions.rb
@@ -1,3 +1,6 @@
+require 'loofah'
+require 'rails-dom-testing'
+
 module ActionDispatch
   module Assertions
     autoload :ResponseAssertions, 'action_dispatch/testing/assertions/response'
@@ -7,6 +10,14 @@ module Assertions
 
     include ResponseAssertions
     include RoutingAssertions
+    include Rails::Dom::Testing::Assertions
+
+    def html_document
+      @html_document ||= if @response.content_type =~ /xml$/
+        Loofah.xml_document(@response.body)
+      else
+        Loofah.document(@response.body)
+      end
+    end
   end
 end
-