require "abstract_unit" module ActionDispatch class FlashHashTest < ActiveSupport::TestCase def setup @hash = Flash::FlashHash.new end def test_set_get @hash[:foo] = "zomg" assert_equal "zomg", @hash[:foo] end def test_keys assert_equal [], @hash.keys @hash["foo"] = "zomg" assert_equal ["foo"], @hash.keys @hash["bar"] = "zomg" assert_equal ["foo", "bar"].sort, @hash.keys.sort end def test_update @hash["foo"] = "bar" @hash.update("foo" => "baz", "hello" => "world") assert_equal "baz", @hash["foo"] assert_equal "world", @hash["hello"] end def test_key @hash["foo"] = "bar" assert @hash.key?("foo") assert @hash.key?(:foo) assert_not @hash.key?("bar") assert_not @hash.key?(:bar) end def test_delete @hash["foo"] = "bar" @hash.delete "foo" assert !@hash.key?("foo") assert_nil @hash["foo"] end def test_to_hash @hash["foo"] = "bar" assert_equal({ "foo" => "bar" }, @hash.to_hash) @hash.to_hash["zomg"] = "aaron" assert !@hash.key?("zomg") assert_equal({ "foo" => "bar" }, @hash.to_hash) end def test_to_session_value @hash["foo"] = "bar" assert_equal({ "discard" => [], "flashes" => { "foo" => "bar" } }, @hash.to_session_value) @hash.now["qux"] = 1 assert_equal({ "flashes" => { "foo" => "bar" }, "discard" => [] }, @hash.to_session_value) @hash.discard("foo") assert_equal(nil, @hash.to_session_value) @hash.sweep assert_equal(nil, @hash.to_session_value) end def test_from_session_value # {"session_id"=>"f8e1b8152ba7609c28bbb17ec9263ba7", "flash"=>#, @closed=false, @flashes={"greeting"=>"Hello", "farewell"=>"Goodbye"}, @now=nil>} rails_3_2_cookie = "BAh7B0kiD3Nlc3Npb25faWQGOgZFRkkiJWY4ZTFiODE1MmJhNzYwOWMyOGJiYjE3ZWM5MjYzYmE3BjsAVEkiCmZsYXNoBjsARm86JUFjdGlvbkRpc3BhdGNoOjpGbGFzaDo6Rmxhc2hIYXNoCToKQHVzZWRvOghTZXQGOgpAaGFzaHsGSSINZmFyZXdlbGwGOwBUVDoMQGNsb3NlZEY6DUBmbGFzaGVzewdJIg1ncmVldGluZwY7AFRJIgpIZWxsbwY7AFRJIg1mYXJld2VsbAY7AFRJIgxHb29kYnllBjsAVDoJQG5vdzA=" session = Marshal.load(Base64.decode64(rails_3_2_cookie)) hash = Flash::FlashHash.from_session_value(session["flash"]) assert_equal({ "greeting" => "Hello" }, hash.to_hash) assert_equal(nil, hash.to_session_value) end def test_from_session_value_on_json_serializer decrypted_data = "{ \"session_id\":\"d98bdf6d129618fc2548c354c161cfb5\", \"flash\":{\"discard\":[\"farewell\"], \"flashes\":{\"greeting\":\"Hello\",\"farewell\":\"Goodbye\"}} }" session = ActionDispatch::Cookies::JsonSerializer.load(decrypted_data) hash = Flash::FlashHash.from_session_value(session["flash"]) assert_equal({ "greeting" => "Hello" }, hash.to_hash) assert_equal(nil, hash.to_session_value) assert_equal "Hello", hash[:greeting] assert_equal "Hello", hash["greeting"] end def test_empty? assert @hash.empty? @hash["zomg"] = "bears" assert !@hash.empty? @hash.clear assert @hash.empty? end def test_each @hash["hello"] = "world" @hash["foo"] = "bar" things = [] @hash.each do |k, v| things << [k, v] end assert_equal([%w{ hello world }, %w{ foo bar }].sort, things.sort) end def test_replace @hash["hello"] = "world" @hash.replace("omg" => "aaron") assert_equal({ "omg" => "aaron" }, @hash.to_hash) end def test_discard_no_args @hash["hello"] = "world" @hash.discard @hash.sweep assert_equal({}, @hash.to_hash) end def test_discard_one_arg @hash["hello"] = "world" @hash["omg"] = "world" @hash.discard "hello" @hash.sweep assert_equal({ "omg" => "world" }, @hash.to_hash) end def test_keep_sweep @hash["hello"] = "world" @hash.sweep assert_equal({ "hello" => "world" }, @hash.to_hash) end def test_update_sweep @hash["hello"] = "world" @hash.update("hi" => "mom") @hash.sweep assert_equal({ "hello" => "world", "hi" => "mom" }, @hash.to_hash) end def test_update_delete_sweep @hash["hello"] = "world" @hash.delete "hello" @hash.update("hello" => "mom") @hash.sweep assert_equal({ "hello" => "mom" }, @hash.to_hash) end def test_delete_sweep @hash["hello"] = "world" @hash["hi"] = "mom" @hash.delete "hi" @hash.sweep assert_equal({ "hello" => "world" }, @hash.to_hash) end def test_clear_sweep @hash["hello"] = "world" @hash.clear @hash.sweep assert_equal({}, @hash.to_hash) end def test_replace_sweep @hash["hello"] = "world" @hash.replace("hi" => "mom") @hash.sweep assert_equal({ "hi" => "mom" }, @hash.to_hash) end def test_discard_then_add @hash["hello"] = "world" @hash["omg"] = "world" @hash.discard "hello" @hash["hello"] = "world" @hash.sweep assert_equal({ "omg" => "world", "hello" => "world" }, @hash.to_hash) end def test_keep_all_sweep @hash["hello"] = "world" @hash["omg"] = "world" @hash.discard "hello" @hash.keep @hash.sweep assert_equal({ "omg" => "world", "hello" => "world" }, @hash.to_hash) end def test_double_sweep @hash["hello"] = "world" @hash.sweep assert_equal({ "hello" => "world" }, @hash.to_hash) @hash.sweep assert_equal({}, @hash.to_hash) end end end