tokyo_store 0.3.0

data/spec/rack/session/tyrant_spec.rb

@@ -0,0 +1,249 @@
+# Code from http://github.com/jodosha/redis-store
+require File.expand_path(File.dirname(__FILE__) + '/../../spec_helper')
+
+module Rack
+  module Session
+    class Tyrant
+      attr_reader :host, :port
+    end
+    describe "Rack::Session::Tyrant" do
+      before(:each) do
+        @session_key = Rack::Session::Tyrant::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::Tyrant.new(@incrementor, :tyrant_server => "127.0.0.1:1978").pool
+        pool.should be_kind_of(TokyoTyrant::RDB)
+        #pool.host.should eql("127.0.0.1")
+        #pool.port.should eql("1978")
+
+        # pool = Rack::Session::Tyrant.new(@incrementor, :tokyo_server => ["localhost:6379", "localhost:6380"]).pool
+        # pool.should be_kind_of(DistributedMarshaledTyrant)
+      end
+
+      it "should not raise error on connect" do
+        lambda{ Rack::Session::Tyrant.new(@incrementor, :tyrant_server => "localhost:6380").pool }.
+          should_not raise_error
+      end
+
+
+      it "creates a new cookie" do
+        pool = Rack::Session::Tyrant.new(@incrementor)
+        res = Rack::MockRequest.new(pool).get("/")
+        res["Set-Cookie"].should match(/#{@session_key}=/)
+        res.body.should eql('{"counter"=>1}')
+      end
+
+      it "determines session from a cookie" do
+        pool = Rack::Session::Tyrant.new(@incrementor)
+        req = Rack::MockRequest.new(pool)
+        res = req.get("/")
+        cookie = res["Set-Cookie"]
+        req.get("/", "HTTP_COOKIE" => cookie).body.should eql('{"counter"=>2}')
+        req.get("/", "HTTP_COOKIE" => cookie).body.should eql('{"counter"=>3}')
+      end
+
+      it "survives nonexistant cookies" do
+        bad_cookie = "rack.session=blsarghfasel"
+        pool = Rack::Session::Tyrant.new(@incrementor)
+        res = Rack::MockRequest.new(pool).get("/", "HTTP_COOKIE" => bad_cookie)
+        res.body.should eql('{"counter"=>1}')
+        cookie = res["Set-Cookie"][@session_match]
+        cookie.should_not match(/#{bad_cookie}/)
+      end
+
+      # Expire isn't supported by cabinet. Implement in ruby?
+      # it "should maintain freshness" do
+      #   pool = Rack::Session::Tyrant.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::Tyrant.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::Tyrant.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::Tyrant.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::Tyrant'
+        pool = Rack::Session::Tyrant.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
+
+      after(:all) do
+         Rack::Session::Tyrant.new(@incrementor).pool.clear
+      end
+    end
+  end
+end

data/spec/spec.opts

@@ -0,0 +1,3 @@
+--colour
+--loadby mtime
+--reverse

data/spec/spec_helper.rb

@@ -0,0 +1,13 @@
+$: << File.join(File.dirname(__FILE__), "/../lib")
+%W{rubygems spec rack/cache activesupport }.each { |l| require l }
+require 'tokyo_store'
+require 'rack/session/rufus_tyrant'
+require 'rack/session/tyrant'
+require 'rack/session/cabinet'
+require 'cache/tokyo_store'
+#require 'rack/cache/tokyo'
+
+#Simple class to test marshal
+class City;  attr_accessor :name, :pop;end
+
+Spec::Runner.configure do |config|;end

data/spec/tokyo_store_spec.rb

@@ -0,0 +1,43 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+if ENV['CABINET']
+  describe "TokyoStore" do
+    it "should store fragment cache" do
+      HDB.should_receive(:new).and_return(@mock_hdb = mock("HDB"))
+      @mock_hdb.should_receive(:open).with('data.tch', 6).and_return(true)
+      store = ActiveSupport::Cache.lookup_store :tokyo_store, "data.tch"
+      store.should be_kind_of ActiveSupport::Cache::TokyoStore
+    end
+
+    it "should fail" do
+      tokyo =  HDB.new
+      tokyo.open('data.tch')
+      HDB.should_not_receive(:new)
+      store = ActiveSupport::Cache.lookup_store :tokyo_store, tokyo
+      store.should be_kind_of ActiveSupport::Cache::TokyoStore
+    end
+
+    describe "Similar" do
+
+      before(:all) do
+        @cache = ActiveSupport::Cache::TokyoStore.new 'data.tcb'
+      end
+
+      it "test_should_read_and_write_strings" do
+        @cache.write('foo', 'bar')
+        @cache.read('foo').should eql('bar')
+      end
+
+      it "test_should_read_and_write_hash" do
+        @cache.write('foo', {:a => "b"})
+        @cache.read('foo').should eql({:a => "b"})
+      end
+
+      it "test_should_read_and_write_hash" do
+        @cache.write('foo', {:a => "b", :c => "d"})
+        @cache.read('foo').should eql({:a => "b", :c => "d"})
+      end
+    end
+
+  end
+end

data/tokyo_store.gemspec

@@ -0,0 +1,65 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{tokyo_store}
+  s.version = "0.3.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Marcos Piccinini"]
+  s.date = %q{2009-07-19}
+  s.email = %q{x@nofxx.com}
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "benchmark/cache.rb",
+     "benchmark/session.rb",
+     "lib/cache/tokyo_store.rb",
+     "lib/rack/cache/tokyo_entitystore.rb",
+     "lib/rack/cache/tokyo_metastore.rb",
+     "lib/rack/session/cabinet.rb",
+     "lib/rack/session/rufus_tyrant.rb",
+     "lib/rack/session/tyrant.rb",
+     "lib/tokyo_store.rb",
+     "spec/cache/tokyo_store_spec.rb",
+     "spec/rack/cache/tokyo_spec.rb",
+     "spec/rack/session/cabinet_spec.rb",
+     "spec/rack/session/rufus_tyrant_spec.rb",
+     "spec/rack/session/tyrant_spec.rb",
+     "spec/spec.opts",
+     "spec/spec_helper.rb",
+     "spec/tokyo_store_spec.rb",
+     "tokyo_store.gemspec"
+  ]
+  s.homepage = %q{http://github.com/nofxx/tokyo_store}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.4}
+  s.summary = %q{Tokyo Tyrant rails session store}
+  s.test_files = [
+    "spec/rack/cache/tokyo_spec.rb",
+     "spec/rack/session/rufus_tyrant_spec.rb",
+     "spec/rack/session/cabinet_spec.rb",
+     "spec/rack/session/tyrant_spec.rb",
+     "spec/tokyo_store_spec.rb",
+     "spec/cache/tokyo_store_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end

metadata

@@ -0,0 +1,85 @@
+--- !ruby/object:Gem::Specification 
+name: tokyo_store
+version: !ruby/object:Gem::Version 
+  version: 0.3.0
+platform: ruby
+authors: 
+- Marcos Piccinini
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-07-19 00:00:00 -03:00
+default_executable: 
+dependencies: []
+
+description: 
+email: x@nofxx.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- benchmark/cache.rb
+- benchmark/session.rb
+- lib/cache/tokyo_store.rb
+- lib/rack/cache/tokyo_entitystore.rb
+- lib/rack/cache/tokyo_metastore.rb
+- lib/rack/session/cabinet.rb
+- lib/rack/session/rufus_tyrant.rb
+- lib/rack/session/tyrant.rb
+- lib/tokyo_store.rb
+- spec/cache/tokyo_store_spec.rb
+- spec/rack/cache/tokyo_spec.rb
+- spec/rack/session/cabinet_spec.rb
+- spec/rack/session/rufus_tyrant_spec.rb
+- spec/rack/session/tyrant_spec.rb
+- spec/spec.opts
+- spec/spec_helper.rb
+- spec/tokyo_store_spec.rb
+- tokyo_store.gemspec
+has_rdoc: true
+homepage: http://github.com/nofxx/tokyo_store
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+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: Tokyo Tyrant rails session store
+test_files: 
+- spec/rack/cache/tokyo_spec.rb
+- spec/rack/session/rufus_tyrant_spec.rb
+- spec/rack/session/cabinet_spec.rb
+- spec/rack/session/tyrant_spec.rb
+- spec/tokyo_store_spec.rb
+- spec/cache/tokyo_store_spec.rb
+- spec/spec_helper.rb