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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
require 'minitest'
# Add support for around and before_all/after_all/around_all hooks to
# minitest spec classes.
module Minitest::Hooks
# Add the class methods to the class. Also, include an additional
# module in the class that before(:all) and after(:all) methods
# work on a class that directly includes this module.
def self.included(mod)
super
mod.instance_exec do
extend(Minitest::Hooks::ClassMethods)
end
end
# Empty method, necessary so that super calls in spec subclasses work.
def before_all
end
# Empty method, necessary so that super calls in spec subclasses work.
def after_all
end
# Method that just yields, so that super calls in spec subclasses work.
def around_all
yield
end
# Method that just yields, so that super calls in spec subclasses work.
def around
yield
end
# Run around hook inside, since time_it is run around every spec.
def time_it
super do
around do
yield
end
end
end
end
module Minitest::Hooks::ClassMethods
# Object used to get an empty new instance, as new by default will return
# a dup of the singleton instance.
NEW = Object.new.freeze
# Unless name is NEW, return a dup singleton instance.
def new(name)
if name.equal?(NEW)
return super('around_all')
end
instance = @instance.dup
instance.name = name
instance.failures = []
instance
end
# When running the specs in the class, first create a singleton instance, the singleton is
# used to implement around_all/before_all/after_all hooks, and each spec will run as a
# dup of the singleton instance.
def with_info_handler(reporter, &block)
@instance = new(NEW)
@instance.time = 0
@instance.name = "around_all"
begin
@instance.around_all do
begin
@instance.capture_exceptions do
@instance.name = "before_all"
@instance.before_all
end
if @instance.failure
failed = true
_record_minitest_hooks_error(reporter, @instance)
else
super(reporter, &block)
end
ensure
@instance.capture_exceptions do
@instance.name = "after_all" unless failed
@instance.after_all
end
if @instance.failure && !failed
failed = true
_record_minitest_hooks_error(reporter, @instance)
end
@instance.name = "around_all" unless failed
end
end
rescue => e
@instance.capture_exceptions do
raise e
end
_record_minitest_hooks_error(reporter, @instance)
end
end
# If type is :all, set the around_all hook, otherwise set the around hook.
def around(type=nil, &block)
meth = type == :all ? :around_all : :around
define_method(meth, &block)
end
# If type is :all, set the before_all hook instead of the before hook.
def before(type=nil, &block)
if type == :all
define_method(:before_all) do
super()
instance_exec(&block)
end
nil
else
super
end
end
# If type is :all, set the after_all hook instead of the after hook.
def after(type=nil, &block)
if type == :all
define_method(:after_all) do
instance_exec(&block)
super()
end
nil
else
super
end
end
private
def _record_minitest_hooks_error(reporter, instance)
# In Minitest 5.11+, use Minitest::Result for wrapping the object to send
# to the reporter.
if(defined?(Minitest::Result))
result = Minitest::Result.from(instance)
# :nocov:
else
result = instance
# :nocov:
end
reporter.record result
end
end
|
