strobemonkey-god 0.7.13

data/test/test_task.rb

@@ -0,0 +1,262 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestTask < Test::Unit::TestCase
+  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
+    no_stdout do
+      assert !@task.valid?
+    end
+  end
+  
+  def test_valid_should_return_false_if_no_valid_states
+    @task.valid_states = nil
+    no_stdout do
+      assert !@task.valid?
+    end
+  end
+  
+  def test_valid_should_return_false_if_no_initial_state
+    @task.initial_state = nil
+    no_stdout do
+      assert !@task.valid?
+    end
+  end
+  
+  # transition
+  
+  def test_transition_should_be_always_if_no_block_was_given
+    @task.transition(:foo, :bar)
+    
+    assert 1, @task.metrics.size
+    assert Conditions::Always, @task.metrics.keys.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_raise NoMethodError do
+      @task.baz = 5
+    end
+  end
+  
+  def test_method_missing_should_raise_for_non_setters
+    assert_raise NoMethodError do
+      @task.baz
+    end
+  end
+  
+  # action
+  
+  def test_action_should_send_string_commands_to_system
+    @task.foo = 'foo'
+    Thread.current.stubs(:==).returns(true)
+    @task.expects(:system).with('foo')
+    no_stdout { @task.action(:foo, nil) }
+  end
+  
+  def test_action_should_call_lambda_commands
+    @task.foo = lambda { }
+    Thread.current.stubs(:==).returns(true)
+    @task.foo.expects(:call)
+    no_stdout { @task.action(:foo, nil) }
+  end
+  
+  def test_action_should_raise_not_implemented_on_non_string_or_lambda_action
+    Thread.current.stubs(:==).returns(true)
+    assert_raise 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)
+    
+    no_stdout do
+      @task.handle_poll(c)
+    end
+  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)
+    
+    no_stdout do
+      @task.handle_poll(c)
+    end
+  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)
+    
+    no_stdout do
+      @task.handle_poll(c)
+    end
+  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)
+    
+    no_stdout do
+      @task.handle_poll(c)
+    end
+  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
+    
+    no_stdout do
+      @task.handle_poll(c)
+    end
+  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)
+    
+    no_stdout do
+      @task.handle_event(c)
+    end
+  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)
+    
+    no_stdout do
+      @task.handle_event(c)
+    end
+  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
+    
+    no_stdout do
+      @task.handle_event(c)
+    end
+  end
+end

data/test/test_timeline.rb

@@ -0,0 +1,37 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestTimeline < Test::Unit::TestCase
+  def setup
+    @timeline = Timeline.new(5)
+  end
+  
+  def test_new_should_be_empty
+    assert_equal 0, @timeline.size
+  end
+  
+  def test_should_not_grow_to_more_than_size
+    (1..10).each do |i|
+      @timeline.push(i)
+    end
+    
+    assert_equal [6, 7, 8, 9, 10], @timeline
+  end
+  
+  def test_clear_should_clear_array
+    @timeline << 1
+    assert_equal [1], @timeline
+    assert_equal [], @timeline.clear
+  end
+  
+  # def test_benchmark
+  #   require 'benchmark'
+  # 
+  #   count = 1_000_000
+  # 
+  #   t = Timeline.new(10)
+  # 
+  #   Benchmark.bmbm do |x|
+  #     x.report("go") { count.times { t.push(5) } }
+  #   end
+  # end
+end

data/test/test_trigger.rb

@@ -0,0 +1,59 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestTrigger < Test::Unit::TestCase
+  def setup
+    Trigger.reset
+  end
+  
+  # base case
+  
+  def test_should_have_empty_triggers
+    assert_equal({}, Trigger.triggers)
+  end
+  
+  # register
+  
+  def test_register_should_add_condition_to_triggers
+    c = Condition.new
+    c.watch = stub(:name => 'foo')
+    Trigger.register(c)
+    
+    assert_equal({'foo' => [c]}, Trigger.triggers)
+  end
+  
+  def test_register_should_add_condition_to_triggers_twice
+    watch = stub(:name => 'foo')
+    c = Condition.new
+    c.watch = watch
+    Trigger.register(c)
+    
+    c2 = Condition.new
+    c2.watch = watch
+    Trigger.register(c2)
+    
+    assert_equal({'foo' => [c, c2]}, Trigger.triggers)
+  end
+  
+  # deregister
+  
+  def test_deregister_should_remove_condition_from_triggers
+    c = Condition.new
+    c.watch = stub(:name => 'foo')
+    Trigger.register(c)
+    Trigger.deregister(c)
+    
+    assert_equal({}, Trigger.triggers)
+  end
+  
+  # broadcast
+  
+  def test_broadcast_should_call_process_on_each_condition
+    c = Condition.new
+    c.watch = stub(:name => 'foo')
+    Trigger.register(c)
+    
+    c.expects(:process).with(:state_change, [:up, :start])
+    
+    Trigger.broadcast(c.watch, :state_change, [:up, :start])
+  end
+end

