mcproc 2016.2.20

data/test/test_prowl.rb

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+require File.dirname(__FILE__) + '/helper'
+
+class TestProwl < Minitest::Test
+  def test_live_notify
+    prowl = God::Contacts::Prowl.new
+    prowl.name = "Prowly"
+    prowl.apikey = 'put_your_apikey_here'
+
+    Prowly.expects(:notify).returns(mock(:succeeded? => true))
+
+    prowl.notify("Test", Time.now, "Test", "Test", "")
+    assert_equal "sent prowl notification to #{prowl.name}", prowl.info
+  end
+end

data/test/test_registry.rb

@@ -0,0 +1,15 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestRegistry < Minitest::Test
+  def setup
+    God.registry.reset
+  end
+
+  def test_add
+    foo = God::Process.new
+    foo.name = 'foo'
+    God.registry.add(foo)
+    assert_equal 1, God.registry.size
+    assert_equal foo, God.registry['foo']
+  end
+end

data/test/test_sensu.rb

@@ -0,0 +1,11 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestSensu < Minitest::Test
+  def test_sensu_notify
+    sensu = God::Contacts::Sensu.new
+    sensu.check_name = "TestSensuContact"
+
+    UDPSocket.any_instance.expects(:send)
+    sensu.notify("Test", Time.now, "Test", "Test", "")
+  end
+end

data/test/test_slack.rb

@@ -0,0 +1,57 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestSlack < Minitest::Test
+  def setup
+    @slack = God::Contacts::Slack.new
+
+    # new slack webhook url contains three random 'tokens' with length of 9, 9 and 24 characters
+    @slack.url = "https://hooks.slack.com/services/ABCABCABC/DEFDEFDEF/ABCDEFABCDEFABCDEFABCDEF"
+
+    @sample_data = {
+      :message => "a sample message",
+      :time => "2038-01-01 00:00:00",
+      :priority => "High",
+      :category => "Test",
+      :host => "example.com"
+    }
+  end
+
+  def test_api_url
+    assert_equal @slack.url, @slack.api_url.to_s
+  end
+
+  def test_notify
+    Net::HTTP.any_instance.expects(:request).returns(Net::HTTPSuccess.new('a', 'b', 'c'))
+
+    @slack.channel = "#ops"
+
+    @slack.notify('msg', Time.now, 'prio', 'cat', 'host')
+    assert_equal "successfully notified slack on channel #ops", @slack.info
+  end
+
+  def test_default_channel
+    Net::HTTP.any_instance.expects(:request).returns(Net::HTTPSuccess.new('a', 'b', 'c'))
+
+    @slack.notify('msg', Time.now, 'prio', 'cat', 'host')
+    assert_equal "successfully notified slack on channel #general", @slack.info
+  end
+
+  def test_default_formatting
+    text = @slack.text(@sample_data)
+    assert_equal "High alert on example.com: a sample message (Test, 2038-01-01 00:00:00)", text
+  end
+
+  def test_custom_formatting
+    @slack.format = "%{host}: %{message}"
+    text = @slack.text(@sample_data)
+    assert_equal "example.com: a sample message", text
+  end
+
+  def test_notify_channel
+    @slack.notify_channel = true
+    @slack.format = ""
+    text = @slack.text(@sample_data)
+    assert_equal "<!channel> ", text
+  end
+end
+

data/test/test_socket.rb

@@ -0,0 +1,34 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestSocket < Minitest::Test
+  def setup
+    silence_warnings do
+      Object.const_set(:DRb, stub_everything)
+    end
+  end
+
+  def test_should_start_a_drb_server
+    DRb.expects(:start_service)
+    God::Socket.new
+  end
+
+  def test_should_use_supplied_port_and_host
+    DRb.expects(:start_service).with { |uri, object| uri == "drbunix:///tmp/god.9999.sock" && object.is_a?(God::Socket) }
+    server = God::Socket.new(9999)
+  end
+
+  def test_should_forward_foreign_method_calls_to_god
+    server = nil
+    server = God::Socket.new
+    God.expects(:send).with(:something_random)
+    server.something_random
+  end
+
+  # ping
+
+  def test_ping_should_return_true
+    server = nil
+    server = God::Socket.new
+    assert server.ping
+  end
+end

data/test/test_statsd.rb

@@ -0,0 +1,22 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestStatsd < Minitest::Test
+  def setup
+    @statsd = God::Contacts::Statsd.new
+  end
+
+  def test_exists
+    God::Contacts::Statsd
+  end
+
+  def test_notify
+    [
+        'cpu out of bounds',
+        'memory out of bounds',
+        'process is flapping'
+    ].each do |event_type|
+      ::Statsd.any_instance.expects(:increment).with("god.#{event_type.gsub(/\s/, '_')}.127_0_0_1.myapp-thin-1234")
+      @statsd.notify("myapp-thin-1234 [trigger] #{event_type}", Time.now, 'some priority', 'and some category', '127.0.0.1')
+    end
+  end
+end

