4e14ccc74d
Instiki now runs on the Rails 2.3.0 Candidate Release.
Among other improvements, this means that it now
automagically selects between WEBrick and Mongrel.
Just run
./instiki --daemon
133 lines
4.1 KiB
Ruby
133 lines
4.1 KiB
Ruby
require 'test/spec'
|
|
|
|
begin
|
|
require 'rack/session/memcache'
|
|
require 'rack/mock'
|
|
require 'rack/response'
|
|
require 'thread'
|
|
|
|
context "Rack::Session::Memcache" do
|
|
incrementor = lambda { |env|
|
|
env["rack.session"]["counter"] ||= 0
|
|
env["rack.session"]["counter"] += 1
|
|
Rack::Response.new(env["rack.session"].inspect).to_a
|
|
}
|
|
|
|
# Keep this first.
|
|
specify "startup" do
|
|
$pid = fork {
|
|
exec "memcached"
|
|
}
|
|
sleep 1
|
|
end
|
|
|
|
specify "faults on no connection" do
|
|
lambda do
|
|
Rack::Session::Memcache.new(incrementor, :memcache_server => '')
|
|
end.should.raise
|
|
end
|
|
|
|
specify "creates a new cookie" do
|
|
cache = Rack::Session::Memcache.new(incrementor)
|
|
res = Rack::MockRequest.new(cache).get("/")
|
|
res["Set-Cookie"].should.match("rack.session=")
|
|
res.body.should.equal '{"counter"=>1}'
|
|
end
|
|
|
|
specify "determines session from a cookie" do
|
|
cache = Rack::Session::Memcache.new(incrementor)
|
|
res = Rack::MockRequest.new(cache).get("/")
|
|
cookie = res["Set-Cookie"]
|
|
res = Rack::MockRequest.new(cache).get("/", "HTTP_COOKIE" => cookie)
|
|
res.body.should.equal '{"counter"=>2}'
|
|
res = Rack::MockRequest.new(cache).get("/", "HTTP_COOKIE" => cookie)
|
|
res.body.should.equal '{"counter"=>3}'
|
|
end
|
|
|
|
specify "survives broken cookies" do
|
|
cache = Rack::Session::Memcache.new(incrementor)
|
|
res = Rack::MockRequest.new(cache).
|
|
get("/", "HTTP_COOKIE" => "rack.session=blarghfasel")
|
|
res.body.should.equal '{"counter"=>1}'
|
|
end
|
|
|
|
specify "maintains freshness" do
|
|
cache = Rack::Session::Memcache.new(incrementor, :expire_after => 3)
|
|
res = Rack::MockRequest.new(cache).get('/')
|
|
res.body.should.include '"counter"=>1'
|
|
cookie = res["Set-Cookie"]
|
|
res = Rack::MockRequest.new(cache).get('/', "HTTP_COOKIE" => cookie)
|
|
res["Set-Cookie"].should.equal cookie
|
|
res.body.should.include '"counter"=>2'
|
|
puts 'Sleeping to expire session' if $DEBUG
|
|
sleep 4
|
|
res = Rack::MockRequest.new(cache).get('/', "HTTP_COOKIE" => cookie)
|
|
res["Set-Cookie"].should.not.equal cookie
|
|
res.body.should.include '"counter"=>1'
|
|
end
|
|
|
|
specify "multithread: should cleanly merge sessions" do
|
|
cache = Rack::Session::Memcache.new(incrementor)
|
|
drop_counter = Rack::Session::Memcache.new(proc do |env|
|
|
env['rack.session'].delete 'counter'
|
|
env['rack.session']['foo'] = 'bar'
|
|
[200, {'Content-Type'=>'text/plain'}, env['rack.session'].inspect]
|
|
end)
|
|
|
|
res = Rack::MockRequest.new(cache).get('/')
|
|
res.body.should.equal '{"counter"=>1}'
|
|
cookie = res["Set-Cookie"]
|
|
sess_id = cookie[/#{cache.key}=([^,;]+)/, 1]
|
|
|
|
res = Rack::MockRequest.new(cache).get('/', "HTTP_COOKIE" => cookie)
|
|
res.body.should.equal '{"counter"=>2}'
|
|
|
|
r = Array.new(rand(7).to_i+2) do |i|
|
|
app = proc do |env|
|
|
env['rack.session'][i] = Time.now
|
|
sleep 2
|
|
env['rack.session'] = env['rack.session'].dup
|
|
env['rack.session'][i] -= Time.now
|
|
incrementor.call(env)
|
|
end
|
|
Thread.new(cache.context(app)) do |run|
|
|
Rack::MockRequest.new(run).
|
|
get('/', "HTTP_COOKIE" => cookie, 'rack.multithread' => true)
|
|
end
|
|
end
|
|
|
|
r.reverse!
|
|
|
|
r.map! do |t|
|
|
p t if $DEBUG
|
|
t.join.value
|
|
end
|
|
|
|
r.each do |res|
|
|
res['Set-Cookie'].should.equal cookie
|
|
res.body.should.include '"counter"=>3'
|
|
end
|
|
|
|
session = cache.pool[sess_id]
|
|
session.size.should.be r.size+1
|
|
session['counter'].should.be 3
|
|
|
|
res = Rack::MockRequest.new(drop_counter).get('/', "HTTP_COOKIE" => cookie)
|
|
res.body.should.include '"foo"=>"bar"'
|
|
|
|
session = cache.pool[sess_id]
|
|
session.size.should.be r.size+1
|
|
session['counter'].should.be.nil?
|
|
session['foo'].should.equal 'bar'
|
|
end
|
|
|
|
# Keep this last.
|
|
specify "shutdown" do
|
|
Process.kill 15, $pid
|
|
Process.wait($pid).should.equal $pid
|
|
end
|
|
end
|
|
rescue LoadError
|
|
$stderr.puts "Skipping Rack::Session::Memcache tests (Memcache is required). `gem install memcache-client` and try again."
|
|
end
|