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
|
