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
|
module Knapsack
class Tracker
include Singleton
attr_reader :global_time, :test_files_with_time
attr_writer :test_path
def initialize
set_defaults
end
def config(opts={})
@config ||= default_config
@config.merge!(opts)
end
def reset!
set_defaults
end
def start_timer
@start_time = now_without_mock_time.to_f
end
def stop_timer
@execution_time = now_without_mock_time.to_f - @start_time
update_global_time
update_test_file_time
@execution_time
end
def test_path
raise("test_path needs to be set by Knapsack Adapter's bind method") unless @test_path
@test_path.sub(/^\.\//, '')
end
def time_exceeded?
global_time > max_node_time_execution
end
def max_node_time_execution
report_distributor.node_time_execution + config[:time_offset_in_seconds]
end
def exceeded_time
global_time - max_node_time_execution
end
private
def default_config
{
enable_time_offset_warning: Config::Tracker.enable_time_offset_warning,
time_offset_in_seconds: Config::Tracker.time_offset_in_seconds,
generate_report: Config::Tracker.generate_report
}
end
def set_defaults
@global_time = 0
@test_files_with_time = {}
@test_path = nil
end
def update_global_time
@global_time += @execution_time
end
def update_test_file_time
@test_files_with_time[test_path] ||= 0
@test_files_with_time[test_path] += @execution_time
end
def report_distributor
@report_distributor ||= Knapsack::Distributors::ReportDistributor.new({
report: Knapsack.report.open,
test_file_pattern: Knapsack::Config::Env.test_file_pattern || Knapsack.report.config[:test_file_pattern],
ci_node_total: Knapsack::Config::Env.ci_node_total,
ci_node_index: Knapsack::Config::Env.ci_node_index
})
end
def now_without_mock_time
if defined?(Timecop)
Time.now_without_mock_time
else
Time.raw_now
end
end
end
end
|
