require 'test/spec'

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
  }

  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'
    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.reverse.map{|t| t.join.value }

    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
end