god 0.11.0 → 0.12.0

data/test/configs/test.rb

@@ -12,17 +12,17 @@ God.watch do |w|
   w.stop = "mongrel_rails stop -P ./log/mongrel.pid -c #{RAILS_ROOT}"
   w.group = 'mongrels'
   w.pid_file = File.join(RAILS_ROOT, "log/mongrel.pid")
-  
+
   # clean pid files before start if necessary
   w.behavior(:clean_pid_file)
-  
+
   # determine the state on startup
   w.transition(:init, { true => :up, false => :start }) do |on|
     on.condition(:process_running) do |c|
       c.running = true
     end
   end
-  
+
   # determine when process has finished starting
   w.transition([:start, :restart], :up) do |on|
     on.condition(:process_running) do |c|
@@ -34,7 +34,7 @@ God.watch do |w|
   w.transition(:up, :start) do |on|
     on.condition(:process_exits)
   end
-  
+
   # restart if memory or cpu is too high
   w.transition(:up, :restart) do |on|
     on.condition(:memory_usage) do |c|
@@ -42,13 +42,13 @@ God.watch do |w|
       c.above = 50.megabytes
       c.times = [3, 5]
     end
-    
+
     on.condition(:cpu_usage) do |c|
       c.interval = 1
       c.above = 10.percent
       c.times = [3, 5]
     end
-    
+
     on.condition(:http_response_code) do |c|
       c.host = 'localhost'
       c.port = port
@@ -58,4 +58,4 @@ God.watch do |w|
       c.times = [3, 5]
     end
   end
-end
+end

data/test/helper.rb

@@ -8,7 +8,7 @@ require 'set'
 
 include God
 
-if Process.uid != 0
+if Process.uid != 0 and RbConfig::CONFIG['host_os'] == "linux"
   abort <<-EOF
 \n
 *********************************************************************
@@ -39,13 +39,13 @@ module God
         true
       end
     end
-  
+
     class FakePollCondition < PollCondition
       def test
         true
       end
     end
-  
+
     class FakeEventCondition < EventCondition
       def register
       end
@@ -53,7 +53,7 @@ module God
       end
     end
   end
-  
+
   module Behaviors
     class FakeBehavior < Behavior
       def before_start
@@ -64,15 +64,15 @@ module God
       end
     end
   end
-  
+
   module Contacts
     class FakeContact < Contact
     end
-    
+
     class InvalidContact
     end
   end
-  
+
   def self.reset
     self.watches = nil
     self.groups = nil
@@ -129,7 +129,7 @@ class Object
     def initialize(ref)
       @ref = ref
     end
-  
+
     def method_missing(sym, *args)
       @ref.__send__(sym, *args)
     end

data/test/test_behavior.rb

@@ -4,15 +4,15 @@ class TestBehavior < Test::Unit::TestCase
   def test_generate_should_return_an_object_corresponding_to_the_given_type
     assert_equal Behaviors::FakeBehavior, Behavior.generate(:fake_behavior, nil).class
   end
-  
+
   def test_generate_should_raise_on_invalid_type
     assert_raise NoSuchBehaviorError do
       Behavior.generate(:foo, nil)
     end
   end
-  
+
   def test_complain
     SysLogger.expects(:log).with(:error, 'foo')
     assert !Behavior.allocate.bypass.complain('foo')
   end
-end
+end

data/test/test_campfire.rb

@@ -1,5 +1,4 @@
 require File.dirname(__FILE__) + '/helper'
-require 'tinder'
 
 class TestCampfire < Test::Unit::TestCase
   def setup
@@ -20,4 +19,4 @@ class TestCampfire < Test::Unit::TestCase
     Marshmallow::Connection.any_instance.expects(:speak).with('danger', body)
     @campfire.notify('msg', time, 'prio', 'cat', 'host')
   end
-end
+end

data/test/test_condition.rb

@@ -4,19 +4,19 @@ class BadlyImplementedCondition < PollCondition
 end
 
 class TestCondition < Test::Unit::TestCase
-  
+
   # generate
