-
Notifications
You must be signed in to change notification settings - Fork 103
/
testsuite.rb
175 lines (153 loc) · 4.8 KB
/
testsuite.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
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
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
#--
#
# Author:: Nathaniel Talbott.
# Copyright:: Copyright (c) 2000-2003 Nathaniel Talbott. All rights reserved.
# Copyright:: Copyright (c) 2008-2011 Kouhei Sutou. All rights reserved.
# License:: Ruby license.
require 'test/unit/error'
module Test
module Unit
# A collection of tests which can be #run.
#
# Note: It is easy to confuse a TestSuite instance with
# something that has a static suite method; I know because _I_
# have trouble keeping them straight. Think of something that
# has a suite method as simply providing a way to get a
# meaningful TestSuite instance.
class TestSuite
attr_reader :name, :tests, :test_case, :start_time, :elapsed_time
# Test suite that has higher priority is ran prior to
# test suites that have lower priority.
attr_accessor :priority
STARTED = name + "::STARTED"
STARTED_OBJECT = name + "::STARTED::OBJECT"
FINISHED = name + "::FINISHED"
FINISHED_OBJECT = name + "::FINISHED::OBJECT"
# Creates a new TestSuite with the given name.
def initialize(name="Unnamed TestSuite", test_case=nil)
@name = name
@tests = []
@test_case = test_case
@n_tests = 0
@priority = 0
@start_time = nil
@elapsed_time = nil
@passed = true
end
# Runs the tests and/or suites contained in this
# TestSuite.
def run(result, &progress_block)
@start_time = Time.now
yield(STARTED, name)
yield(STARTED_OBJECT, self)
run_startup(result)
while test = @tests.shift
@n_tests += test.size
run_test(test, result, &progress_block)
@passed = false unless test.passed?
end
ensure
begin
run_shutdown(result)
ensure
@elapsed_time = Time.now - @start_time
yield(FINISHED, name)
yield(FINISHED_OBJECT, self)
end
end
# Adds the test to the suite.
def <<(test)
@tests << test
self
end
def delete(test)
@tests.delete(test)
end
def delete_tests(tests)
@tests -= tests
end
# Retuns the rolled up number of tests in this suite;
# i.e. if the suite contains other suites, it counts the
# tests within those suites, not the suites themselves.
def size
total_size = @n_tests
@tests.each { |test| total_size += test.size }
total_size
end
def empty?
size.zero?
end
# Overridden to return the name given the suite at
# creation.
def to_s
@name
end
# It's handy to be able to compare TestSuite instances.
def ==(other)
return false unless(other.kind_of?(self.class))
return false unless(@name == other.name)
@tests == other.tests
end
def passed?
@passed
end
private
def run_startup(result)
return if @test_case.nil? or !@test_case.respond_to?(:startup)
begin
@test_case.startup
rescue Exception
raise unless handle_exception($!, result)
end
end
def run_test(test, result)
finished_is_yielded = false
finished_object_is_yielded = false
previous_event_name = nil
test.run(result) do |event_name, *args|
case previous_event_name
when Test::Unit::TestCase::STARTED
if event_name != Test::Unit::TestCase::STARTED_OBJECT
yield(Test::Unit::TestCase::STARTED_OBJECT, test)
end
when Test::Unit::TestCase::FINISHED
if event_name != Test::Unit::TestCase::FINISHED_OBJECT
yield(Test::Unit::TestCase::FINISHED_OBJECT, test)
end
finished_object_is_yielded = true
end
case event_name
when Test::Unit::TestCase::STARTED
finished_is_yielded = false
finished_object_is_yielded = false
when Test::Unit::TestCase::FINISHED
finished_is_yielded = true
end
previous_event_name = event_name
yield(event_name, *args)
end
if finished_is_yielded and not finished_object_is_yielded
yield(Test::Unit::TestCase::FINISHED_OBJECT, test)
end
end
def run_shutdown(result)
return if @test_case.nil? or !@test_case.respond_to?(:shutdown)
begin
@test_case.shutdown
rescue Exception
raise unless handle_exception($!, result)
end
end
def handle_exception(exception, result)
case exception
when *ErrorHandler::PASS_THROUGH_EXCEPTIONS
false
else
result.add_error(Error.new(@test_case.name, exception))
@passed = false
true
end
end
end
end
end