1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175
|
require 'openid/extensions/oauth'
require 'openid/message'
require 'openid/server'
require 'openid/consumer/responses'
require 'openid/consumer/discovery'
module OpenID
module OAuthTest
class OAuthRequestTestCase < Test::Unit::TestCase
def setup
@req = OAuth::Request.new
end
def test_construct
assert_nil(@req.consumer)
assert_nil(@req.scope)
assert_equal('oauth', @req.ns_alias)
req2 = OAuth::Request.new("CONSUMER","http://sample.com/some_scope")
assert_equal("CONSUMER",req2.consumer)
assert_equal("http://sample.com/some_scope",req2.scope)
end
def test_add_consumer
@req.consumer="CONSUMER"
assert_equal("CONSUMER",@req.consumer)
end
def test_add_scope
@req.scope="http://sample.com/some_scope"
assert_equal("http://sample.com/some_scope",@req.scope)
end
def test_get_extension_args
assert_equal({}, @req.get_extension_args)
@req.consumer="CONSUMER"
assert_equal({'consumer' => 'CONSUMER'}, @req.get_extension_args)
@req.scope="http://sample.com/some_scope"
assert_equal({'consumer' => 'CONSUMER', 'scope' => 'http://sample.com/some_scope'}, @req.get_extension_args)
end
def test_parse_extension_args
args = {'consumer' => 'CONSUMER', 'scope' => 'http://sample.com/some_scope'}
@req.parse_extension_args(args)
assert_equal("CONSUMER",@req.consumer)
assert_equal("http://sample.com/some_scope",@req.scope)
end
def test_parse_extension_args_empty
@req.parse_extension_args({})
assert_nil( @req.consumer )
assert_nil( @req.scope )
end
def test_from_openid_request
openid_req_msg = Message.from_openid_args({
'mode' => 'checkid_setup',
'ns' => OPENID2_NS,
'ns.oauth' => OAuth::NS_URI,
'oauth.consumer' => 'CONSUMER',
'oauth.scope' => "http://sample.com/some_scope"
})
oid_req = Server::OpenIDRequest.new
oid_req.message = openid_req_msg
req = OAuth::Request.from_openid_request(oid_req)
assert_equal("CONSUMER",req.consumer)
assert_equal("http://sample.com/some_scope",req.scope)
end
def test_from_openid_request_no_oauth
message = Message.new
openid_req = Server::OpenIDRequest.new
openid_req.message = message
oauth_req = OAuth::Request.from_openid_request(openid_req)
assert(oauth_req.nil?)
end
end
class DummySuccessResponse
attr_accessor :message
def initialize(message, signed_stuff)
@message = message
@signed_stuff = signed_stuff
end
def get_signed_ns(ns_uri)
return @signed_stuff
end
end
class OAuthResponseTestCase < Test::Unit::TestCase
def setup
@req = OAuth::Response.new
end
def test_construct
assert_nil(@req.request_token)
assert_nil(@req.scope)
req2 = OAuth::Response.new("REQUESTTOKEN","http://sample.com/some_scope")
assert_equal("REQUESTTOKEN",req2.request_token)
assert_equal("http://sample.com/some_scope",req2.scope)
end
def test_add_request_token
@req.request_token="REQUESTTOKEN"
assert_equal("REQUESTTOKEN",@req.request_token)
end
def test_add_scope
@req.scope="http://sample.com/some_scope"
assert_equal("http://sample.com/some_scope",@req.scope)
end
def test_get_extension_args
assert_equal({}, @req.get_extension_args)
@req.request_token="REQUESTTOKEN"
assert_equal({'request_token' => 'REQUESTTOKEN'}, @req.get_extension_args)
@req.scope="http://sample.com/some_scope"
assert_equal({'request_token' => 'REQUESTTOKEN', 'scope' => 'http://sample.com/some_scope'}, @req.get_extension_args)
end
def test_parse_extension_args
args = {'request_token' => 'REQUESTTOKEN', 'scope' => 'http://sample.com/some_scope'}
@req.parse_extension_args(args)
assert_equal("REQUESTTOKEN",@req.request_token)
assert_equal("http://sample.com/some_scope",@req.scope)
end
def test_parse_extension_args_empty
@req.parse_extension_args({})
assert_nil( @req.request_token )
assert_nil( @req.scope )
end
def test_from_success_response
openid_req_msg = Message.from_openid_args({
'mode' => 'id_res',
'ns' => OPENID2_NS,
'ns.oauth' => OAuth::NS_URI,
'ns.oauth' => OAuth::NS_URI,
'oauth.request_token' => 'REQUESTTOKEN',
'oauth.scope' => "http://sample.com/some_scope"
})
signed_stuff = {
'request_token' => 'REQUESTTOKEN',
'scope' => "http://sample.com/some_scope"
}
oid_req = DummySuccessResponse.new(openid_req_msg, signed_stuff)
req = OAuth::Response.from_success_response(oid_req)
assert_equal("REQUESTTOKEN",req.request_token)
assert_equal("http://sample.com/some_scope",req.scope)
end
def test_from_success_response_unsigned
openid_req_msg = Message.from_openid_args({
'mode' => 'id_res',
'ns' => OPENID2_NS,
'ns.oauth' => OAuth::NS_URI,
'oauth.request_token' => 'REQUESTTOKEN',
'oauth.scope' => "http://sample.com/some_scope"
})
signed_stuff = {}
endpoint = OpenIDServiceEndpoint.new
oid_req = Consumer::SuccessResponse.new(endpoint, openid_req_msg, signed_stuff)
req = OAuth::Response.from_success_response(oid_req)
assert(req.nil?, req.inspect)
end
end
end
end
|