diff options
author | Yann Vanhalewyn <yann.vanhalewyn@gmail.com> | 2016-10-24 14:47:14 +0200 |
---|---|---|
committer | Yann Vanhalewyn <yann.vanhalewyn@gmail.com> | 2016-10-24 15:26:00 +0200 |
commit | c4244cc55b21061cecc00a03f056689ac2469d1e (patch) | |
tree | de769d69e0d08f66952cf31fcf99b44313545c39 | |
parent | 7dd7646a932419f4d308bf2fbc91f4e77edfd4cb (diff) | |
download | rack-c4244cc55b21061cecc00a03f056689ac2469d1e.tar.gz |
Improve and test SessionHash#fetch.
-rw-r--r-- | lib/rack/session/abstract/id.rb | 12 | ||||
-rw-r--r-- | test/spec_session_abstract_session_hash.rb | 13 |
2 files changed, 24 insertions, 1 deletions
diff --git a/lib/rack/session/abstract/id.rb b/lib/rack/session/abstract/id.rb index 4c139637..ca1a2628 100644 --- a/lib/rack/session/abstract/id.rb +++ b/lib/rack/session/abstract/id.rb @@ -18,6 +18,8 @@ module Rack include Enumerable attr_writer :id + Unspecified = Object.new + def self.find(req) req.get_header RACK_SESSION end @@ -54,7 +56,15 @@ module Rack load_for_read! @data[key.to_s] end - alias :fetch :[] + + def fetch(key, default=Unspecified, &block) + load_for_read! + if default == Unspecified + @data.fetch(key.to_s, &block) + else + @data.fetch(key.to_s, default, &block) + end + end def has_key?(key) load_for_read! diff --git a/test/spec_session_abstract_session_hash.rb b/test/spec_session_abstract_session_hash.rb index 6d73a80a..16b06c7b 100644 --- a/test/spec_session_abstract_session_hash.rb +++ b/test/spec_session_abstract_session_hash.rb @@ -25,4 +25,17 @@ describe Rack::Session::Abstract::SessionHash do assert_equal [:bar, :qux], hash.values end + describe "#fetch" do + it "returns value for a matching key" do + assert_equal :bar, hash.fetch(:foo) + end + + it "works with a default value" do + assert_equal :default, hash.fetch(:unknown, :default) + end + + it "works with a block" do + assert_equal :default, hash.fetch(:unkown) { :default } + end + end end |