-  
+
   def test_generate_should_return_an_object_corresponding_to_the_given_type
     assert_equal Conditions::ProcessRunning, Condition.generate(:process_running, nil).class
   end
-  
+
   def test_generate_should_raise_on_invalid_type
     assert_raise NoSuchConditionError do
       Condition.generate(:foo, nil)
     end
   end
-  
+
   def test_generate_should_abort_on_event_condition_without_loaded_event_system
     God::EventHandler.stubs(:operational?).returns(false)
     assert_abort do
@@ -24,25 +24,25 @@ class TestCondition < Test::Unit::TestCase
       Condition.generate(:process_exits, nil).class
     end
   end
-  
+
   def test_generate_should_return_a_good_error_message_for_invalid_types
     emsg = "No Condition found with the class name God::Conditions::FooBar"
     rmsg = nil
-    
+
     begin
       Condition.generate(:foo_bar, nil)
     rescue => e
       rmsg = e.message
     end
-    
+
     assert_equal emsg, rmsg
   end
-  
+
   # test
-  
+
   def test_test_should_raise_if_not_defined_in_subclass
     c = BadlyImplementedCondition.new
-    
+
     assert_raise AbstractMethodNotOverriddenError do
       c.test
     end

data/test/test_conditions_disk_usage.rb

@@ -2,49 +2,49 @@ require File.dirname(__FILE__) + '/helper'
 
 class TestConditionsDiskUsage < Test::Unit::TestCase
   # valid?
-  
+
   def test_valid_should_return_false_if_no_above_given
     c = Conditions::DiskUsage.new
     c.mount_point = '/'
     c.watch = stub(:name => 'foo')
     assert_equal false, c.valid?
   end
-  
+
   def test_valid_should_return_false_if_no_mount_point_given
     c = Conditions::DiskUsage.new
     c.above = 90
     c.watch = stub(:name => 'foo')
     assert_equal false, c.valid?
   end
-  
+
   def test_valid_should_return_true_if_required_options_all_set
     c = Conditions::DiskUsage.new
     c.above = 90
     c.mount_point = '/'
     c.watch = stub(:name => 'foo')
-    
+
     assert_equal true, c.valid?
   end
-  
+
   # test
-  
+
   def test_test_should_return_true_if_above_limit
     c = Conditions::DiskUsage.new
     c.above = 90
     c.mount_point = '/'
-    
+
     c.expects(:`).returns('91')
-    
+
     assert_equal true, c.test
   end
-  
+
   def test_test_should_return_false_if_below_limit
     c = Conditions::DiskUsage.new
     c.above = 90
     c.mount_point = '/'
-    
+
     c.expects(:`).returns('90')
-    
+
     assert_equal false, c.test
   end
-end
+end

data/test/test_conditions_http_response_code.rb

@@ -14,96 +14,96 @@ class TestHttpResponseCode < Test::Unit::TestCase
     c.prepare
     c
   end
-  
+
   # valid?
-  
+
   def test_valid_condition_is_valid
     c = valid_condition
     assert c.valid?
   end
-  
+
   def test_valid_should_return_false_if_both_code_is_and_code_is_not_are_set
     c = valid_condition do |cc|
       cc.code_is_not = 500
     end
     assert !c.valid?
   end
-  
+
   def test_valid_should_return_false_if_no_host_set
     c = valid_condition do |cc|
       cc.host = nil
     end
     assert !c.valid?
   end
-  
+
   # test
-  
+
   def test_test_should_return_false_if_code_is_is_set_to_200_but_response_is_500
     c = valid_condition
-    Net::HTTP.expects(:start).yields(stub(:read_timeout= => nil, :get => stub(:code => 500)))
+    Net::HTTP.any_instance.expects(:start).yields(mock(:read_timeout= => nil, :get => mock(:code => 500)))
     assert_equal false, c.test
   end
-  
+
   def test_test_should_return_false_if_code_is_not_is_set_to_200_and_response_is_200
     c = valid_condition do |cc|
       cc.code_is = nil
       cc.code_is_not = [200]
     end
