module ActionDispatch::Assertions

def html_document
  @html_document ||= if @response.content_type === Mime::XML
    Nokogiri::XML::Document.parse(@response.body)
  else
    Nokogiri::HTML::Document.parse(@response.body)
  end
end