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
|
# frozen_string_literal: true
module TestProf
# Add ability to run only a specified number of example groups (randomly selected)
module MinitestSample
# Do not add these classes to resulted sample
CORE_RUNNABLES = [
Minitest::Test,
Minitest::Unit::TestCase,
Minitest::Spec
].freeze
class << self
def suites
# Make sure that sample contains only _real_ suites
Minitest::Runnable.runnables
.reject { |suite| CORE_RUNNABLES.include?(suite) }
end
def sample_groups(sample_size)
saved_suites = suites
Minitest::Runnable.reset
saved_suites.sample(sample_size).each { |r| Minitest::Runnable.runnables << r }
end
def sample_examples(sample_size)
all_examples = suites.flat_map do |runnable|
runnable.runnable_methods.map { |method| [runnable, method] }
end
sample = all_examples.sample(sample_size)
# Filter examples by overriding #runnable_methods for all suites
suites.each do |runnable|
runnable.define_singleton_method(:runnable_methods) do
super() & sample.select { |ex| ex.first.equal?(runnable) }.map(&:last)
end
end
end
def call
if ENV["SAMPLE"]
::TestProf::MinitestSample.sample_examples(ENV["SAMPLE"].to_i)
elsif ENV["SAMPLE_GROUPS"]
::TestProf::MinitestSample.sample_groups(ENV["SAMPLE_GROUPS"].to_i)
end
end
end
end
end
|