data/test/test_sugar.rb

@@ -0,0 +1,42 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestSugar < Minitest::Test
+  def test_seconds
+    assert_equal 1, 1.seconds
+    assert_equal 1, 1.second
+  end
+
+  def test_minutes
+    assert_equal 60, 1.minutes
+    assert_equal 60, 1.minute
+  end
+
+  def test_hours
+    assert_equal 3600, 1.hours
+    assert_equal 3600, 1.hour
+  end
+
+  def test_days
+    assert_equal 86400, 1.days
+    assert_equal 86400, 1.day
+  end
+
+  def test_kilobytes
+    assert_equal 1, 1.kilobytes
+    assert_equal 1, 1.kilobyte
+  end
+
+  def test_megabytes
+    assert_equal 1024, 1.megabytes
+    assert_equal 1024, 1.megabyte
+  end
+
+  def test_gigabytes
+    assert_equal 1024 ** 2, 1.gigabytes
+    assert_equal 1024 ** 2, 1.gigabyte
+  end
+
+  def test_percent
+    assert_equal 1, 1.percent
+  end
+end

data/test/test_system_portable_poller.rb

@@ -0,0 +1,17 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestSystemPortablePoller < Minitest::Test
+  def setup
+    pid = Process.pid
+    @process = System::PortablePoller.new(pid)
+  end
+
+  def test_time_string_to_seconds
+    assert_equal 0, @process.bypass.time_string_to_seconds('0:00:00')
+    assert_equal 0, @process.bypass.time_string_to_seconds('0:00:55')
+    assert_equal 27, @process.bypass.time_string_to_seconds('0:27:32')
+    assert_equal 75, @process.bypass.time_string_to_seconds('1:15:13')
+    assert_equal 735, @process.bypass.time_string_to_seconds('12:15:13')
+  end
+end
+

data/test/test_system_process.rb

@@ -0,0 +1,30 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestSystemProcess < Minitest::Test
+  def setup
+    pid = Process.pid
+    @process = System::Process.new(pid)
+  end
+
+  def test_exists_should_return_true_for_running_process
+    assert_equal true, @process.exists?
+  end
+
+  def test_exists_should_return_false_for_non_existant_process
+    assert_equal false, System::Process.new(9999999).exists?
+  end
+
+  def test_memory
+    assert_kind_of Integer, @process.memory
+    assert @process.memory > 0
+  end
+
+  def test_percent_memory
+    assert_kind_of Float, @process.percent_memory
+  end
+
+  def test_percent_cpu
+    assert_kind_of Float, @process.percent_cpu
+  end
+end
+

data/test/test_task.rb