-    Net::HTTP.expects(:start).yields(stub(:read_timeout= => nil, :get => stub(:code => 200)))
+    Net::HTTP.any_instance.expects(:start).yields(mock(:read_timeout= => nil, :get => mock(:code => 200)))
     assert_equal false, c.test
   end
-  
+
   def test_test_should_return_true_if_code_is_is_set_to_200_and_response_is_200
     c = valid_condition
-    Net::HTTP.expects(:start).yields(stub(:read_timeout= => nil, :get => stub(:code => 200)))
+    Net::HTTP.any_instance.expects(:start).yields(mock(:read_timeout= => nil, :get => mock(:code => 200)))
     assert_equal true, c.test
   end
-  
+
   def test_test_should_return_false_if_code_is_not_is_set_to_200_but_response_is_500
     c = valid_condition do |cc|
       cc.code_is = nil
       cc.code_is_not = [200]
     end
-    Net::HTTP.expects(:start).yields(stub(:read_timeout= => nil, :get => stub(:code => 500)))
+    Net::HTTP.any_instance.expects(:start).yields(mock(:read_timeout= => nil, :get => mock(:code => 500)))
     assert_equal true, c.test
   end
-  
+
   def test_test_should_return_false_if_code_is_is_set_to_200_but_response_times_out
     c = valid_condition
-    Net::HTTP.expects(:start).raises(Timeout::Error, '')
+    Net::HTTP.any_instance.expects(:start).raises(Timeout::Error, '')
     assert_equal false, c.test
   end
-  
+
   def test_test_should_return_true_if_code_is_not_is_set_to_200_and_response_times_out
     c = valid_condition do |cc|
       cc.code_is = nil
       cc.code_is_not = [200]
     end
-    Net::HTTP.expects(:start).raises(Timeout::Error, '')
+    Net::HTTP.any_instance.expects(:start).raises(Timeout::Error, '')
     assert_equal true, c.test
   end
-  
+
   def test_test_should_return_false_if_code_is_is_set_to_200_but_cant_connect
     c = valid_condition
-    Net::HTTP.expects(:start).raises(Errno::ECONNREFUSED, '')
+    Net::HTTP.any_instance.expects(:start).raises(Errno::ECONNREFUSED, '')
     assert_equal false, c.test
   end
-  
+
   def test_test_should_return_true_if_code_is_not_is_set_to_200_and_cant_connect
     c = valid_condition do |cc|
       cc.code_is = nil
       cc.code_is_not = [200]
     end
-    Net::HTTP.expects(:start).raises(Errno::ECONNREFUSED, '')
+    Net::HTTP.any_instance.expects(:start).raises(Errno::ECONNREFUSED, '')
     assert_equal true, c.test
   end
-  
+
   def test_test_should_return_true_if_code_is_is_set_to_200_and_response_is_200_twice_for_times_two_of_two
     c = valid_condition do |cc|
       cc.times = [2, 2]
     end
-    Net::HTTP.expects(:start).yields(stub(:read_timeout= => nil, :get => stub(:code => 200))).times(2)
+    Net::HTTP.any_instance.expects(:start).yields(stub(:read_timeout= => nil, :get => stub(:code => 200))).times(2)
     assert_equal false, c.test
     assert_equal true, c.test
   end
-end
+end

data/test/test_conditions_process_running.rb

@@ -9,32 +9,32 @@ class TestConditionsProcessRunning < Test::Unit::TestCase
       assert_equal !r, c.test
     end
   end
-  
+
   def test_not_running_returns_opposite
     [true, false].each do |r|
       c = Conditions::ProcessRunning.new
       c.running = r
-    
+
       File.stubs(:exist?).returns(true)
       c.stubs(:watch).returns(stub(:pid => 123))
       File.stubs(:read).returns('5')
       System::Process.any_instance.stubs(:exists?).returns(false)
-    
+
       assert_equal !r, c.test
     end
   end
-  
+
   def test_running_returns_same
     [true, false].each do |r|
       c = Conditions::ProcessRunning.new
       c.running = r
-    
+
       File.stubs(:exist?).returns(true)
       c.stubs(:watch).returns(stub(:pid => 123))
       File.stubs(:read).returns('5')
       System::Process.any_instance.stubs(:exists?).returns(true)
