mcproc 2016.2.20

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_airbrake.rb

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+require File.dirname(__FILE__) + '/helper'
+
+class TestAirbrake < Minitest::Test
+  def test_notify
+    airbrake = God::Contacts::Airbrake.new
+    airbrake.apikey = "put_your_apikey_here"
+    airbrake.name = "Airbrake"
+
+    Airbrake.expects(:notify).returns "123"
+
+    airbrake.notify("Test message for airbrake", Time.now, "airbrake priority", "airbrake category", "")
+  end
+end

data/test/test_behavior.rb

@@ -0,0 +1,18 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestBehavior < Minitest::Test
+  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_raises 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,22 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestCampfire < Minitest::Test
+  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,52 @@
+require File.dirname(__FILE__) + '/helper'
+
+class BadlyImplementedCondition < PollCondition
+end
+
+class TestCondition < Minitest::Test
+
+  # 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_raises 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
+  ensure
+    God::EventHandler.stop
+  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_raises 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 < Minitest::Test
+  # 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 < Minitest::Test
+  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.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.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.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.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.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.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.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.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.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

data/test/test_conditions_process_running.rb

@@ -0,0 +1,40 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestConditionsProcessRunning < Minitest::Test
+  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

data/test/test_conditions_socket_responding.rb

@@ -0,0 +1,176 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestConditionsSocketResponding < Minitest::Test
+  # 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
+    assert_equal c.responding, false
+    # 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'
+    assert_equal c.responding, false
+    # path should not be set for unix domain sockets
+    assert_equal c.port, 0
+  end
+
+  # test : responding = false
+
+  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
+
+  # test : responding = true
+
+  def test_test_tcp_should_return_true_if_socket_is_listening_with_responding_true
+    c = Conditions::SocketResponding.new
+    c.responding = true
+    c.prepare
+
+    TCPSocket.expects(:new).returns(0)
+    assert_equal true, c.test
+  end
+
+  def test_test_tcp_should_return_false_if_no_socket_is_listening_with_responding_true
+    c = Conditions::SocketResponding.new
+    c.responding = true
+    c.prepare
+
+    TCPSocket.expects(:new).returns(nil)
+    assert_equal false, c.test
+  end
+
+  def test_test_unix_should_return_true_if_socket_is_listening_with_responding_true
+    c = Conditions::SocketResponding.new
+    c.responding = true
+    c.socket = 'unix:/some/path'
+
+    c.prepare
+    UNIXSocket.expects(:new).returns(0)
+    assert_equal true, c.test
+  end
+
+  def test_test_unix_should_return_false_if_no_socket_is_listening_with_responding_true
+    c = Conditions::SocketResponding.new
+    c.socket = 'unix:/some/path'
+    c.responding = true
+    c.prepare
+
+    UNIXSocket.expects(:new).returns(nil)
+    assert_equal false, c.test
+  end
+
+  def test_test_unix_should_return_false_if_socket_is_listening_2_times_with_responding_true
+    c = Conditions::SocketResponding.new
+    c.socket = 'unix:/some/path'
+    c.responding = true
+    c.times = [2, 2]
+    c.prepare
+
+    UNIXSocket.expects(:new).returns(nil).times(2)
+    assert_equal false, c.test
+    assert_equal false, c.test
+  end
+end