test/unit_test/tests/recurring_downtime_Test.php

   1 <?php
   2 /**
   3  * @package    NINJA
   4  * @author     op5
   5  * @license    GPL
   6  */
   7 class Recurring_downtime_Test extends PHPUnit_Framework_TestCase {
   8         var $scheduletimeofday = "12:00";
   9         var $scheduleendtime = "15:00";
  10         var $basecomment = 'Recurring Downtime Test Schedule For ';
  11
  12         /**
  13          *      Set up prerequisities for this test
  14          */
  15         public function setUp() {
  16                 $this->auth = Auth::instance(array('session_key' => false))->force_user(new Op5User_AlwaysAuth());
  17
  18                 $this->sd = new ScheduleDate_Model();
  19
  20                 $this->basictests = array(
  21                         mktime(23, 50, 0, 11, 11, 2036) => 'Schedule on a monday',
  22                         mktime(23, 50, 0, 11, 12, 2036) => 'Schedule on a tuesday',
  23                         mktime(23, 50, 0, 11, 13, 2036) => 'Schedule on a wednesday',
  24                         mktime(23, 50, 0, 11, 14, 2036) => 'Schedule on a thursday',
  25                         mktime(23, 50, 0, 11, 15, 2036) => 'Schedule on a friday',
  26                         mktime(23, 50, 0, 11, 16, 2036) => 'Schedule on a saturday',
  27                         mktime(23, 50, 0, 11, 17, 2036) => 'Schedule on a sunday',
  28                         mktime(23, 50, 0, 11, 30, 2036) => 'Schedule the first of a new month',
  29                         mktime(23, 50, 0, 2, 28, 2036) => 'Schedule on a leap day',
  30                         mktime(23, 50, 0, 12, 31, 2036) => 'Schedule on new years day',
  31                 );
  32         }
  33
  34         public function tearDown() {
  35                 $db = Database::instance();
  36                 $db->query("TRUNCATE TABLE recurring_downtime");
  37                 $db->query("TRUNCATE TABLE recurring_downtime_objects");
  38
  39         }
  40
  41         public function resubmit_and_cleanup($tests, $type) {
  42                 $ls = Livestatus::instance();
  43                 $comment = $this->basecomment . $type;
  44                 $current_number = count($ls->getDowntimes(array('filter' => array('comment' => $comment))));
  45
  46                 $old_count = array();
  47
  48                 foreach ($tests as $time => $description) {
  49                         $old_count[$time] = count($ls->getDowntimes(array('filter' => array('start_time' => strtotime("{$this->scheduletimeofday} +1 day", $time)), 'columns' => array('id'))));
  50                         $output = '';
  51                         // Test that we cannot create duplicate schedules. Function should return skipping to STDERR if duplicate.
  52                         exec('/usr/bin/php index.php default/cron/downtime/'.$time.' 2>&1', $output, $status);
  53                         $this->assertEquals(0, $status, 'Return code should be zero');
  54                         $this->assertNotEmpty($output, "$description twice should give error");
  55                 }
  56
  57                 $ids = array();
  58                 foreach ($tests as $time => $description) {
  59                         $dt = $ls->getDowntimes(array('filter' => array('start_time' => strtotime("{$this->scheduletimeofday} +1 day", $time)), 'columns' => array('id')));
  60                         $this->assertCount($old_count[$time], $dt, 'There should still be the same number of matching downtimes from '.$description);
  61                         foreach ($dt as $row) {
  62                                 $ids[] = $row['id'];
  63                         }
  64                 }
  65
  66                 $this->assertCount($current_number, $ls->getDowntimes(array('filter' => array('comment' => $comment))), 'Still same number of downtimes in total with our comment');
  67
  68                 // Remove downtimes when tests are done.
  69                 $cmd = (strpos($type, 'host') !== false) ? 'DEL_HOST_DOWNTIME;' : 'DEL_SVC_DOWNTIME;';
  70                 foreach ($ids as $id) {
  71                         $res = nagioscmd::submit_to_nagios($cmd . $id);
  72                         $this->assertTrue($res, 'Host delete command was submitted');
  73                 }
  74
  75                 $this->assertCount(0, $ls->getDowntimes(array('filter' => array('comment' => $comment))), "Downtimes are gone after deleting them");
  76         }
  77
  78         public function cron($tests, $type, $expected_number) {
  79                 $comment = $this->basecomment . $type;
  80                 foreach ($tests as $time => $description) {
  81                         $output = '';
  82                         exec('/usr/bin/php index.php default/cron/downtime/'.$time.' 2>&1', $output, $status);
  83                         $this->assertEmpty($output, $description ." returned output: ".implode("\n", $output));
  84                         $this->assertEquals(0, $status, 'Return code should be zero');
  85                 }
  86
  87                 sleep(3); # Y U SO SLOW?
  88
  89                 $ls = Livestatus::instance();
  90                 foreach ($tests as $time => $description) {
  91                         $dt = $ls->getDowntimes(array('filter' => array('start_time' => strtotime("{$this->scheduletimeofday} +1 day", $time)), 'columns' => array('comment'), 'auth' => false));
  92                         $this->assertCount($expected_number, $dt, "Unexpected number of downtimes created from $description");
  93                         foreach ($dt as $row) {
  94                                 $this->assertEquals('AUTO: ' . $comment, $row['comment'], "Downtimes matching $description should have proper comment");
  95                         }
  96                 }
  97         }
  98
  99         /**
 100          *      Test if everyday schedule for hosts work
 101          */
 102         public function test_schedule_hosts() {
 103                 $comment = $this->basecomment.'hosts';
 104                 $data = array(
 105                         'author' => 'me',
 106                         "downtime_type" => "hosts",
 107                         "objects" => array("monitor"),
 108                         "comment" => $comment,
 109                         "start_time" => $this->scheduletimeofday,
 110                         "end_time" => $this->scheduleendtime,
 111                         "duration" => "2:00",
 112                         "fixed" => "1",
 113                         "weekdays" => array("1","2","3","4","5","6","0"),
 114                         "months" => array("1","2","3","4","5","6","7","8","9","10","11","12"));
 115                 $id;
 116                 $this->assertTrue($this->sd->edit_schedule($data, $id));
 117                 $db = Database::instance();
 118                 $sql = "SELECT id FROM recurring_downtime WHERE comment = {$db->escape($comment)} ORDER BY id DESC";
 119                 $result = $db->query($sql);
 120                 $this->assertCount(1, $result, "After creating a new schedule, there's only one with that name");
 121
 122                 $this->cron($this->basictests, 'hosts', 1);
 123                 $this->resubmit_and_cleanup($this->basictests, 'hosts');
 124
 125                 $sql = "DELETE FROM recurring_downtime WHERE comment = {$db->escape($comment)}";
 126                 $result = $db->query($sql);
 127                 $this->assertCount(1, $result, "One schedule was deleted.");
 128         }
 129
 130         /**
 131          *      Test if everyday schedule for hosts work
 132          */
 133         public function test_schedule_hostgroups() {
 134                 $comment = $this->basecomment . 'hostgroups';
 135                 $data = array(
 136                         'author' => 'me',
 137                         "downtime_type" => "hostgroups",
 138                         "objects" => array("hostgroup_up", "hostgroup_all"),
 139                         "comment" => $comment,
 140                         "start_time" => $this->scheduletimeofday,
 141                         'end_time' => $this->scheduleendtime,
 142                         "duration" => "2:00",
 143                         "fixed" => "1",
 144                         "weekdays" => array("1","2","3","4","5","6","0"),
 145                         "months" => array("1","2","3","4","5","6","7","8","9","10","11","12"));
 146
 147                 # The number is wrong.
 148                 # Any overlapping hosts will be added twice.
 149                 # However, it's slightly better with two downtimes than none.
 150                 $number = 0;
 151                 $ls = Livestatus::instance();
 152                 foreach ($data['objects'] as $group) {
 153                         $number += count($ls->getHosts(array('columns' => array('name'), 'filter' => array('groups' => array('>=' => $group)))));
 154                 }
 155
 156                 $id;
 157                 $this->assertTrue($this->sd->edit_schedule($data, $id));
 158                 $db = Database::instance();
 159                 $sql = "SELECT id FROM recurring_downtime WHERE comment = {$db->escape($comment)} ORDER BY id DESC";
 160                 $result = $db->query($sql);
 161                 $this->assertCount(1, $result, "After creating a new schedule, there's only one with that name");
 162
 163                 $this->cron($this->basictests, 'hostgroups', $number);
 164                 $this->resubmit_and_cleanup($this->basictests, 'hostgroups');
 165
 166                 $sql = "DELETE FROM recurring_downtime WHERE comment = {$db->escape($comment)}";
 167                 $result = $db->query($sql);
 168                 $this->assertCount(1, $result, "One schedule was deleted.");
 169         }
 170
 171         /**
 172          *↦ Test if everyday schedule for services work
 173          */
 174         public function test_schedule_services() {
 175                 $comment = $this->basecomment . 'services';
 176                 $data = array(
 177                         'author' => 'me',
 178                         "downtime_type" => "services",
 179                         "objects" => array("host_down;service ok", "host_down;service critical"),
 180                         "comment" => $comment,
 181                         "start_time" => $this->scheduletimeofday,
 182                         'end_time' => $this->scheduleendtime,
 183                         "duration" => "2:00",
 184                         "fixed" => "1",
 185                         "weekdays" => array("1","2","3","4","5","6","0"),
 186                         "months" => array("1","2","3","4","5","6","7","8","9","10","11","12"));
 187                 $id;
 188                 $this->assertTrue($this->sd->edit_schedule($data, $id));
 189                 $db = Database::instance();
 190                 $sql = "SELECT id FROM recurring_downtime WHERE comment = {$db->escape($comment)} ORDER BY id DESC";
 191                 $result = $db->query($sql);
 192                 $this->assertCount(1, $result, "After creating a new schedule, there's only one with that name");
 193
 194                 $this->cron($this->basictests, 'services', 2);
 195                 $this->resubmit_and_cleanup($this->basictests, 'services');
 196
 197                 $db = Database::instance();
 198                 $sql = "DELETE FROM recurring_downtime WHERE comment = {$db->escape($comment)}";
 199                 $result = $db->query($sql);
 200                 $this->assertCount(1, $result, "One schedule was deleted.");
 201         }
 202
 203         /**
 204          *↦ Test if everyday schedule for services work
 205          */
 206         public function test_schedule_servicegroups() {
 207                 $comment = $this->basecomment . 'servicegroups';
 208                 $data = array(
 209                         'author' => 'me',
 210                         "downtime_type" => "servicegroups",
 211                         "objects" => array("servicegroup_ok", "servicegroup_critical"),
 212                         "comment" => $comment,
 213                         "start_time" => $this->scheduletimeofday,
 214                         'end_time' => $this->scheduleendtime,
 215                         "duration" => "2:00",
 216                         "fixed" => "1",
 217                         "weekdays" => array("1","2","3","4","5","6","0"),
 218                         "months" => array("1","2","3","4","5","6","7","8","9","10","11","12"));
 219                 $id;
 220                 $this->assertTrue($this->sd->edit_schedule($data, $id));
 221                 $db = Database::instance();
 222                 $sql = "SELECT id FROM recurring_downtime WHERE comment = {$db->escape($comment)} ORDER BY id DESC";
 223                 $result = $db->query($sql);
 224                 $this->assertCount(1 ,$result, "After creating a new schedule, there's only one with that name");
 225
 226                 # The number is wrong.
 227                 # Any overlapping hosts will be added twice.
 228                 # However, it's slightly better with two downtimes than none.
 229                 $number = 0;
 230                 $ls = Livestatus::instance();
 231                 foreach ($data['objects'] as $group) {
 232                         $number += count($ls->getServices(array('filter' => array('groups' => array('>=' => $group)))));
 233                 }
 234
 235                 $this->cron($this->basictests, 'servicegroups', $number);
 236                 $this->resubmit_and_cleanup($this->basictests, 'servicegroups');
 237
 238                 $db = Database::instance();
 239                 $sql = "DELETE FROM recurring_downtime WHERE comment = {$db->escape($comment)}";
 240                 $result = $db->query($sql);
 241                 $this->assertCount(1, $result, "One schedule was deleted.");
 242         }
 243
 244         public function test_host_noschedule() {
 245                 $comment = $this->basecomment . "hosts";
 246                 $data = array(
 247                         'author' => 'me',
 248                         "downtime_type" => "hosts",
 249                         "objects" => array("monitor"),
 250                         "comment" => $comment,
 251                         "start_time" => $this->scheduletimeofday,
 252                         'end_time' => $this->scheduleendtime,
 253                         "duration" => "2:00",
 254                         "fixed" => "1",
 255                         "weekdays" => array("1","2","3","4","5","6"),
 256                         "months" => array("1","2","3","4","5","6","7","8","9","10","11","12"));
 257                 $id;
 258                 $this->assertTrue($this->sd->edit_schedule($data, $id));
 259
 260                 $tests_expected = array(
 261                         strtotime("2036-01-17 23:50") => "Schedule on thursday when sunday is excluded",
 262                 );
 263                 $tests_unexpected = array(
 264                         strtotime("2036-01-19 23:50") => "Schedule on saturday when sunday is excluded",
 265                 );
 266
 267                 $this->cron($tests_expected, 'hosts', 1);
 268                 $this->resubmit_and_cleanup($tests_expected, 'hosts');
 269
 270                 $this->cron($tests_unexpected, 'hosts', 0);
 271                 # Honey, I swear, Nothing Happened, so there's nothing to clean up!
 272
 273                 $db = Database::instance();
 274                 $sql = "DELETE FROM recurring_downtime WHERE comment = {$db->escape($comment)}";
 275                 $result = $db->query($sql);
 276                 $this->assertCount(1, $result, "One schedule should be deleted.");
 277         }
 278
 279         public function test_host_acrossmidnight() {
 280                 $comment = $this->basecomment . "hosts";
 281                 $data = array(
 282                         'author' => 'me',
 283                         "downtime_type" => "hosts",
 284                         "objects" => array("monitor"),
 285                         "comment" => $comment,
 286                         "start_time" => $this->scheduleendtime,
 287                         'end_time' => $this->scheduletimeofday,
 288                         "duration" => "2:00",
 289                         "fixed" => "1",
 290                         "weekdays" => array("1","2","3","4","5","6","0"),
 291                         "months" => array("1","2","3","4","5","6","7","8","9","10","11","12"));
 292                 $id;
 293                 $this->assertTrue($this->sd->edit_schedule($data, $id));
 294                 $time = mktime(23, 50, 0, 11, 11, 2036);
 295
 296                 $output = '';
 297                 exec('/usr/bin/php index.php default/cron/downtime/'.$time.' 2>&1', $output, $status);
 298                 $this->assertEmpty($output, "acrossmidnight returned output: ".implode("\n", $output));
 299                 $this->assertEquals(0, $status, 'Return code should be zero');
 300
 301                 sleep(3); # Y U SO SLOW?
 302
 303                 $ls = Livestatus::instance();
 304                 $dt = $ls->getDowntimes(array('filter' => array('start_time' => strtotime("{$this->scheduleendtime} +1 day", $time), 'end_time' => strtotime("{$this->scheduletimeofday} +2 days", $time)), 'columns' => array('id'), 'auth' => false));
 305                 $this->assertCount(1, $dt, "Unexpected number of downtimes created from acrossmidnight");
 306
 307                 // Remove downtimes when tests are done.
 308                 $cmd = 'DEL_HOST_DOWNTIME;';
 309                 foreach ($dt as $id) {
 310                         $res = nagioscmd::submit_to_nagios($cmd . $id['id']);
 311                         $this->assertTrue($res, 'Host delete command was submitted');
 312                 }
 313
 314                 $this->assertCount(0, $ls->getDowntimes(array('filter' => array('start_time' => strtotime("{$this->scheduleendtime} +1 day", $time), 'end_time' => strtotime("{$this->scheduletimeofday} +2 days", $time)))), "Downtimes are gone after deleting them");
 315         }
 316
 317         public function test_realistic_time() {
 318                 $comment = $this->basecomment . "tomorrow";
 319                 $data = array(
 320                         'author' => 'me',
 321                         "downtime_type" => "hosts",
 322                         "objects" => array("monitor"),
 323                         "comment" => $comment,
 324                         "start_time" => $this->scheduleendtime,
 325                         'end_time' => $this->scheduletimeofday,
 326                         "duration" => "2:00",
 327                         "fixed" => "1",
 328                         "weekdays" => array("1","2","3","4","5","6","0"),
 329                         "months" => array("1","2","3","4","5","6","7","8","9","10","11","12"));
 330                 $id;
 331                 $this->assertTrue($this->sd->edit_schedule($data, $id));
 332
 333                 $output = '';
 334                 exec('/usr/bin/php index.php default/cron/downtime 2>&1', $output, $status);
 335                 $this->assertEmpty($output, "acrossmidnight returned output: ".implode("\n", $output));
 336                 $this->assertEquals(0, $status, 'Return code should be zero');
 337
 338                 sleep(3); # Y U SO SLOW?
 339
 340                 $ls = Livestatus::instance();
 341                 $dt = $ls->getDowntimes(array('filter' => array('start_time' => strtotime("{$this->scheduleendtime} +1 day"), 'end_time' => strtotime("{$this->scheduletimeofday} +2 days")), 'columns' => array('id'), 'auth' => false));
 342                 $this->assertCount(1, $dt, "Unexpected number of downtimes created from acrossmidnight");
 343
 344                 // Remove downtimes when tests are done.
 345                 $cmd = 'DEL_HOST_DOWNTIME;';
 346                 foreach ($dt as $id) {
 347                         $res = nagioscmd::submit_to_nagios($cmd . $id['id']);
 348                         $this->assertTrue($res, 'Host delete command was submitted');
 349                 }
 350
 351                 $this->assertCount(0, $ls->getDowntimes(array('filter' => array('start_time' => strtotime("{$this->scheduleendtime} +1 day"), 'end_time' => strtotime("{$this->scheduletimeofday} +2 days")))), "Downtimes are gone after deleting them");
 352         }
 353 }