aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/test/cases/identity_map/middleware_test.rb
blob: 60dcad4586841f5df11f3b981745280446e89371 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
require "cases/helper"

module ActiveRecord
  module IdentityMap
    class MiddlewareTest < ActiveRecord::TestCase
      def setup
        super
        @enabled = IdentityMap.enabled
        IdentityMap.enabled = false
      end

      def teardown
        super
        IdentityMap.enabled = @enabled
        IdentityMap.clear
      end

      def test_delegates
        called = false
        mw = Middleware.new lambda { |env|
          called = true
        }
        mw.call({})
        assert called, 'middleware delegated'
      end

      def test_im_enabled_during_delegation
        mw = Middleware.new lambda { |env|
          assert IdentityMap.enabled?, 'identity map should be enabled'
        }
        mw.call({})
      end

      class Enum < Struct.new(:iter)
        def each(&b)
          iter.call(&b)
        end
      end

      def test_im_enabled_during_body_each
        mw = Middleware.new lambda { |env|
          [200, {}, Enum.new(lambda { |&b|
            assert IdentityMap.enabled?, 'identity map should be enabled'
            b.call "hello"
          })]
        }
        body = mw.call({}).last
        body.each { |x| assert_equal 'hello', x }
      end

      def test_im_disabled_after_body_close
        mw = Middleware.new lambda { |env| [200, {}, []] }
        body = mw.call({}).last
        assert IdentityMap.enabled?, 'identity map should be enabled'
        body.close
        assert !IdentityMap.enabled?, 'identity map should be disabled'
      end

      def test_im_cleared_after_body_close
        mw = Middleware.new lambda { |env| [200, {}, []] }
        body = mw.call({}).last

        IdentityMap.repository['hello'] = 'world'
        assert !IdentityMap.repository.empty?, 'repo should not be empty'

        body.close
        assert IdentityMap.repository.empty?, 'repo should be empty'
      end
    end
  end
end