activerecord/test/cases/identity_map/middleware_test.rb


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

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
      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
    end
  end
end