configurator: allow hooks to be passed callable objects

Premade lambda/proc/Proc objects may all be passed, to the
hooks, not just anonymous blocks.

1 files changed, 15 insertions, 0 deletions

diff --git a/test/unit/test_configurator.rb b/test/unit/test_configurator.rb
index 284d727..46fc248 100644
--- a/test/unit/test_configurator.rb
+++ b/test/unit/test_configurator.rb
@@ -91,4 +91,19 @@ class TestConfigurator < Test::Unit::TestCase
     end
   end
 
+  def test_after_fork_proc
+    [ proc { |a,b| }, Proc.new { |a,b| }, lambda { |a,b| } ].each do |my_proc|
+      Unicorn::Configurator.new(:after_fork => my_proc).commit!(self)
+      assert_equal my_proc, @after_fork
+    end
+  end
+
+  def test_after_fork_wrong_arity
+    [ proc { |a| }, Proc.new { }, lambda { |a,b,c| } ].each do |my_proc|
+      assert_raises(ArgumentError) do
+        Unicorn::Configurator.new(:after_fork => my_proc)
+      end
+    end
+  end
+
 end