Handle for SIGTERM

lib/rake/application.rb

@@ -94,6 +94,8 @@ def init(app_name="rake", argv = ARGV)
           # Backward compatibility for capistrano
           args = handle_options
         end
+
+        setup_signal_handling
         load_debug_at_stop_feature
         collect_command_line_tasks(args)
       end
@@ -857,5 +859,12 @@ def set_default_options # :nodoc:
       options.trace_rules                = false
     end
 
+    def setup_signal_handling
+      Signal.trap("TERM") do
+        puts "SIGTERM received, exiting..."
+        exit 143 # 128 + Signal.list["TERM"] (15)
+      end
+    end
+
   end
 end

test/support/rakefile_definitions.rb

@@ -514,4 +514,18 @@ def rakefile_stand_alone_filelist
       io << "puts FL\n"
     end
   end
+
+  def rakefile_with_long_running_task
+    rakefile <<-TEST_TASK
+  require 'rake/testtask'
+
+  task :default => :test
+  Rake::TestTask.new(:test) do |t|
+  t.test_files = ['a_test.rb']
+  end
+    TEST_TASK
+    open "a_test.rb", "w" do |io|
+       io << "sleep 20"
+     end
+  end
 end

test/test_rake_functional.rb

@@ -516,6 +516,20 @@ def test_stand_alone_filelist
     assert_equal 0, @exit.exitstatus unless uncertain_exit_status?
   end
 
+  # Test that SIGTERM is handled gracefully
+  def test_sigterm_handling
+    if !jruby? && can_detect_signals?
+      rakefile_with_long_running_task
+      pid = Process.spawn("rake")
+      sleep 1
+      Process.kill("TERM", pid)
+      _, status = Process.wait2(pid)
+      assert_equal(143, status.exitstatus, "Process should exit with status 143")
+    else
+      omit "Signal detection seems broken on this system"
+    end
+  end
+
   private
 
   # We are unable to accurately verify that Rake returns a proper