rake/test/test_rake_multi_task.rb

86 lines
2.2 KiB
Ruby

require File.expand_path("../helper", __FILE__)
require "thread"
class TestRakeMultiTask < Rake::TestCase
include Rake
include Rake::DSL
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
end