-
-
Save mrchrisadams/408636 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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