@@ -0,0 +1,246 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestTask < Minitest::Test
+  def setup
+    God.internal_init
+    @task = Task.new
+    @task.name = 'foo'
+    @task.valid_states = [:foo, :bar]
+    @task.initial_state = :foo
+    @task.interval = 5
+    @task.prepare
+  end
+
+  # valid?
+
+  def test_valid_should_return_false_if_no_name
+    @task.name = nil
+    assert !@task.valid?
+  end
+
+  def test_valid_should_return_false_if_no_valid_states
+    @task.valid_states = nil
+    assert !@task.valid?
+  end
+
+  def test_valid_should_return_false_if_no_initial_state
+    @task.initial_state = nil
+    assert !@task.valid?
+  end
+
+  # transition
+
+  def test_transition_should_be_always_if_no_block_was_given
+    @task.transition(:foo, :bar)
+
+    assert_equal 5, @task.metrics.size
+    assert_equal Conditions::Always, @task.metrics[:foo].first.conditions.first.class
+  end
+
+  # method_missing
+
+  def test_method_missing_should_create_accessor_for_states
+    assert_nothing_raised do
+      @task.foo = 'testing'
+    end
+  end
+
+  def test_method_missing_should_raise_for_non_states
+    assert_raises NoMethodError do
+      @task.baz = 5
+    end
+  end
+
+  def test_method_missing_should_raise_for_non_setters
+    assert_raises NoMethodError do
+      @task.baz
+    end
+  end
+
+  # action
+
+  def test_action_should_send_string_commands_to_system
+    @task.foo = 'foo'
+    @task.driver.stubs(:in_driver_context?).returns(true)
+    @task.expects(:system).with('foo')
+    @task.action(:foo, nil)
+  end
+
+  def test_action_should_call_lambda_commands
+    @task.foo = lambda { }
+    @task.driver.stubs(:in_driver_context?).returns(true)
+    @task.foo.expects(:call)
+    @task.action(:foo, nil)
+  end
+
+  def test_action_should_raise_not_implemented_on_non_string_or_lambda_action
+    @task.driver.stubs(:in_driver_context?).returns(true)
+    assert_raises NotImplementedError do
+      @task.foo = 7
+      @task.action(:foo, nil)
+    end
+  end
+
+  def test_action_from_outside_driver_should_send_message_to_driver
+    @task.foo = 'foo'
+    @task.driver.expects(:message).with(:action, [:foo, nil])
+    @task.action(:foo, nil)
+  end
+
+  # attach
+
+  def test_attach_should_schedule_for_poll_condition
+    c = Conditions::FakePollCondition.new
+    @task.driver.expects(:schedule).with(c, 0)
+    @task.attach(c)
+  end
+
+  def test_attach_should_regsiter_for_event_condition
+    c = Conditions::FakeEventCondition.new
+    c.expects(:register)
+    @task.attach(c)
+  end
+
+  # detach
+
+  def test_detach_should_reset_poll_condition
+    c = Conditions::FakePollCondition.new
+    c.expects(:reset)
+    c.expects(:deregister).never
+    @task.detach(c)
+  end
+
+  def test_detach_should_deregister_event_conditions
+    c = Conditions::FakeEventCondition.new
+    c.expects(:deregister).once
+    @task.detach(c)
+  end
+
+  # trigger
+
+  def test_trigger_should_send_message_to_driver
+    c = Conditions::FakePollCondition.new
+    @task.driver.expects(:message).with(:handle_event, [c])
+    @task.trigger(c)
+  end
+
+  # handle_poll
+
+  def test_handle_poll_no_change_should_reschedule
+    c = Conditions::FakePollCondition.new
+    c.watch = @task
+    c.interval = 10
+
+    m = Metric.new(@task, {true => :up})
+    @task.directory[c] = m
+
+    c.expects(:test).returns(false)
+    @task.driver.expects(:schedule)
+    @task.handle_poll(c)
+  end
+
+  def test_handle_poll_change_should_move
+    c = Conditions::FakePollCondition.new
+    c.watch = @task
+    c.interval = 10
+
+    m = Metric.new(@task, {true => :up})
+    @task.directory[c] = m
+
+    c.expects(:test).returns(true)
+    @task.expects(:move).with(:up)
+    @task.handle_poll(c)
+  end
+
+  def test_handle_poll_should_use_overridden_transition
+    c = Conditions::Tries.new
+    c.watch = @task
+    c.times = 1
+    c.transition = :start
+    c.prepare
+
+    m = Metric.new(@task, {true => :up})
+    @task.directory[c] = m
+
+    @task.expects(:move).with(:start)
+    @task.handle_poll(c)
+  end
+
+  def test_handle_poll_should_notify_if_triggering
+    c = Conditions::FakePollCondition.new
+    c.watch = @task
+    c.interval = 10
+    c.notify = 'tom'
+
+    m = Metric.new(@task, {true => :up})
+    @task.directory[c] = m
+
+    c.expects(:test).returns(true)
+    @task.expects(:notify)
+    @task.handle_poll(c)
+  end
+
+  def test_handle_poll_should_not_notify_if_not_triggering
+    c = Conditions::FakePollCondition.new
+    c.watch = @task
+    c.interval = 10
+    c.notify = 'tom'
+
+    m = Metric.new(@task, {true => :up})
+    @task.directory[c] = m
+
+    c.expects(:test).returns(false)
+    @task.expects(:notify).never
+    @task.handle_poll(c)
+  end
+
+  def test_handle_poll_unexpected_exception_should_reschedule
+    c = Conditions::FakePollCondition.new
+    c.watch = @task
+    c.interval = 10
+
+    m = Metric.new(@task, {true => :up})
+    @task.directory[c] = m
+
+    c.expects(:test).raises(StandardError)
+    @task.driver.expects(:schedule)
+
+    @task.handle_poll(c)
+  end
+
+  # handle_event
+
+  def test_handle_event_should_move
+    c = Conditions::FakeEventCondition.new
+    c.watch = @task
+
+    m = Metric.new(@task, {true => :up})
+    @task.directory[c] = m
+
+    @task.expects(:move).with(:up)
+    @task.handle_event(c)
+  end
+
+  def test_handle_event_should_notify_if_triggering
+    c = Conditions::FakeEventCondition.new
+    c.watch = @task
+    c.notify = 'tom'
+
+    m = Metric.new(@task, {true => :up})
+    @task.directory[c] = m
+
+    @task.expects(:notify)
+    @task.handle_event(c)
+  end
+
+  def test_handle_event_should_not_notify_if_no_notify_set
+    c = Conditions::FakeEventCondition.new
+    c.watch = @task
+
+    m = Metric.new(@task, {true => :up})
+    @task.directory[c] = m
+
+    @task.expects(:notify).never
+    @task.handle_event(c)
+  end
+end