firenxis-god 0.11.0

data/test/helper.rb

@@ -0,0 +1,141 @@
+require 'rubygems'
+require File.join(File.dirname(__FILE__), *%w[.. lib god sys_logger])
+require File.join(File.dirname(__FILE__), *%w[.. lib god])
+God::EventHandler.load
+
+require 'test/unit'
+require 'set'
+
+include God
+
+if Process.uid != 0
+  abort <<-EOF
+\n
+*********************************************************************
+*                                                                   *
+*               You need to run these tests as root                 *
+*           chroot and netlink (linux only) require it              *
+*                                                                   *
+*********************************************************************
+EOF
+end
+
+begin
+  require 'mocha'
+rescue LoadError
+  unless gems ||= false
+    require 'rubygems'
+    gems = true
+    retry
+  else
+    abort "=> You need the Mocha gem to run these tests."
+  end
+end
+
+module God
+  module Conditions
+    class FakeCondition < Condition
+      def test
+        true
+      end
+    end
+  
+    class FakePollCondition < PollCondition
+      def test
+        true
+      end
+    end
+  
+    class FakeEventCondition < EventCondition
+      def register
+      end
+      def deregister
+      end
+    end
+  end
+  
+  module Behaviors
+    class FakeBehavior < Behavior
+      def before_start
+        'foo'
+      end
+      def after_start
+        'bar'
+      end
+    end
+  end
+  
+  module Contacts
+    class FakeContact < Contact
+    end
+    
+    class InvalidContact
+    end
+  end
+  
+  def self.reset
+    self.watches = nil
+    self.groups = nil
+    self.server = nil
+    self.inited = nil
+    self.host = nil
+    self.port = nil
+    self.pid_file_directory = nil
+    self.registry.reset
+  end
+end
+
+def silence_warnings
+  old_verbose, $VERBOSE = $VERBOSE, nil
+  yield
+ensure
+  $VERBOSE = old_verbose
+end
+
+LOG.instance_variable_set(:@io, StringIO.new('/dev/null'))
+
+module Kernel
+  def abort(text)
+    raise SystemExit
+  end
+  def exit(code)
+    raise SystemExit
+  end
+end
+
+module Test::Unit::Assertions
+  def assert_abort
+    assert_raise SystemExit do
+      yield
+    end
+  end
+end
+
+# This allows you to be a good OOP citizen and honor encapsulation, but
+# still make calls to private methods (for testing) by doing
+#
+#   obj.bypass.private_thingie(arg1, arg2)
+#
+# Which is easier on the eye than
+#
+#   obj.send(:private_thingie, arg1, arg2)
+#
+class Object
+  class Bypass
+    instance_methods.each do |m|
+      undef_method m unless m =~ /^(__|object_id)/
+    end
+
+    def initialize(ref)
+      @ref = ref
+    end
+  
+    def method_missing(sym, *args)
+      @ref.__send__(sym, *args)
+    end
+  end
+
+  def bypass
+    Bypass.new(self)
+  end
+end

data/test/suite.rb

@@ -0,0 +1,6 @@
+require 'test/unit'
+
+tests = Dir["#{File.dirname(__FILE__)}/test_*.rb"]
+tests.each do |file|
+  require file
+end

data/test/test_behavior.rb

@@ -0,0 +1,18 @@
+require File.dirname(__FILE__) + '/helper'
+
+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

data/test/test_campfire.rb

@@ -0,0 +1,23 @@
+require File.dirname(__FILE__) + '/helper'
+require 'tinder'
+
+class TestCampfire < Test::Unit::TestCase
+  def setup
+    @campfire = God::Contacts::Campfire.new
+  end
+
+  def test_exists
+    God::Contacts::Campfire
+  end
+
+  def test_notify
+    @campfire.subdomain = 'github'
+    @campfire.token = 'abc'
+    @campfire.room = 'danger'
+
+    time = Time.now
+    body = "[#{time.strftime('%H:%M:%S')}] host - msg"
+    Marshmallow::Connection.any_instance.expects(:speak).with('danger', body)
+    @campfire.notify('msg', time, 'prio', 'cat', 'host')
+  end
+end

data/test/test_condition.rb

@@ -0,0 +1,50 @@
+require File.dirname(__FILE__) + '/helper'
+
+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
+      God::EventHandler.start
+      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
+  end
+end

data/test/test_conditions_disk_usage.rb

@@ -0,0 +1,50 @@
+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

data/test/test_conditions_http_response_code.rb

@@ -0,0 +1,109 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestHttpResponseCode < Test::Unit::TestCase
+  def valid_condition
+    c = Conditions::HttpResponseCode.new()
+    c.watch = stub(:name => 'foo')
+    c.host = 'localhost'
+    c.port = 8080
+    c.path = '/'
+    c.timeout = 10
+    c.code_is = 200
+    c.times = 1
+    yield(c) if block_given?
+    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)))
+    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)))
+    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)))
+    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)))
+    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, '')
+    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, '')
+    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, '')
+    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, '')
+    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)
+    assert_equal false, c.test
+    assert_equal true, c.test
+  end
+end

data/test/test_conditions_process_running.rb

@@ -0,0 +1,40 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestConditionsProcessRunning < Test::Unit::TestCase
+  def test_missing_pid_file_returns_opposite
+    [true, false].each do |r|
+      c = Conditions::ProcessRunning.new
+      c.running = r
+      c.stubs(:watch).returns(stub(:pid => 99999999, :name => 'foo'))
+      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