diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/tc_mauve_alert_changed.rb | 2 | ||||
| -rw-r--r-- | test/tc_mauve_alert_group.rb | 129 | ||||
| -rw-r--r-- | test/tc_mauve_configuration_builders_people_list.rb | 14 | ||||
| -rw-r--r-- | test/tc_mauve_notification.rb | 253 | ||||
| -rw-r--r-- | test/tc_mauve_people_list.rb | 4 |
5 files changed, 351 insertions, 51 deletions
diff --git a/test/tc_mauve_alert_changed.rb b/test/tc_mauve_alert_changed.rb index 6cd51fb..679b9bb 100644 --- a/test/tc_mauve_alert_changed.rb +++ b/test/tc_mauve_alert_changed.rb @@ -43,6 +43,7 @@ alert_group("test_group") { notify("test_person") { every 5.minutes + during { true } } } @@ -118,6 +119,7 @@ alert_group("test_group") { notify("test_person") { every 5.minutes + during { true } } } diff --git a/test/tc_mauve_alert_group.rb b/test/tc_mauve_alert_group.rb index 7e7c4dd..92a9d61 100644 --- a/test/tc_mauve_alert_group.rb +++ b/test/tc_mauve_alert_group.rb @@ -1,17 +1,33 @@ $:.unshift "../lib" require 'th_mauve' -require 'mauve/alert_group' require 'th_mauve_resolv' +require 'mauve/alert_group' +require 'mauve/server' +require 'mauve/configuration' +require 'mauve/configuration_builder' +require 'mauve/configuration_builders' require 'pp' class TcMauveAlertGroup < Mauve::UnitTest + include Mauve + + def setup + super + setup_database + end + + def teardown + teardown_database + super + end + def test_matches_alert - alert = Mauve::Alert.new + alert = Alert.new - alert_group = Mauve::AlertGroup.new("test") + alert_group = AlertGroup.new("test") alert_group.includes = Proc.new { true } assert( alert_group.matches_alert?(alert) ) @@ -38,7 +54,112 @@ class TcMauveAlertGroup < Mauve::UnitTest assert( ! alert_group.matches_alert?(alert) ) end - + def test_notify + config=<<EOF +server { + database "sqlite3::memory:" + use_notification_buffer false +} + +notification_method("email") { + debug! + deliver_to_queue [] + disable_normal_delivery! +} + +person ("test1") { + email "test1@example.com" + all { email } + notify { + during { hours_in_day 0 } + } +} + +person ("test2") { + email "test2@example.com" + all { email } + notify { + during { hours_in_day 0,1 } + } +} + +person ("test3") { + email "test3@example.com" + all { email } + notify { + during { true } + } +} + +alert_group("default") { + includes{ true } + notify("test1") + notify("test2") { + during { hours_in_day 1 } + } +} +EOF + Configuration.current = ConfigurationBuilder.parse(config) + notification_buffer = Configuration.current.notification_methods["email"].deliver_to_queue + Server.instance.setup + + a = Alert.new( + :alert_id => "test", + :source => "test", + :subject => "test" + ) + + a.raise! + assert_equal("test1@example.com", notification_buffer.pop[2]) + assert(notification_buffer.empty?) + + Timecop.freeze(Time.now + 5.minutes) + a.acknowledge!(Configuration.current.people["test2"], Time.now + 5.minutes) + assert_equal(2, notification_buffer.length) + assert_equal(["test1@example.com", "test2@example.com"], notification_buffer.collect{|m| m[2]}.sort) + notification_buffer.pop until notification_buffer.empty? + + Timecop.freeze(Time.now + 5.minutes) + a.clear! + assert_equal(2, notification_buffer.length) + assert_equal(["test1@example.com", "test2@example.com"], notification_buffer.collect{|m| m[2]}.sort) + notification_buffer.pop until notification_buffer.empty? + + # + # If we raise it again, test2 shouldn't get notified. + # + Timecop.freeze(Time.now + 5.minutes) + a.raise! + assert_equal("test1@example.com", notification_buffer.pop[2]) + assert(notification_buffer.empty?) + + Timecop.freeze(Time.now + 5.minutes) + a.clear! + assert_equal("test1@example.com", notification_buffer.pop[2]) + assert(notification_buffer.empty?) + + # + # Freeze to 1am + # + Timecop.freeze(Time.local(2012,5,2,1,0,0)) + + a.raise! + assert_equal("test2@example.com", notification_buffer.pop[2]) + assert(notification_buffer.empty?) + + Timecop.freeze(Time.now + 5.minutes) + a.acknowledge!(Configuration.current.people["test1"], Time.now + 5.minutes) + assert_equal("test2@example.com", notification_buffer.pop[2]) + assert(notification_buffer.empty?) + + # + # Test1 shouldn't get notified, even though he ack'd it. + # + Timecop.freeze(Time.now + 5.minutes) + a.clear! + assert_equal("test2@example.com", notification_buffer.pop[2]) + assert(notification_buffer.empty?) + end end diff --git a/test/tc_mauve_configuration_builders_people_list.rb b/test/tc_mauve_configuration_builders_people_list.rb index 7e0840f..d74f25a 100644 --- a/test/tc_mauve_configuration_builders_people_list.rb +++ b/test/tc_mauve_configuration_builders_people_list.rb @@ -39,9 +39,9 @@ people_list("garmin-cervelo", %w( EOF x = nil assert_nothing_raised { x = Mauve::ConfigurationBuilder.parse(config) } - assert_equal(2, x.people_lists.keys.length) - assert_equal(["team sky","garmin-cervelo"].sort,x.people_lists.keys.sort) - assert_equal(%w(geraint edvald bradley rigoberto ben), x.people_lists["team sky"].list) + assert_equal(2, x.people.keys.length) + assert_equal(["team sky","garmin-cervelo"].sort,x.people.keys.sort) + assert_equal(%w(geraint edvald bradley rigoberto ben), x.people["team sky"].list) end def test_duplicate_people_list @@ -61,13 +61,7 @@ EOF # * duplicate list # * Lars already being on a list # - assert_nothing_raised { x = Mauve::ConfigurationBuilder.parse(config) } - - assert_match(/Lars/, logger_pop()) - assert_match(/Duplicate/, logger_pop()) - - assert_equal(1, x.people_lists.keys.length) - assert_equal(["mark c","mark r","Lars","Bernie","Danny"].sort, x.people_lists["htc-highroad"].list.sort) + assert_raise(ArgumentError) { x = Mauve::ConfigurationBuilder.parse(config) } end end diff --git a/test/tc_mauve_notification.rb b/test/tc_mauve_notification.rb index 00005e1..f78ec27 100644 --- a/test/tc_mauve_notification.rb +++ b/test/tc_mauve_notification.rb @@ -316,10 +316,12 @@ alert_group("default") { level URGENT notify("test1") { + during { true } every 10.minutes } notify("testers") { + during { true } every 15.minutes } @@ -563,48 +565,229 @@ EOF :subject => "test" ) - # - # This should only alert test1 - # - assert_equal(0, Time.now.hour) - alert.raise! - assert_equal(1, notification_buffer.size, "Wrong number of notifications sent") - assert_equal("test1@example.com", notification_buffer.pop[2]) + # At midnight just test1 + # At 1am just test2 + # At 2am, both test1 and test2 (via testers) + # At 3am both test1 and test2 (via testers clause with during) + # At 4am no-one - alert.clear! - assert_equal(1, notification_buffer.size, "Wrong number of notifications sent") - assert_equal("test1@example.com", notification_buffer.pop[2]) + [ + [0, %w(test1)], + [1, %w(test2)], + [2, %w(test1 test2)], + [3, %w(test1 test2)], + [4, []] + ].each do |hour, people| + + assert_equal(hour, Time.now.hour) + alert.raise! + assert_equal(people.length, notification_buffer.size, "Wrong number of notifications sent after a raise at #{hour} am") + sent = [] + sent << notification_buffer.pop while !notification_buffer.empty? + assert_equal(people.collect{|u| "#{u}@example.com"}.sort, sent.collect{|n| n[2]}.sort) + + alert.clear! + assert_equal(people.length, notification_buffer.size, "Wrong number of notifications sent after a clear at #{hour} am") + sent = [] + sent << notification_buffer.pop while !notification_buffer.empty? + assert_equal(people.collect{|u| "#{u}@example.com"}.sort, sent.collect{|n| n[2]}.sort) + + # + # Wind the clock forward for the next round of tests. + # + Timecop.freeze(Time.now+1.hours) + end + end + + def test_individual_notification_preferences_part_deux + config=<<EOF +server { + use_notification_buffer false +} + +notification_method("email") { + debug! + deliver_to_queue [] + disable_normal_delivery! +} + +person ("test1") { + email "test1@example.com" + all { email } + notify { + every 300 + during { hours_in_day(0) } + } +} + +person ("test2") { + email "test2@example.com" + all { email } + notify { + every 300 + during { hours_in_day(1) } + } +} + +people_list("testers", %w(test1 test2)) + +alert_group("test") { + level URGENT + + notify("testers") + + notify("testers") { + every 60 + during { hours_in_day (2) } + } +} + +EOF + + Configuration.current = ConfigurationBuilder.parse(config) + notification_buffer = Configuration.current.notification_methods["email"].deliver_to_queue + + Server.instance.setup + alert = Alert.new( + :alert_id => "test", + :source => "test", + :subject => "test" + ) + # - # Wind forward to 1am when test2 should get alerted - # - Timecop.freeze(Time.now+1.hours) + # At midnight just test1 should be notified. + # At 1am, just test2 + # At 2am both test1 and test2 + + [ + [0, %w(test1)], + [1, %w(test2)], + [2, %w(test1 test2)] + ].each do |hour, people| + + assert_equal(hour, Time.now.hour) + alert.raise! + assert_equal(people.length, notification_buffer.size, "Wrong number of notifications sent after a raise at #{hour} am") + sent = [] + sent << notification_buffer.pop while !notification_buffer.empty? + assert_equal(people.collect{|u| "#{u}@example.com"}.sort, sent.collect{|n| n[2]}.sort) + + alert.clear! + assert_equal(people.length, notification_buffer.size, "Wrong number of notifications sent after a clear at #{hour} am") + sent = [] + sent << notification_buffer.pop while !notification_buffer.empty? + assert_equal(people.collect{|u| "#{u}@example.com"}.sort, sent.collect{|n| n[2]}.sort) + + # + # Wind the clock forward for the next round of tests. + # + Timecop.freeze(Time.now+1.hours) + end - assert_equal(1, Time.now.hour) - alert.raise! - assert_equal(1, notification_buffer.size, "Wrong number of notifications sent") - assert_equal("test2@example.com", notification_buffer.pop[2]) - - alert.clear! - assert_equal(1, notification_buffer.size, "Wrong number of notifications sent") - assert_equal("test2@example.com", notification_buffer.pop[2]) + end - # - # Wind forward to 2am when the testers group should get alerted - # - Timecop.freeze(Time.now+1.hours) + def test_notify_array_of_usernames + config=<<EOF +server { + use_notification_buffer false +} - assert_equal(2, Time.now.hour) - alert.raise! - assert_equal(2, notification_buffer.size, "Wrong number of notifications sent") - assert_equal("test2@example.com", notification_buffer.pop[2]) - assert_equal("test1@example.com", notification_buffer.pop[2]) +notification_method("email") { + debug! + deliver_to_queue [] + disable_normal_delivery! +} - alert.clear! - assert_equal(2, notification_buffer.size, "Wrong number of notifications sent") - assert_equal("test2@example.com", notification_buffer.pop[2]) - assert_equal("test1@example.com", notification_buffer.pop[2]) +person ("test1") { + email "test1@example.com" + all { email } + notify { + every 100 + during { hours_in_day(1) } + } +} - end +person ("test2") { + email "test2@example.com" + all { email } + notify { + every 200 + during { hours_in_day(2) } + } +} +person ("test3") { + email "test3@example.com" + all { email } + notify { + every 300 + during { hours_in_day(3) } + } +} + +people_list("testers1", %w(test1 test2)) + +people_list("testers2", "testers1", "test3") { + notify { + every 500 + during { hours_in_day(4) } + } +} + +alert_group("test") { + level URGENT + + includes { source == "test" } + + notify("test1", "testers1", "testers2") +} +EOF + + Configuration.current = ConfigurationBuilder.parse(config) + notification_buffer = Configuration.current.notification_methods["email"].deliver_to_queue + + Server.instance.setup + + alert = Alert.new( + :alert_id => "test", + :source => "test", + :subject => "test" + ) + assert_equal("test", alert.alert_group.name) + + # At midnight no-one should be alerted + # At 1am just test1 should be alerted + # At 2am just test2 should be alerted (via the testers1 group) + # At 3am no-one should be alerted + # At 4am test1, test2, and test3 should all be alerted (via the testers2 group) + + [ + [0, []], + [1, %w(test1)], + [2, %w(test2)], + [3, []], + [4, %w(test1 test2 test3)] + ].each do |hour, people| + + assert_equal(hour, Time.now.hour) + alert.raise! + assert_equal(people.length, notification_buffer.size, "Wrong number of notifications sent after a raise at #{hour} am") + sent = [] + sent << notification_buffer.pop while !notification_buffer.empty? + assert_equal(people.collect{|u| "#{u}@example.com"}.sort, sent.collect{|n| n[2]}.sort) + + alert.clear! + assert_equal(people.length, notification_buffer.size, "Wrong number of notifications sent after a clear at #{hour} am") + sent = [] + sent << notification_buffer.pop while !notification_buffer.empty? + assert_equal(people.collect{|u| "#{u}@example.com"}.sort, sent.collect{|n| n[2]}.sort) + + # + # Wind the clock forward for the next round of tests. + # + Timecop.freeze(Time.now+1.hours) + end + + end end diff --git a/test/tc_mauve_people_list.rb b/test/tc_mauve_people_list.rb index 78586d2..2f6755a 100644 --- a/test/tc_mauve_people_list.rb +++ b/test/tc_mauve_people_list.rb @@ -60,7 +60,7 @@ EOF notification_buffer = Configuration.current.notification_methods["email"].deliver_to_queue Server.instance.setup - people_list = Configuration.current.people_lists["testers"] + people_list = Configuration.current.people["testers"] alert = Alert.new( :alert_id => "test", @@ -142,7 +142,7 @@ EOF stub_request(:get, "http://localhost/api/attendees/support_shift/2011-08-01T00:05:00"). to_return(:status => 200, :body => YAML.dump(%w(test2))) - people_list = Configuration.current.people_lists["testers"] + people_list = Configuration.current.people["testers"] assert_equal([Configuration.current.people["test1"]], people_list.people) assert_equal([Configuration.current.people["test2"]], people_list.people(Time.now + 5.minutes)) end |