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
|
module Test
module Unit
module Collector
def initialize
@filters = []
end
def filter=(filters)
@filters = case(filters)
when Proc
[filters]
when Array
filters
end
end
def add_suite(destination, suite)
to_delete = suite.tests.find_all do |test|
test.is_a?(TestCase) and !include?(test)
end
suite.delete_tests(to_delete)
destination << suite unless suite.empty?
end
def add_test_cases(suite, test_cases)
children_map = {}
test_cases.each do |test_case|
ancestor_classes = test_case.ancestors.find_all do |ancestor|
ancestor.is_a?(Class)
end
parent = ancestor_classes[1]
children_map[parent] ||= []
children_map[parent] << test_case
end
root_test_cases = children_map.keys - test_cases
root_test_cases.each do |root_test_case|
add_test_case(suite, root_test_case, children_map)
end
end
def include?(test)
return true if(@filters.empty?)
@filters.each do |filter|
return false if filter[test] == false
end
true
end
def sort(suites)
suites.sort_by do |suite|
[suite.priority, suite.name || suite.to_s]
end
end
private
def add_test_case(suite, test_case, children_map)
children = children_map[test_case]
return if children.nil?
sub_suites = []
children.each do |child|
sub_suite = child.suite
add_test_case(sub_suite, child, children_map)
add_suite(sub_suites, sub_suite)
end
sort(sub_suites).each do |sub_suite|
suite << sub_suite
end
end
def adjust_ractor_tests(suite)
return if suite.nil?
ractor_suites = extract_ractor_tests(suite)
ractor_suites.each do |ractor_suite|
suite << ractor_suite
end
end
def extract_ractor_tests(suite)
ractor_suites = []
ractor_tests = []
suite.tests.each do |test|
case test
when TestSuite
ractor_suites.concat(extract_ractor_tests(test))
else
next unless test[:ractor]
ractor_tests << test
end
end
unless ractor_tests.empty?
suite.delete_tests(ractor_tests)
ractor_suite = TestSuite.new(suite.name, suite.test_case)
ractor_tests.each do |ractor_test|
ractor_suite << ractor_test
end
ractor_suites << ractor_suite
end
ractor_suites
end
end
end
end
|