-
Notifications
You must be signed in to change notification settings - Fork 614
/
test_rake_multi_task.rb
106 lines (88 loc) · 2.57 KB
/
test_rake_multi_task.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
# frozen_string_literal: true
require File.expand_path("../helper", __FILE__)
require "thread"
class TestRakeMultiTask < Rake::TestCase
include Rake
def setup
super
Task.clear
@runs = Array.new
@mutex = Mutex.new
end
def teardown
Rake.application.thread_pool.join
super
end
def add_run(obj)
@mutex.synchronize do
@runs << obj
end
end
def test_running_multitasks
task :a do 3.times do |i| add_run("A#{i}"); sleep 0.01; end end
task :b do 3.times do |i| add_run("B#{i}"); sleep 0.01; end end
multitask both: [:a, :b]
Task[:both].invoke
assert_equal 6, @runs.size
assert @runs.index("A0") < @runs.index("A1")
assert @runs.index("A1") < @runs.index("A2")
assert @runs.index("B0") < @runs.index("B1")
assert @runs.index("B1") < @runs.index("B2")
end
def test_all_multitasks_wait_on_slow_prerequisites
task :slow do 3.times do |i| add_run("S#{i}"); sleep 0.05 end end
task a: [:slow] do 3.times do |i| add_run("A#{i}"); sleep 0.01 end end
task b: [:slow] do 3.times do |i| add_run("B#{i}"); sleep 0.01 end end
multitask both: [:a, :b]
Task[:both].invoke
assert_equal 9, @runs.size
assert @runs.index("S0") < @runs.index("S1")
assert @runs.index("S1") < @runs.index("S2")
assert @runs.index("S2") < @runs.index("A0")
assert @runs.index("S2") < @runs.index("B0")
assert @runs.index("A0") < @runs.index("A1")
assert @runs.index("A1") < @runs.index("A2")
assert @runs.index("B0") < @runs.index("B1")
assert @runs.index("B1") < @runs.index("B2")
end
def test_multitasks_with_parameters
task :a, [:arg] do |t, args| add_run(args[:arg]) end
multitask :b, [:arg] => [:a] do |t, args| add_run(args[:arg] + "mt") end
Task[:b].invoke "b"
assert @runs[0] == "b"
assert @runs[1] == "bmt"
end
def test_cross_thread_prerequisite_failures
failed = false
multitask :fail_once do
fail_now = !failed
failed = true
raise "failing once" if fail_now
end
task a: :fail_once
task b: :fail_once
assert_raises RuntimeError do
Rake::Task[:a].invoke
end
assert_raises RuntimeError do
Rake::Task[:b].invoke
end
end
def test_task_not_executed_if_dependant_task_failed_concurrently
multitask default: [:one, :two]
task :one do
raise
end
task_two_was_executed = false
task two: :one do
task_two_was_executed = true
end
begin
Rake::Task[:default].invoke
rescue RuntimeError
ensure
sleep 0.5
refute task_two_was_executed
end
end
end