mavenlink · February 4, 2010 07:21
diff --git a/fixture_generation_methods.rb b/fixture_generation_methods.rb
 # Add fixture-generation methods to ControllerExampleGroup. We load
 # this file within our spec_helper.rb

 Spec::Rails::Example::ControllerExampleGroup.class_eval do

  # Saves the markup to a fixture file using the given name
  def save_fixture(markup, name)
    fixture_path = File.join(RAILS_ROOT, '/tmp/js_dom_fixtures')
    Dir.mkdir(fixture_path) unless File.exists?(fixture_path)

    fixture_file = File.join(fixture_path, "#{name}.fixture.html.erb")
    File.open(fixture_file, 'w') do |file|
      file.puts(markup)
    end
  end

  # From the controller spec response body, extracts html identified
  # by the css selector.
  def html_for(selector)
    doc = Nokogiri::HTML(response.body)

    remove_third_party_scripts(doc)
    content = doc.css(selector).first.to_s

    return convert_body_tag_to_div(content)
  end

  # Remove scripts such as Google Analytics to avoid running them
  # when we load into the dom during js specs.
  def remove_third_party_scripts(doc)
    scripts = doc.at('#third-party-scripts')
    scripts.remove if scripts
  end

  # Many of our css and jQuery selectors rely on a class attribute we
  # normally embed in the <body>. For example:
  #
  # <body class="workspaces show">
  #
  # Here we convert the body tag to a div so that we can load it into
  # the document running js specs without embedding a <body> within a <body>.
  def convert_body_tag_to_div(markup)
    return markup.gsub("<body", '<div').gsub("</body>", "</div>")
  end
 end
	# Add fixture-generation methods to ControllerExampleGroup. We load
	# this file within our spec_helper.rb

	Spec::Rails::Example::ControllerExampleGroup.class_eval do

	# Saves the markup to a fixture file using the given name
	def save_fixture(markup, name)
	fixture_path = File.join(RAILS_ROOT, '/tmp/js_dom_fixtures')
	Dir.mkdir(fixture_path) unless File.exists?(fixture_path)

	fixture_file = File.join(fixture_path, "#{name}.fixture.html.erb")
	File.open(fixture_file, 'w') do \|file\|
	file.puts(markup)
	end
	end

	# From the controller spec response body, extracts html identified
	# by the css selector.
	def html_for(selector)
	doc = Nokogiri::HTML(response.body)

	remove_third_party_scripts(doc)
	content = doc.css(selector).first.to_s

	return convert_body_tag_to_div(content)
	end

	# Remove scripts such as Google Analytics to avoid running them
	# when we load into the dom during js specs.
	def remove_third_party_scripts(doc)
	scripts = doc.at('#third-party-scripts')
	scripts.remove if scripts
	end

	# Many of our css and jQuery selectors rely on a class attribute we
	# normally embed in the <body>. For example:
	#
	# <body class="workspaces show">
	#
	# Here we convert the body tag to a div so that we can load it into
	# the document running js specs without embedding a <body> within a <body>.
	def convert_body_tag_to_div(markup)
	return markup.gsub("<body", '<div').gsub("</body>", "</div>")
	end
	end