require 'loofah'
require 'rails-dom-testing'

module ActionDispatch
  module Assertions
    autoload :ResponseAssertions, 'action_dispatch/testing/assertions/response'
    autoload :RoutingAssertions, 'action_dispatch/testing/assertions/routing'

    extend ActiveSupport::Concern

    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