module ActionDispatch::Assertions

def html_document
  @html_document ||= if @response.content_type.to_s =~ /xml\z/
    Nokogiri::XML::Document.parse(@response.body)
  else
    Nokogiri::HTML::Document.parse(@response.body)
  end
end