redis-store 0.3.6

data/spec/rack/session/redis_spec.rb

@@ -0,0 +1,238 @@
+require File.join(File.dirname(__FILE__), "/../../spec_helper")
+
+module Rack
+  module Session
+    describe "Rack::Session::Redis" do
+      before(:each) do
+        @session_key = Rack::Session::Redis::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
+        @drop_session = proc do |env|
+          env['rack.session.options'][:drop] = true
+          @incrementor.call(env)
+        end
+        @renew_session = proc do |env|
+          env['rack.session.options'][:renew] = true
+          @incrementor.call(env)
+        end
+        @defer_session = proc do |env|
+          env['rack.session.options'][:defer] = true
+          @incrementor.call(env)
+        end
+      end
+
+      it "should specify connection params" do
+        pool = Rack::Session::Redis.new(@incrementor, :redis_server => "localhost:6380/1").pool
+        pool.should be_kind_of(MarshaledRedis)
+        pool.host.should == "localhost"
+        pool.port.should == 6380
+        pool.db.should == 1
+
+        pool = Rack::Session::Redis.new(@incrementor, :redis_server => ["localhost:6379", "localhost:6380"]).pool
+        pool.should be_kind_of(DistributedMarshaledRedis)
+      end
+
+      it "creates a new cookie" do
+        pool = Rack::Session::Redis.new(@incrementor)
+        res = Rack::MockRequest.new(pool).get("/")
+        res["Set-Cookie"].should match(/#{@session_key}=/)
+        res.body.should == '{"counter"=>1}'
+      end
+
+      it "determines session from a cookie" do
+        pool = Rack::Session::Redis.new(@incrementor)
+        req = Rack::MockRequest.new(pool)
+        res = req.get("/")
+        cookie = res["Set-Cookie"]
+        req.get("/", "HTTP_COOKIE" => cookie).
+          body.should == '{"counter"=>2}'
+        req.get("/", "HTTP_COOKIE" => cookie).
+          body.should == '{"counter"=>3}'
+      end
+      
+      it "survives nonexistant cookies" do
+        bad_cookie = "rack.session=blarghfasel"
+        pool = Rack::Session::Redis.new(@incrementor)
+        res = Rack::MockRequest.new(pool).
+          get("/", "HTTP_COOKIE" => bad_cookie)
+        res.body.should == '{"counter"=>1}'
+        cookie = res["Set-Cookie"][@session_match]
+        cookie.should_not match(/#{bad_cookie}/)
+      end
+
+      it "should maintain freshness" do
+        pool = Rack::Session::Redis.new(@incrementor, :expire_after => 3)
+        res = Rack::MockRequest.new(pool).get('/')
+        res.body.should include('"counter"=>1')
+        cookie = res["Set-Cookie"]
+        res = Rack::MockRequest.new(pool).get('/', "HTTP_COOKIE" => cookie)
+        res["Set-Cookie"].should == cookie
+        res.body.should include('"counter"=>2')
+        puts 'Sleeping to expire session' if $DEBUG
+        sleep 4
+        res = Rack::MockRequest.new(pool).get('/', "HTTP_COOKIE" => cookie)
+        res["Set-Cookie"].should_not == cookie
+        res.body.should include('"counter"=>1')
+      end
+
+      it "deletes cookies with :drop option" do
+        pool = Rack::Session::Redis.new(@incrementor)
+        req = Rack::MockRequest.new(pool)
+        drop = Rack::Utils::Context.new(pool, @drop_session)
+        dreq = Rack::MockRequest.new(drop)
+
+        res0 = req.get("/")
+        session = (cookie = res0["Set-Cookie"])[@session_match]
+        res0.body.should == '{"counter"=>1}'
+
+        res1 = req.get("/", "HTTP_COOKIE" => cookie)
+        res1["Set-Cookie"][@session_match].should == session
+        res1.body.should == '{"counter"=>2}'
+
+        res2 = dreq.get("/", "HTTP_COOKIE" => cookie)
+        res2["Set-Cookie"].should be_nil
+        res2.body.should == '{"counter"=>3}'
+
+        res3 = req.get("/", "HTTP_COOKIE" => cookie)
+        res3["Set-Cookie"][@session_match].should_not == session
+        res3.body.should == '{"counter"=>1}'
+      end
+
+      it "provides new session id with :renew option" do
+        pool = Rack::Session::Redis.new(@incrementor)
+        req = Rack::MockRequest.new(pool)
+        renew = Rack::Utils::Context.new(pool, @renew_session)
+        rreq = Rack::MockRequest.new(renew)
+
+        res0 = req.get("/")
+        session = (cookie = res0["Set-Cookie"])[@session_match]
+        res0.body.should == '{"counter"=>1}'
+
+        res1 = req.get("/", "HTTP_COOKIE" => cookie)
+        res1["Set-Cookie"][@session_match].should == session
+        res1.body.should == '{"counter"=>2}'
+
+        res2 = rreq.get("/", "HTTP_COOKIE" => cookie)
+        new_cookie = res2["Set-Cookie"]
+        new_session = new_cookie[@session_match]
+        new_session.should_not == session
+        res2.body.should == '{"counter"=>3}'
+
+        res3 = req.get("/", "HTTP_COOKIE" => new_cookie)
+        res3["Set-Cookie"][@session_match].should == new_session
+        res3.body.should == '{"counter"=>4}'
+      end
+
+      specify "omits cookie with :defer option" do
+        pool = Rack::Session::Redis.new(@incrementor)
+        req = Rack::MockRequest.new(pool)
+        defer = Rack::Utils::Context.new(pool, @defer_session)
+        dreq = Rack::MockRequest.new(defer)
+
+        res0 = req.get("/")
+        session = (cookie = res0["Set-Cookie"])[@session_match]
+        res0.body.should == '{"counter"=>1}'
+
+        res1 = req.get("/", "HTTP_COOKIE" => cookie)
+        res1["Set-Cookie"][@session_match].should == session
+        res1.body.should == '{"counter"=>2}'
+
+        res2 = dreq.get("/", "HTTP_COOKIE" => cookie)
+        res2["Set-Cookie"].should be_nil
+        res2.body.should == '{"counter"=>3}'
+
+        res3 = req.get("/", "HTTP_COOKIE" => cookie)
+        res3["Set-Cookie"][@session_match].should == session
+        res3.body.should == '{"counter"=>4}'
+      end
+
+      # anyone know how to do this better?
+      specify "multithread: should cleanly merge sessions" do
+        next unless $DEBUG
+        warn 'Running multithread test for Session::Redis'
+        pool = Rack::Session::Redis.new(@incrementor)
+        req = Rack::MockRequest.new(pool)
+
+        res = req.get('/')
+        res.body.should == '{"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 |res|
+          res['Set-Cookie'].should == cookie
+          res.body.should include('"counter"=>2')
+        end
+
+        session = pool.pool.get(sess_id)
+        session.size.should == tnum+1 # counter
+        session['counter'].should == 2 # meeeh
+
+        tnum = rand(7).to_i+5
+        r = Array.new(tnum) do |i|
+          delta_time = proc do |env|
+            env['rack.session'][i]  = Time.now
+            Thread.stop
+            env['rack.session']     = env['rack.session'].dup
+            env['rack.session'][i] -= Time.now
+            @incrementor.call(env)
+          end
+          app = Rack::Utils::Context.new pool, time_delta
+          req = Rack::MockRequest.new app
+          Thread.new(req) do |run|
+            run.get('/', "HTTP_COOKIE" => cookie, 'rack.multithread' => true)
+          end
+        end.reverse.map{|t| t.run.join.value }
+        r.each do |res|
+          res['Set-Cookie'].should == cookie
+          res.body.should include('"counter"=>3')
+        end
+
+        session = pool.pool.get(sess_id)
+        session.size.should == tnum+1
+        session['counter'].should == 3
+
+        drop_counter = proc do |env|
+          env['rack.session'].delete 'counter'
+          env['rack.session']['foo'] = 'bar'
+          [200, {'Content-Type'=>'text/plain'}, env['rack.session'].inspect]
+        end
+        tses = Rack::Utils::Context.new pool, drop_counter
+        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 |res|
+          res['Set-Cookie'].should == cookie
+          res.body.should include('"foo"=>"bar"')
+        end
+
+        session = pool.pool.get(sess_id)
+        session.size.should == r.size+1
+        session['counter'].should be_nil
+        session['foo'].should == 'bar'
+      end
+    end
+  end
+end

data/spec/redis/distributed_marshaled_redis_spec.rb

@@ -0,0 +1,35 @@
+require File.join(File.dirname(__FILE__), "/../spec_helper")
+
+describe "DistributedMarshaledRedis" do
+  before(:each) do
+    @dmr = DistributedMarshaledRedis.new [
+      {:host => "localhost", :port => "6380", :db => 0},
+      {:host => "localhost", :port => "6381", :db => 0}
+    ]
+    @rabbit = OpenStruct.new :name => "bunny"
+    @white_rabbit = OpenStruct.new :color => "white"
+    @dmr.set "rabbit", @rabbit
+  end
+
+  after(:all) do
+    @dmr.ring.nodes.each { |server| server.flush_db }
+  end
+
+  it "should accept connection params" do
+    dmr = DistributedMarshaledRedis.new [ :host => "localhost", :port => "6380", :db => "1" ]
+    dmr.ring.should have(1).node
+    mr = dmr.ring.nodes.first
+    mr.host.should == "localhost"
+    mr.port.should == 6380
+    mr.db.should == 1
+  end
+
+  it "should set an object" do
+    @dmr.set "rabbit", @white_rabbit
+    @dmr.get("rabbit").should == @white_rabbit
+  end
+
+  it "should get an object" do
+    @dmr.get("rabbit").should == @rabbit
+  end
+end

data/spec/redis/marshaled_redis_spec.rb

@@ -0,0 +1,49 @@
+require File.join(File.dirname(__FILE__), "/../spec_helper")
+
+describe "MarshaledRedis" do
+  before(:each) do
+    @mr = MarshaledRedis.new
+    @rabbit = OpenStruct.new :name => "bunny"
+    @white_rabbit = OpenStruct.new :color => "white"
+    @mr.set    "rabbit", @rabbit
+    @mr.delete "rabbit2"
+  end
+
+  it "should unmarshal an object on get" do
+    @mr.get("rabbit").should === @rabbit
+  end
+
+  it "should marshal object on set" do
+    @mr.set "rabbit", @white_rabbit
+    @mr.get("rabbit").should === @white_rabbit
+  end
+
+  it "should not unmarshal object on get if raw option is true" do
+    @mr.get("rabbit", :raw => true).should == "\004\bU:\017OpenStruct{\006:\tname\"\nbunny"
+  end
+
+  it "should not marshal object on set if raw option is true" do
+    @mr.set "rabbit", @white_rabbit, :raw => true
+    @mr.get("rabbit", :raw => true).should == %(#<OpenStruct color="white">)
+  end
+
+  it "should not unmarshal object if getting an empty string" do
+    @mr.set "empty_string", ""
+    lambda { @mr.get("empty_string").should == "" }.should_not raise_error
+  end
+
+  it "should not set an object if already exist" do
+    @mr.set_unless_exists "rabbit", @white_rabbit
+    @mr.get("rabbit").should === @rabbit
+  end
+
+  it "should marshal object on set_unless_exists" do
+    @mr.set_unless_exists "rabbit2", @white_rabbit
+    @mr.get("rabbit2").should === @white_rabbit
+  end
+
+  it "should not marshal object on set_unless_exists if raw option is true" do
+    @mr.set_unless_exists "rabbit2", @white_rabbit, :raw => true
+    @mr.get("rabbit2", :raw => true).should == %(#<OpenStruct color="white">)
+  end
+end

data/spec/redis/redis_factory_spec.rb

@@ -0,0 +1,34 @@
+require File.join(File.dirname(__FILE__), "/../spec_helper")
+
+describe "RedisFactory" do
+  it "should instantiate a MarshaledRedis store" do
+    store = RedisFactory.create
+    store.should be_kind_of(MarshaledRedis)
+    store.host.should == "127.0.0.1"
+    store.port.should == 6379
+    store.db.should == 0
+  end
+
+  it "should allow to specify host" do
+    store = RedisFactory.create "localhost"
+    store.host.should == "localhost"
+  end
+
+  it "should allow to specify port" do
+    store = RedisFactory.create "localhost:6380"
+    store.host.should == "localhost"
+    store.port.should == 6380
+  end
+
+  it "should allow to specify db" do
+    store = RedisFactory.create "localhost:6380/13"
+    store.host.should == "localhost"
+    store.port.should == 6380
+    store.db.should == 13
+  end
+
+  it "should instantiate a DistributedMarshaledRedis store" do
+    store = RedisFactory.create "localhost:6379", "localhost:6380"
+    store.should be_kind_of(DistributedMarshaledRedis)
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,16 @@
+$: << File.join(File.dirname(__FILE__), "/../lib")
+require "rubygems"
+require "ostruct"
+require "spec"
+require "redis"
+require "merb"
+require "rack/cache"
+require "rack/cache/metastore"
+require "rack/cache/entitystore"
+require "redis-store"
+require "activesupport"
+require "cache/rails/redis_store"
+require "cache/sinatra/redis_store"
+
+class Redis; attr_reader :host, :port, :db end
+$DEBUG = ENV["DEBUG"] === "true"

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification 
+name: redis-store
+version: !ruby/object:Gem::Version 
+  version: 0.3.6
+platform: ruby
+authors: 
+- Luca Guidi
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-06-18 00:00:00 +02:00
+default_executable: 
+dependencies: []
+
+description: Rack::Session, Rack::Cache and cache Redis stores for Ruby web frameworks.
+email: guidi.luca@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.textile
+files: 
+- MIT-LICENSE
+- README.textile
+- Rakefile
+- lib/cache/merb/redis_store.rb
+- lib/cache/rails/redis_store.rb
+- lib/cache/sinatra/redis_store.rb
+- lib/rack/cache/redis_entitystore.rb
+- lib/rack/cache/redis_metastore.rb
+- lib/rack/session/merb.rb
+- lib/rack/session/redis.rb
+- lib/redis-store.rb
+- lib/redis/distributed_marshaled_redis.rb
+- lib/redis/marshaled_redis.rb
+- lib/redis/redis_factory.rb
+- redis-store.gemspec
+- spec/cache/merb/redis_store_spec.rb
+- spec/cache/rails/redis_store_spec.rb
+- spec/cache/sinatra/redis_store_spec.rb
+- spec/config/master.conf
+- spec/config/single.conf
+- spec/config/slave.conf
+- spec/rack/cache/entitystore/pony.jpg
+- spec/rack/cache/entitystore/redis_spec.rb
+- spec/rack/cache/metastore/redis_spec.rb
+- spec/rack/session/redis_spec.rb
+- spec/redis/distributed_marshaled_redis_spec.rb
+- spec/redis/marshaled_redis_spec.rb
+- spec/redis/redis_factory_spec.rb
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://lucaguidi.com
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Rack::Session, Rack::Cache and cache Redis stores for Ruby web frameworks.
+test_files: 
+- spec/cache/merb/redis_store_spec.rb
+- spec/cache/rails/redis_store_spec.rb
+- spec/cache/sinatra/redis_store_spec.rb
+- spec/rack/cache/entitystore/redis_spec.rb
+- spec/rack/cache/metastore/redis_spec.rb
+- spec/rack/session/redis_spec.rb
+- spec/redis/distributed_marshaled_redis_spec.rb
+- spec/redis/marshaled_redis_spec.rb
+- spec/redis/redis_factory_spec.rb