module ActionController::TestResponseBehavior

def html(flavor=nil)
  if flavor == :hpricot
    @_nokogiri_html_hpricot ||= Nokogiri::Hpricot(body)
  else
    @_nokogiri_html_vanilla ||= Nokogiri::HTML(body)
  end
end

def xml
  @_nokogiri_xml ||= Nokogiri::XML(body)
end