2 files changed, 13 insertions, 7 deletions
diff --git a/actionpack/lib/action_controller/test_case.rb b/actionpack/lib/action_controller/test_case.rb
index 98ed1c2382..c307702f5d 100644
--- a/actionpack/lib/action_controller/test_case.rb
+++ b/actionpack/lib/action_controller/test_case.rb
@@ -684,14 +684,9 @@ module ActionController
       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
index 5f923abbf9..44815a75a8 100644
--- 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 ActionDispatch
 
     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
-