83 lines
3.2 KiB
Ruby
83 lines
3.2 KiB
Ruby
|
require 'test/spec'
|
||
|
|
|
||
|
|
require 'rack/session/cookie'
|
||
|
|
require 'rack/mock'
|
||
|
|
require 'rack/response'
|
||
|
|
|
||
|
|
context "Rack::Session::Cookie" do
|
||
|
|
incrementor = lambda { |env|
|
||
|
|
env["rack.session"]["counter"] ||= 0
|
||
|
|
env["rack.session"]["counter"] += 1
|
||
|
|
Rack::Response.new(env["rack.session"].inspect).to_a
|
||
|
|
}
|
||
|
|
|
||
|
|
specify "creates a new cookie" do
|
||
|
|
res = Rack::MockRequest.new(Rack::Session::Cookie.new(incrementor)).get("/")
|
||
|
|
res["Set-Cookie"].should.match("rack.session=")
|
||
|
|
res.body.should.equal '{"counter"=>1}'
|
||
|
|
end
|
||
|
|
|
||
|
|
specify "loads from a cookie" do
|
||
|
|
res = Rack::MockRequest.new(Rack::Session::Cookie.new(incrementor)).get("/")
|
||
|
|
cookie = res["Set-Cookie"]
|
||
|
|
res = Rack::MockRequest.new(Rack::Session::Cookie.new(incrementor)).
|
||
|
|
get("/", "HTTP_COOKIE" => cookie)
|
||
|
|
res.body.should.equal '{"counter"=>2}'
|
||
|
|
cookie = res["Set-Cookie"]
|
||
|
|
res = Rack::MockRequest.new(Rack::Session::Cookie.new(incrementor)).
|
||
|
|
get("/", "HTTP_COOKIE" => cookie)
|
||
|
|
res.body.should.equal '{"counter"=>3}'
|
||
|
|
end
|
||
|
|
|
||
|
|
specify "survives broken cookies" do
|
||
|
|
res = Rack::MockRequest.new(Rack::Session::Cookie.new(incrementor)).
|
||
|
|
get("/", "HTTP_COOKIE" => "rack.session=blarghfasel")
|
||
|
|
res.body.should.equal '{"counter"=>1}'
|
||
|
|
end
|
||
|
|
|
||
|
|
bigcookie = lambda { |env|
|
||
|
|
env["rack.session"]["cookie"] = "big" * 3000
|
||
|
|
Rack::Response.new(env["rack.session"].inspect).to_a
|
||
|
|
}
|
||
|
|
|
||
|
|
specify "barks on too big cookies" do
|
||
|
|
lambda {
|
||
|
|
Rack::MockRequest.new(Rack::Session::Cookie.new(bigcookie)).
|
||
|
|
get("/", :fatal => true)
|
||
|
|
}.should.raise(Rack::MockRequest::FatalWarning)
|
||
|
|
end
|
||
|
|
|
||
|
|
specify "creates a new cookie with integrity hash" do
|
||
|
|
res = Rack::MockRequest.new(Rack::Session::Cookie.new(incrementor, :secret => 'test')).get("/")
|
||
|
|
if RUBY_VERSION < "1.9"
|
||
|
|
res["Set-Cookie"].should.match("rack.session=BAh7BiIMY291bnRlcmkG%0A--1439b4d37b9d4b04c603848382f712d6fcd31088")
|
||
|
|
else
|
||
|
|
res["Set-Cookie"].should.match("rack.session=BAh7BkkiDGNvdW50ZXIGOg1lbmNvZGluZyINVVMtQVNDSUlpBg%3D%3D%0A--d7a6637b94d2728194a96c18484e1f7ed9074a83")
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
specify "loads from a cookie wih integrity hash" do
|
||
|
|
res = Rack::MockRequest.new(Rack::Session::Cookie.new(incrementor, :secret => 'test')).get("/")
|
||
|
|
cookie = res["Set-Cookie"]
|
||
|
|
res = Rack::MockRequest.new(Rack::Session::Cookie.new(incrementor, :secret => 'test')).
|
||
|
|
get("/", "HTTP_COOKIE" => cookie)
|
||
|
|
res.body.should.equal '{"counter"=>2}'
|
||
|
|
cookie = res["Set-Cookie"]
|
||
|
|
res = Rack::MockRequest.new(Rack::Session::Cookie.new(incrementor, :secret => 'test')).
|
||
|
|
get("/", "HTTP_COOKIE" => cookie)
|
||
|
|
res.body.should.equal '{"counter"=>3}'
|
||
|
|
end
|
||
|
|
|
||
|
|
specify "ignores tampered with session cookies" do
|
||
|
|
app = Rack::Session::Cookie.new(incrementor, :secret => 'test')
|
||
|
|
response1 = Rack::MockRequest.new(app).get("/")
|
||
|
|
_, digest = response1["Set-Cookie"].split("--")
|
||
|
|
tampered_with_cookie = "hackerman-was-here" + "--" + digest
|
||
|
|
response2 = Rack::MockRequest.new(app).get("/", "HTTP_COOKIE" =>
|
||
|
|
tampered_with_cookie)
|
||
|
|
|
||
|
|
# The tampered-with cookie is ignored, so we get back an identical Set-Cookie
|
||
|
|
response2["Set-Cookie"].should.equal(response1["Set-Cookie"])
|
||
|
|
end
|
||
|
|
end
|