1 files changed, 12 insertions, 1 deletions
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
-