data/test/test_watch.rb

@@ -0,0 +1,279 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestWatch < Test::Unit::TestCase
+  def setup
+    God.internal_init
+    @watch = Watch.new
+    @watch.name = 'foo'
+    @watch.start = lambda { }
+    @watch.stop = lambda { }
+    @watch.prepare
+  end
+  
+  # new
+  
+  def test_new_should_have_no_behaviors
+    assert_equal [], @watch.behaviors
+  end
+  
+  def test_new_should_have_no_metrics
+    Watch::VALID_STATES.each do |state|
+      assert_equal [], @watch.metrics[state]
+    end
+  end
+    
+  def test_new_should_have_standard_attributes
+    assert_nothing_raised do
+      @watch.name = 'foo'
+      @watch.start = 'start'
+      @watch.stop = 'stop'
+      @watch.restart = 'restart'
+      @watch.interval = 30
+      @watch.grace = 5
+    end
+  end
+  
+  def test_new_should_have_unmonitored_state
+    assert_equal :unmonitored, @watch.state
+  end
+  
+  # valid?
+  
+  def test_valid?
+    God::Process.any_instance.expects(:valid?)
+    @watch.valid?
+  end
+  
+  # behavior
+  
+  def test_behavior_should_record_behavior
+    beh = nil
+    @watch.behavior(:fake_behavior) { |b| beh = b }
+    assert_equal 1, @watch.behaviors.size
+    assert_equal beh, @watch.behaviors.first
+  end
+  
+  def test_invalid_behavior_should_abort
+    assert_abort do
+      @watch.behavior(:invalid)
+    end
+  end
+  
+  # transition
+  
+  def test_transition_should_abort_on_invalid_start_state
+    assert_abort do
+      @watch.transition(:foo, :bar)
+    end
+  end
+  
+  def test_transition_should_accept_all_valid_start_states
+    assert_nothing_raised do
+      Watch::VALID_STATES.each do |state|
+        @watch.transition(state, :bar) { }
+      end
+    end
+  end
+  
+  def test_transition_should_create_and_record_a_metric_for_the_given_start_state
+    @watch.transition(:init, :start) { }
+    assert_equal 1, @watch.metrics[:init].size
+  end
+  
+  # lifecycle
+  
+  def test_lifecycle_should_create_and_record_a_metric_for_nil_start_state
+    @watch.lifecycle { }
+    assert_equal 1, @watch.metrics[nil].size
+  end
+  
+  # start_if
+  
+  def test_start_if_should_place_a_metric_on_up_state
+    @watch.start_if { }
+    assert_equal 1, @watch.metrics[:up].size
+  end
+  
+  # restart_if
+  
+  def test_restart_if_should_place_a_metric_on_up_state
+    @watch.restart_if { }
+    assert_equal 1, @watch.metrics[:up].size
+  end
+  
+  # monitor
+  
+  def test_monitor_should_move_to_init_if_available
+    @watch.instance_eval do
+      transition(:init, :up) { }
+    end
+    @watch.expects(:move).with(:init)
+    @watch.monitor
+  end
+  
+  def test_monitor_should_move_to_up_if_no_init_available
+    @watch.expects(:move).with(:up)
+    @watch.monitor
+  end
+  
+  # unmonitor
+  
+  def test_unmonitor_should_move_to_nil
+    @watch.expects(:move).with(:unmonitored)
+    @watch.unmonitor
+  end
+  
+  # move
+  
+  def test_move_should_not_clean_up_if_from_state_is_nil
+    Thread.current.stubs(:==).returns(true)
+    @watch.driver.expects(:message).never
+    
+    metric = nil
+    
+    @watch.instance_eval do
+      transition(:init, :up) do |on|
+        metric = on
+        on.condition(:process_running) do |c|
+          c.running = true
+          c.interval = 10
+        end
+      end
+    end
+    
+    metric.expects(:disable).never
+    
+    no_stdout { @watch.move(:init) }
+  end
+  
+  def test_move_should_clean_up_from_state_if_not_nil
+    Thread.current.stubs(:==).returns(true)
+    @watch.driver.expects(:message).never
+    
+    metric = nil
+    
+    @watch.instance_eval do
+      transition(:init, :up) do |on|
+        metric = on
+        on.condition(:process_running) do |c|
+          c.running = true
+          c.interval = 10
+        end
+      end
+    end
+    
+    no_stdout { @watch.move(:init) }
+    
+    metric.expects(:disable)
+    
+    no_stdout { @watch.move(:up) }
+  end
+  
+  def test_move_should_call_action
+    Thread.current.stubs(:==).returns(true)
+    @watch.driver.expects(:message).never
+    
+    @watch.expects(:action).with(:start)
+    
+    no_stdout { @watch.move(:start) }
+  end
+  
+  def test_move_should_move_to_up_state_if_no_start_or_restart_metric
+    Thread.current.stubs(:==).returns(true)
+    @watch.driver.expects(:message).never
+    
+    [:start, :restart].each do |state|
+      @watch.expects(:action)
+      no_stdout { @watch.move(state) }
+      assert_equal :up, @watch.state
+    end
+  end
+  
+  def test_move_should_enable_destination_metric
+    Thread.current.stubs(:==).returns(true)
+    @watch.driver.expects(:message).never
+    
+    metric = nil
+    
+    @watch.instance_eval do
+      transition(:init, :up) do |on|
+        metric = on
+        on.condition(:process_running) do |c|
+          c.running = true
+          c.interval = 10
+        end
+      end
+    end
+    
+    metric.expects(:enable)
+    
+    no_stdout { @watch.move(:init) }
+  end
+  
+  # action
+  
+  def test_action_should_pass_start_and_stop_actions_to_call_action
+    Thread.current.stubs(:==).returns(true)
+    @watch.driver.expects(:message).never
+    
+    c = Conditions::FakePollCondition.new
+    [:start, :stop].each do |cmd|
+      @watch.expects(:call_action).with(c, cmd)
+      @watch.action(cmd, c)
+    end
+  end
+  
+  def test_action_should_do_stop_then_start_if_no_restart_command
+    Thread.current.stubs(:==).returns(true)
+    @watch.driver.expects(:message).never
+    
+    c = Conditions::FakePollCondition.new
+    @watch.expects(:call_action).with(c, :stop)
+    @watch.expects(:call_action).with(c, :start)
+    @watch.action(:restart, c)
+  end
+  
+  def test_action_should_restart_to_call_action_if_present
+    Thread.current.stubs(:==).returns(true)
+    @watch.driver.expects(:message).never
+    
+    @watch.restart = lambda { }
+    c = Conditions::FakePollCondition.new
+    @watch.expects(:call_action).with(c, :restart)
+    @watch.action(:restart, c)
+  end
+  
+  # call_action
+  
+  def test_call_action
+    c = Conditions::FakePollCondition.new
+    God::Process.any_instance.expects(:call_action).with(:start)
+    no_stdout { @watch.call_action(c, :start) }
+  end
+  
+  def test_call_action_should_call_before_start_when_behavior_has_that
+    @watch.behavior(:fake_behavior)
+    c = Conditions::FakePollCondition.new
+    God::Process.any_instance.expects(:call_action).with(:start)
+    Behaviors::FakeBehavior.any_instance.expects(:before_start)
+    no_stdout { @watch.call_action(c, :start) }
+  end
+  
+  def test_call_action_should_call_after_start_when_behavior_has_that
+    @watch.behavior(:fake_behavior)
+    c = Conditions::FakePollCondition.new
+    God::Process.any_instance.expects(:call_action).with(:start)
+    Behaviors::FakeBehavior.any_instance.expects(:after_start)
+    no_stdout { @watch.call_action(c, :start) }
+  end
+  
+  # canonical_hash_form
+  
+  def test_canonical_hash_form_should_convert_symbol_to_hash
+    assert_equal({true => :foo}, @watch.canonical_hash_form(:foo))
+  end
+  
+  def test_canonical_hash_form_should_convert_hash_to_hash
+    assert_equal({true => :foo}, @watch.canonical_hash_form(true => :foo))
+  end
+end