-    
+
       assert_equal r, c.test
     end
   end
-end
+end

data/test/test_conditions_socket_responding.rb

@@ -0,0 +1,122 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestConditionsSocketResponding < Test::Unit::TestCase
+  # valid?
+
+  def test_valid_should_return_false_if_no_options_set
+    c = Conditions::SocketResponding.new
+    c.watch = stub(:name => 'foo')
+    assert_equal false, c.valid?
+
+  end
+
+  def test_valid_should_return_true_if_required_options_set_for_default
+    c = Conditions::SocketResponding.new
+    c.port = 443
+    assert_equal true, c.valid?
+  end
+
+  def test_valid_should_return_true_if_required_options_set_for_tcp
+    c = Conditions::SocketResponding.new
+    c.family = 'tcp'
+    c.port = 443
+    assert_equal true, c.valid?
+  end
+
+  def test_valid_should_return_true_if_required_options_set_for_unix
+    c = Conditions::SocketResponding.new
+    c.path = 'some-path'
+    c.family = 'unix'
+    assert_equal true, c.valid?
+  end
+
+  def test_valid_should_return_true_if_family_is_tcp
+    c = Conditions::SocketResponding.new
+    c.port = 443
+    c.family = 'tcp'
+    assert_equal true, c.valid?
+  end
+
+  def test_valid_should_return_true_if_family_is_unix
+    c = Conditions::SocketResponding.new
+    c.path = 'some-path'
+    c.family = 'unix'
+    c.watch = stub(:name => 'foo')
+    assert_equal true, c.valid?
+  end
+
+  # socket method
+  def test_socket_should_return_127_0_0_1_for_default_addr
+    c = Conditions::SocketResponding.new
+    c.socket = 'tcp:443'
+    assert_equal c.addr, '127.0.0.1'
+  end
+
+  def test_socket_should_set_properties_for_tcp
+    c = Conditions::SocketResponding.new
+    c.socket = 'tcp:127.0.0.1:443'
+    assert_equal c.family, 'tcp'
+    assert_equal c.addr, '127.0.0.1'
+    assert_equal c.port, 443
+    # path should not be set for tcp sockets
+    assert_equal c.path, nil
+  end
+
+  def test_socket_should_set_properties_for_unix
+    c = Conditions::SocketResponding.new
+    c.socket = 'unix:/tmp/process.sock'
+    assert_equal c.family, 'unix'
+    assert_equal c.path, '/tmp/process.sock'
+    # path should not be set for unix domain sockets
+    assert_equal c.port, 0
+  end
+
+  # test
+
+  def test_test_tcp_should_return_false_if_socket_is_listening
+    c = Conditions::SocketResponding.new
+    c.prepare
+
+    TCPSocket.expects(:new).returns(0)
+    assert_equal false, c.test
+  end
+
+  def test_test_tcp_should_return_true_if_no_socket_is_listening
+    c = Conditions::SocketResponding.new
+    c.prepare
+
+    TCPSocket.expects(:new).returns(nil)
+    assert_equal true, c.test
+  end
+
+  def test_test_unix_should_return_false_if_socket_is_listening
+    c = Conditions::SocketResponding.new
+    c.socket = 'unix:/some/path'
+
+    c.prepare
+    UNIXSocket.expects(:new).returns(0)
+    assert_equal false, c.test
+  end
+
+  def test_test_unix_should_return_true_if_no_socket_is_listening
+
+    c = Conditions::SocketResponding.new
+    c.socket = 'unix:/some/path'
+    c.prepare
+
+    UNIXSocket.expects(:new).returns(nil)
+    assert_equal true, c.test
+  end
+
+  def test_test_unix_should_return_true_if_socket_is_listening_2_times
+
+    c = Conditions::SocketResponding.new
+    c.socket = 'unix:/some/path'
+    c.times = [2, 2]
+    c.prepare
+
+    UNIXSocket.expects(:new).returns(nil).times(2)
+    assert_equal false, c.test
+    assert_equal true, c.test
+  end
+end