Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

210 lines (174 sloc) 6.691 kb
require 'thread'
require 'rack/lint'
require 'rack/mock'
require 'rack/session/pool'
describe Rack::Session::Pool do
session_key = Rack::Session::Pool::DEFAULT_OPTIONS[:key]
session_match = /#{session_key}=[0-9a-fA-F]+;/
incrementor = lambda do |env|
env["rack.session"]["counter"] ||= 0
env["rack.session"]["counter"] += 1
Rack::Response.new(env["rack.session"].inspect).to_a
end
session_id = Rack::Lint.new(lambda do |env|
Rack::Response.new(env["rack.session"].inspect).to_a
end)
nothing = Rack::Lint.new(lambda do |env|
Rack::Response.new("Nothing").to_a
end)
drop_session = Rack::Lint.new(lambda do |env|
env['rack.session.options'][:drop] = true
incrementor.call(env)
end)
renew_session = Rack::Lint.new(lambda do |env|
env['rack.session.options'][:renew] = true
incrementor.call(env)
end)
defer_session = Rack::Lint.new(lambda do |env|
env['rack.session.options'][:defer] = true
incrementor.call(env)
end)
incrementor = Rack::Lint.new(incrementor)
it "creates a new cookie" do
pool = Rack::Session::Pool.new(incrementor)
res = Rack::MockRequest.new(pool).get("/")
res["Set-Cookie"].should.match session_match
res.body.should.equal '{"counter"=>1}'
end
it "determines session from a cookie" do
pool = Rack::Session::Pool.new(incrementor)
req = Rack::MockRequest.new(pool)
cookie = req.get("/")["Set-Cookie"]
req.get("/", "HTTP_COOKIE" => cookie).
body.should.equal '{"counter"=>2}'
req.get("/", "HTTP_COOKIE" => cookie).
body.should.equal '{"counter"=>3}'
end
it "survives nonexistant cookies" do
pool = Rack::Session::Pool.new(incrementor)
res = Rack::MockRequest.new(pool).
get("/", "HTTP_COOKIE" => "#{session_key}=blarghfasel")
res.body.should.equal '{"counter"=>1}'
end
it "does not send the same session id if it did not change" do
pool = Rack::Session::Pool.new(incrementor)
req = Rack::MockRequest.new(pool)
res0 = req.get("/")
cookie = res0["Set-Cookie"][session_match]
res0.body.should.equal '{"counter"=>1}'
pool.pool.size.should.equal 1
res1 = req.get("/", "HTTP_COOKIE" => cookie)
res1["Set-Cookie"].should.be.nil
res1.body.should.equal '{"counter"=>2}'
pool.pool.size.should.equal 1
res2 = req.get("/", "HTTP_COOKIE" => cookie)
res2["Set-Cookie"].should.be.nil
res2.body.should.equal '{"counter"=>3}'
pool.pool.size.should.equal 1
end
it "deletes cookies with :drop option" do
pool = Rack::Session::Pool.new(incrementor)
req = Rack::MockRequest.new(pool)
drop = Rack::Utils::Context.new(pool, drop_session)
dreq = Rack::MockRequest.new(drop)
res1 = req.get("/")
session = (cookie = res1["Set-Cookie"])[session_match]
res1.body.should.equal '{"counter"=>1}'
pool.pool.size.should.equal 1
res2 = dreq.get("/", "HTTP_COOKIE" => cookie)
res2["Set-Cookie"].should.be.nil
res2.body.should.equal '{"counter"=>2}'
pool.pool.size.should.equal 0
res3 = req.get("/", "HTTP_COOKIE" => cookie)
res3["Set-Cookie"][session_match].should.not.equal session
res3.body.should.equal '{"counter"=>1}'
pool.pool.size.should.equal 1
end
it "provides new session id with :renew option" do
pool = Rack::Session::Pool.new(incrementor)
req = Rack::MockRequest.new(pool)
renew = Rack::Utils::Context.new(pool, renew_session)
rreq = Rack::MockRequest.new(renew)
res1 = req.get("/")
session = (cookie = res1["Set-Cookie"])[session_match]
res1.body.should.equal '{"counter"=>1}'
pool.pool.size.should.equal 1
res2 = rreq.get("/", "HTTP_COOKIE" => cookie)
new_cookie = res2["Set-Cookie"]
new_session = new_cookie[session_match]
new_session.should.not.equal session
res2.body.should.equal '{"counter"=>2}'
pool.pool.size.should.equal 1
res3 = req.get("/", "HTTP_COOKIE" => new_cookie)
res3.body.should.equal '{"counter"=>3}'
pool.pool.size.should.equal 1
res4 = req.get("/", "HTTP_COOKIE" => cookie)
res4.body.should.equal '{"counter"=>1}'
pool.pool.size.should.equal 2
end
it "omits cookie with :defer option" do
pool = Rack::Session::Pool.new(incrementor)
defer = Rack::Utils::Context.new(pool, defer_session)
dreq = Rack::MockRequest.new(defer)
res1 = dreq.get("/")
res1["Set-Cookie"].should.equal nil
res1.body.should.equal '{"counter"=>1}'
pool.pool.size.should.equal 1
end
# anyone know how to do this better?
it "should merge sessions when multithreaded" do
unless $DEBUG
1.should.equal 1
next
end
warn 'Running multithread tests for Session::Pool'
pool = Rack::Session::Pool.new(incrementor)
req = Rack::MockRequest.new(pool)
res = req.get('/')
res.body.should.equal '{"counter"=>1}'
cookie = res["Set-Cookie"]
sess_id = cookie[/#{pool.key}=([^,;]+)/,1]
delta_incrementor = lambda do |env|
# emulate disconjoinment of threading
env['rack.session'] = env['rack.session'].dup
Thread.stop
env['rack.session'][(Time.now.usec*rand).to_i] = true
incrementor.call(env)
end
tses = Rack::Utils::Context.new pool, delta_incrementor
treq = Rack::MockRequest.new(tses)
tnum = rand(7).to_i+5
r = Array.new(tnum) do
Thread.new(treq) do |run|
run.get('/', "HTTP_COOKIE" => cookie, 'rack.multithread' => true)
end
end.reverse.map{|t| t.run.join.value }
r.each do |resp|
resp['Set-Cookie'].should.equal cookie
resp.body.should.include '"counter"=>2'
end
session = pool.pool[sess_id]
session.size.should.equal tnum+1 # counter
session['counter'].should.equal 2 # meeeh
end
it "does not return a cookie if cookie was not read/written" do
app = Rack::Session::Pool.new(nothing)
res = Rack::MockRequest.new(app).get("/")
res["Set-Cookie"].should.be.nil
end
it "does not return a cookie if cookie was not written (only read)" do
app = Rack::Session::Pool.new(session_id)
res = Rack::MockRequest.new(app).get("/")
res["Set-Cookie"].should.be.nil
end
it "returns even if not read/written if :expire_after is set" do
app = Rack::Session::Pool.new(nothing, :expire_after => 3600)
res = Rack::MockRequest.new(app).get("/", 'rack.session' => {'not' => 'empty'})
res["Set-Cookie"].should.not.be.nil
end
it "returns no cookie if no data was written and no session was created previously, even if :expire_after is set" do
app = Rack::Session::Pool.new(nothing, :expire_after => 3600)
res = Rack::MockRequest.new(app).get("/")
res["Set-Cookie"].should.be.nil
end
end
Jump to Line
Something went wrong with that request. Please try again.