spec_rack_session_memcache.rb

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