10d483eaaa
* fix-u2f-fixtures: generate u2f frontend fixtures dynamically force utf-8 encoding for prior to fixture parsing to prevent nokogiri issues exclude script type=text/template from dom scrubbing
63 lines
1.8 KiB
Ruby
63 lines
1.8 KiB
Ruby
require 'action_dispatch/testing/test_request'
|
|
require 'fileutils'
|
|
require 'gitlab/popen'
|
|
|
|
module JavaScriptFixturesHelpers
|
|
include Gitlab::Popen
|
|
|
|
FIXTURE_PATH = 'spec/javascripts/fixtures'
|
|
|
|
# Public: Removes all fixture files from given directory
|
|
#
|
|
# directory_name - directory of the fixtures (relative to FIXTURE_PATH)
|
|
#
|
|
def clean_frontend_fixtures(directory_name)
|
|
directory_name = File.expand_path(directory_name, FIXTURE_PATH)
|
|
Dir[File.expand_path('*.html.raw', directory_name)].each do |file_name|
|
|
FileUtils.rm(file_name)
|
|
end
|
|
end
|
|
|
|
# Public: Store a response object as fixture file
|
|
#
|
|
# response - string or response object to store
|
|
# fixture_file_name - file name to store the fixture in (relative to FIXTURE_PATH)
|
|
#
|
|
def store_frontend_fixture(response, fixture_file_name)
|
|
fixture_file_name = File.expand_path(fixture_file_name, FIXTURE_PATH)
|
|
fixture = response.respond_to?(:body) ? parse_response(response) : response
|
|
|
|
FileUtils.mkdir_p(File.dirname(fixture_file_name))
|
|
File.write(fixture_file_name, fixture)
|
|
end
|
|
|
|
private
|
|
|
|
# Private: Prepare a response object for use as a frontend fixture
|
|
#
|
|
# response - response object to prepare
|
|
#
|
|
def parse_response(response)
|
|
fixture = response.body
|
|
fixture.force_encoding("utf-8")
|
|
|
|
response_mime_type = Mime::Type.lookup(response.content_type)
|
|
if response_mime_type.html?
|
|
doc = Nokogiri::HTML::DocumentFragment.parse(fixture)
|
|
|
|
link_tags = doc.css('link')
|
|
link_tags.remove
|
|
|
|
scripts = doc.css("script:not([type='text/template'])")
|
|
scripts.remove
|
|
|
|
fixture = doc.to_html
|
|
|
|
# replace relative links
|
|
test_host = ActionDispatch::TestRequest::DEFAULT_ENV['HTTP_HOST']
|
|
fixture.gsub!(%r{="/}, "=\"http://#{test_host}/")
|
|
end
|
|
|
|
fixture
|
|
end
|
|
end
|