rails--rails/actionpack/test/dispatch/request/session_test.rb

require 'abstract_unit'
require 'action_dispatch/middleware/session/abstract_store'

module ActionDispatch
  class Request
    class SessionTest < ActiveSupport::TestCase
      def test_create_adds_itself_to_env
        env = {}
        s = Session.create(store, env, {})
        assert_equal s, env[Rack::Session::Abstract::ENV_SESSION_KEY]
      end

      def test_to_hash
        env = {}
        s = Session.create(store, env, {})
        s['foo'] = 'bar'
        assert_equal 'bar', s['foo']
        assert_equal({'foo' => 'bar'}, s.to_hash)
      end

      def test_create_merges_old
        env = {}
        s = Session.create(store, env, {})
        s['foo'] = 'bar'

        s1 = Session.create(store, env, {})
        assert_not_equal s, s1
        assert_equal 'bar', s1['foo']
      end

      def test_find
        env = {}
        assert_nil Session.find(env)

        s = Session.create(store, env, {})
        assert_equal s, Session.find(env)
      end

      def test_keys
        env = {}
        s = Session.create(store, env, {})
        s['rails'] = 'ftw'
        s['adequate'] = 'awesome'
        assert_equal %w[rails adequate], s.keys
      end

      def test_values
        env = {}
        s = Session.create(store, env, {})
        s['rails'] = 'ftw'
        s['adequate'] = 'awesome'
        assert_equal %w[ftw awesome], s.values
      end

      def test_clear
        env = {}
        s = Session.create(store, env, {})
        s['rails'] = 'ftw'
        s['adequate'] = 'awesome'
        s.clear
        assert_equal([], s.values)
      end

      def test_fetch
        session = Session.create(store, {}, {})

        session['one'] = '1'
        assert_equal '1', session.fetch(:one)

        assert_equal '2', session.fetch(:two, '2')
        assert_nil session.fetch(:two, nil)

        assert_equal 'three', session.fetch(:three) {|el| el.to_s }

        assert_raise KeyError do
          session.fetch(:three)
        end
      end

      private
      def store
        Class.new {
          def load_session(env); [1, {}]; end
          def session_exists?(env); true; end
        }.new
      end
    end
  end
end