2010-04-30 10:46:30 -04:00
|
|
|
require 'abstract_unit'
|
|
|
|
|
|
|
|
class HttpTokenAuthenticationTest < ActionController::TestCase
|
|
|
|
class DummyController < ActionController::Base
|
2012-12-07 14:46:06 -05:00
|
|
|
before_action :authenticate, only: :index
|
|
|
|
before_action :authenticate_with_request, only: :display
|
|
|
|
before_action :authenticate_long_credentials, only: :show
|
2010-04-30 10:46:30 -04:00
|
|
|
|
|
|
|
def index
|
|
|
|
render :text => "Hello Secret"
|
|
|
|
end
|
|
|
|
|
|
|
|
def display
|
|
|
|
render :text => 'Definitely Maybe'
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2010-04-30 10:46:30 -04:00
|
|
|
def show
|
|
|
|
render :text => 'Only for loooooong credentials'
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
def authenticate
|
2014-01-24 20:25:15 -05:00
|
|
|
authenticate_or_request_with_http_token do |token, _|
|
2010-04-30 10:46:30 -04:00
|
|
|
token == 'lifo'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def authenticate_with_request
|
|
|
|
if authenticate_with_http_token { |token, options| token == '"quote" pretty' && options[:algorithm] == 'test' }
|
|
|
|
@logged_in = true
|
|
|
|
else
|
2015-05-01 14:28:00 -04:00
|
|
|
request_http_token_authentication("SuperSecret", "Authentication Failed\n")
|
2010-04-30 10:46:30 -04:00
|
|
|
end
|
|
|
|
end
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2010-04-30 10:46:30 -04:00
|
|
|
def authenticate_long_credentials
|
|
|
|
authenticate_or_request_with_http_token do |token, options|
|
|
|
|
token == '1234567890123456789012345678901234567890' && options[:algorithm] == 'test'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
AUTH_HEADERS = ['HTTP_AUTHORIZATION', 'X-HTTP_AUTHORIZATION', 'X_HTTP_AUTHORIZATION', 'REDIRECT_X_HTTP_AUTHORIZATION']
|
|
|
|
|
|
|
|
tests DummyController
|
|
|
|
|
|
|
|
AUTH_HEADERS.each do |header|
|
|
|
|
test "successful authentication with #{header.downcase}" do
|
|
|
|
@request.env[header] = encode_credentials('lifo')
|
|
|
|
get :index
|
|
|
|
|
|
|
|
assert_response :success
|
|
|
|
assert_equal 'Hello Secret', @response.body, "Authentication failed for request header #{header}"
|
|
|
|
end
|
|
|
|
test "successful authentication with #{header.downcase} and long credentials" do
|
|
|
|
@request.env[header] = encode_credentials('1234567890123456789012345678901234567890', :algorithm => 'test')
|
|
|
|
get :show
|
2010-08-14 01:13:00 -04:00
|
|
|
|
2010-04-30 10:46:30 -04:00
|
|
|
assert_response :success
|
|
|
|
assert_equal 'Only for loooooong credentials', @response.body, "Authentication failed for request header #{header} and long credentials"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
AUTH_HEADERS.each do |header|
|
|
|
|
test "unsuccessful authentication with #{header.downcase}" do
|
|
|
|
@request.env[header] = encode_credentials('h4x0r')
|
|
|
|
get :index
|
|
|
|
|
|
|
|
assert_response :unauthorized
|
|
|
|
assert_equal "HTTP Token: Access denied.\n", @response.body, "Authentication didn't fail for request header #{header}"
|
|
|
|
end
|
|
|
|
test "unsuccessful authentication with #{header.downcase} and long credentials" do
|
|
|
|
@request.env[header] = encode_credentials('h4x0rh4x0rh4x0rh4x0rh4x0rh4x0rh4x0rh4x0r')
|
|
|
|
get :show
|
|
|
|
|
|
|
|
assert_response :unauthorized
|
|
|
|
assert_equal "HTTP Token: Access denied.\n", @response.body, "Authentication didn't fail for request header #{header} and long credentials"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2012-07-10 19:56:38 -04:00
|
|
|
test "authentication request with badly formatted header" do
|
2015-02-26 10:01:08 -05:00
|
|
|
@request.env['HTTP_AUTHORIZATION'] = 'Token token$"lifo"'
|
2012-07-10 20:07:25 -04:00
|
|
|
get :index
|
2012-07-10 19:56:38 -04:00
|
|
|
|
2012-07-10 20:07:25 -04:00
|
|
|
assert_response :unauthorized
|
|
|
|
assert_equal "HTTP Token: Access denied.\n", @response.body, "Authentication header was not properly parsed"
|
2012-07-10 19:56:38 -04:00
|
|
|
end
|
|
|
|
|
2015-02-26 10:01:26 -05:00
|
|
|
test "successful authentication request with Bearer instead of Token" do
|
|
|
|
@request.env['HTTP_AUTHORIZATION'] = 'Bearer lifo'
|
|
|
|
get :index
|
|
|
|
|
|
|
|
assert_response :success
|
|
|
|
end
|
|
|
|
|
2010-04-30 10:46:30 -04:00
|
|
|
test "authentication request without credential" do
|
|
|
|
get :display
|
|
|
|
|
|
|
|
assert_response :unauthorized
|
2015-05-01 14:28:00 -04:00
|
|
|
assert_equal "Authentication Failed\n", @response.body
|
2010-04-30 10:46:30 -04:00
|
|
|
assert_equal 'Token realm="SuperSecret"', @response.headers['WWW-Authenticate']
|
|
|
|
end
|
|
|
|
|
|
|
|
test "authentication request with invalid credential" do
|
|
|
|
@request.env['HTTP_AUTHORIZATION'] = encode_credentials('"quote" pretty')
|
|
|
|
get :display
|
|
|
|
|
|
|
|
assert_response :unauthorized
|
2015-05-01 14:28:00 -04:00
|
|
|
assert_equal "Authentication Failed\n", @response.body
|
2010-04-30 10:46:30 -04:00
|
|
|
assert_equal 'Token realm="SuperSecret"', @response.headers['WWW-Authenticate']
|
|
|
|
end
|
|
|
|
|
2012-10-07 18:36:01 -04:00
|
|
|
test "token_and_options returns correct token" do
|
|
|
|
token = "rcHu+HzSFw89Ypyhn/896A=="
|
|
|
|
actual = ActionController::HttpAuthentication::Token.token_and_options(sample_request(token)).first
|
|
|
|
expected = token
|
|
|
|
assert_equal(expected, actual)
|
|
|
|
end
|
|
|
|
|
2012-12-28 14:39:29 -05:00
|
|
|
test "token_and_options returns correct token with value after the equal sign" do
|
2012-10-07 18:36:01 -04:00
|
|
|
token = 'rcHu+=HzSFw89Ypyhn/896A==f34'
|
|
|
|
actual = ActionController::HttpAuthentication::Token.token_and_options(sample_request(token)).first
|
|
|
|
expected = token
|
|
|
|
assert_equal(expected, actual)
|
|
|
|
end
|
2010-04-30 10:46:30 -04:00
|
|
|
|
2012-12-28 14:39:29 -05:00
|
|
|
test "token_and_options returns correct token with slashes" do
|
2012-10-07 18:36:01 -04:00
|
|
|
token = 'rcHu+\\\\"/896A'
|
|
|
|
actual = ActionController::HttpAuthentication::Token.token_and_options(sample_request(token)).first
|
|
|
|
expected = token
|
|
|
|
assert_equal(expected, actual)
|
|
|
|
end
|
|
|
|
|
2012-12-28 14:39:29 -05:00
|
|
|
test "token_and_options returns correct token with quotes" do
|
2012-10-07 18:36:01 -04:00
|
|
|
token = '\"quote\" pretty'
|
|
|
|
actual = ActionController::HttpAuthentication::Token.token_and_options(sample_request(token)).first
|
|
|
|
expected = token
|
|
|
|
assert_equal(expected, actual)
|
2010-04-30 10:46:30 -04:00
|
|
|
end
|
|
|
|
|
2014-06-13 02:19:53 -04:00
|
|
|
test "token_and_options returns empty string with empty token" do
|
|
|
|
token = ''
|
|
|
|
actual = ActionController::HttpAuthentication::Token.token_and_options(sample_request(token)).first
|
|
|
|
expected = token
|
|
|
|
assert_equal(expected, actual)
|
2012-10-07 18:36:01 -04:00
|
|
|
end
|
|
|
|
|
2014-06-11 03:11:27 -04:00
|
|
|
test "token_and_options returns correct token with nounce option" do
|
|
|
|
token = "rcHu+HzSFw89Ypyhn/896A="
|
|
|
|
nonce_hash = {nonce: "123abc"}
|
|
|
|
actual = ActionController::HttpAuthentication::Token.token_and_options(sample_request(token, nonce_hash))
|
|
|
|
expected_token = token
|
|
|
|
expected_nonce = {"nonce" => nonce_hash[:nonce]}
|
|
|
|
assert_equal(expected_token, actual.first)
|
|
|
|
assert_equal(expected_nonce, actual.last)
|
|
|
|
end
|
|
|
|
|
2014-06-13 02:19:53 -04:00
|
|
|
test "token_and_options returns nil with no value after the equal sign" do
|
|
|
|
actual = ActionController::HttpAuthentication::Token.token_and_options(malformed_request).first
|
|
|
|
expected = nil
|
|
|
|
assert_equal(expected, actual)
|
2010-04-30 10:46:30 -04:00
|
|
|
end
|
2014-06-13 02:19:53 -04:00
|
|
|
|
2014-06-11 03:11:27 -04:00
|
|
|
test "raw_params returns a tuple of two key value pair strings" do
|
|
|
|
auth = sample_request("rcHu+HzSFw89Ypyhn/896A=").authorization.to_s
|
|
|
|
actual = ActionController::HttpAuthentication::Token.raw_params(auth)
|
|
|
|
expected = ["token=\"rcHu+HzSFw89Ypyhn/896A=\"", "nonce=\"def\""]
|
|
|
|
assert_equal(expected, actual)
|
|
|
|
end
|
|
|
|
|
2014-10-06 00:33:07 -04:00
|
|
|
test "token_and_options returns right token when token key is not specified in header" do
|
|
|
|
token = "rcHu+HzSFw89Ypyhn/896A="
|
|
|
|
|
|
|
|
actual = ActionController::HttpAuthentication::Token.token_and_options(
|
|
|
|
sample_request_without_token_key(token)
|
|
|
|
).first
|
|
|
|
|
|
|
|
expected = token
|
|
|
|
assert_equal(expected, actual)
|
|
|
|
end
|
|
|
|
|
2014-06-13 02:19:53 -04:00
|
|
|
private
|
|
|
|
|
2014-06-11 03:11:27 -04:00
|
|
|
def sample_request(token, options = {nonce: "def"})
|
|
|
|
authorization = options.inject([%{Token token="#{token}"}]) do |arr, (k, v)|
|
|
|
|
arr << "#{k}=\"#{v}\""
|
|
|
|
end.join(", ")
|
2014-10-06 00:33:07 -04:00
|
|
|
mock_authorization_request(authorization)
|
2014-06-13 02:19:53 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def malformed_request
|
2014-10-06 00:33:07 -04:00
|
|
|
mock_authorization_request(%{Token token=})
|
|
|
|
end
|
|
|
|
|
|
|
|
def sample_request_without_token_key(token)
|
|
|
|
mock_authorization_request(%{Token #{token}})
|
|
|
|
end
|
|
|
|
|
|
|
|
def mock_authorization_request(authorization)
|
|
|
|
OpenStruct.new(authorization: authorization)
|
2014-06-13 02:19:53 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def encode_credentials(token, options = {})
|
|
|
|
ActionController::HttpAuthentication::Token.encode_credentials(token, options)
|
|
|
|
end
|
2010-04-30 10:46:30 -04:00
|
|
|
end
|