Skip to content

Instantly share code, notes, and snippets.

@mrchrisadams
Forked from vigetlabs/http_redirect_test.rb
Created May 21, 2010 08:58
Show Gist options
  • Save mrchrisadams/408636 to your computer and use it in GitHub Desktop.
Save mrchrisadams/408636 to your computer and use it in GitHub Desktop.
require 'resource_path'
require 'redirect_check'
require 'test/unit'
class HTTPRedirectTest < Test::Unit::TestCase
def default_test; end # placeholder to stop Test::Unit from complaining
def self.domain=(domain)
RedirectCheck.domain = domain
end
def self.should_not_redirect(path)
class_eval <<-CODE
def test_#{name_for(path)}_should_not_redirect
check = RedirectCheck.new('#{path}')
assert_equal false, check.redirected?, "#{path} is redirecting"
assert_equal true, check.success?, "#{path} is not a success response"
end
CODE
end
def self.name_for(path)
name = path.gsub(/\W+/, '_')
name.gsub!(/^_+/, '')
name.gsub!(/_+$/, '')
name = 'root' if name == ''
name
end
def self.should_redirect(source, options)
source_path = ResourcePath.new(source, :param => 'subdir').to_s
destination_path = ResourcePath.new(options[:to], :param => 'subdir').to_s
permanent = options.fetch(:permanent, true)
class_eval <<-CODE
def test_#{name_for(source_path)}_should_redirect_to_#{name_for(destination_path)}
redirection = RedirectCheck.new('#{source_path}', '#{destination_path}')
assert_equal true, redirection.redirected?, "'#{source_path}' is not redirecting"
assert_equal '#{destination_path}', redirection.redirected_path,
"'#{source_path}' is not redirecting to '#{destination_path}'"
if #{permanent}
assert_equal true, redirection.permanent_redirect?,
"The redirection from '#{source_path}' to '#{destination_path}' doesn't appear to be a permanent redirect"
end
end
CODE
end
end
require 'uri'
require 'net/http'
class RedirectCheck
def self.domain=(domain)
@domain = domain
end
def self.domain
@domain
end
attr_reader :source_path, :destination_path
def initialize(source_path, destination_path = nil)
@source_path = source_path.to_s
@destination_path = destination_path.to_s
end
def uri
URI.parse("http://#{self.class.domain}#{source_path}")
end
def response
@response ||= Net::HTTP.start(uri.host, uri.port) {|http| http.head(uri.path) }
end
def success?
response.is_a?(Net::HTTPOK)
end
def redirected?
response.is_a?(Net::HTTPRedirection)
end
def permanent_redirect?
redirected? && response.is_a?(Net::HTTPMovedPermanently)
end
def redirected_path
response['location'].sub(/#{Regexp.escape("#{uri.scheme}://#{uri.host}")}/, '') if redirected?
end
end
class ResourcePath
attr_writer :param
def initialize(path, options = {})
@path = path
@param = options[:param]
end
def param
@param ||= (0...8).map{65.+(rand(25)).chr}.join
end
def to_s
@path.gsub('*', param)
end
end
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment