test/test_notification_warning.py

   1 #!/usr/bin/env python2.6
   2 #Copyright (C) 2009-2010 :
   3 #    Gabes Jean, naparuba@gmail.com
   4 #    Gerhard Lausser, Gerhard.Lausser@consol.de
   5 #
   6 #This file is part of Shinken.
   7 #
   8 #Shinken is free software: you can redistribute it and/or modify
   9 #it under the terms of the GNU Affero General Public License as published by
  10 #the Free Software Foundation, either version 3 of the License, or
  11 #(at your option) any later version.
  12 #
  13 #Shinken is distributed in the hope that it will be useful,
  14 #but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 #GNU Affero General Public License for more details.
  17 #
  18 #You should have received a copy of the GNU Affero General Public License
  19 #along with Shinken.  If not, see <http://www.gnu.org/licenses/>.
  20
  21 #
  22 # This file is used to test reading and processing of config files
  23 #
  24
  25
  26 from shinken_test import unittest, ShinkenTest
  27
  28 from shinken.notification import Notification
  29
  30
  31 class TestConfig(ShinkenTest):
  32     #setUp is in shinken_test
  33
  34     #Change ME :)
  35     def test_raise_warning_on_notification_errors(self):
  36         host = self.sched.hosts.find_by_name("test_host_0")
  37         host.checks_in_progress = []
  38         host.act_depend_of = [] # ignore the router
  39         cmd = "/error/pl"
  40         #Create a dummy notif
  41         n = Notification('PROBLEM', 'scheduled', 'BADCOMMAND', cmd, host, None, 0)
  42         n.execute()
  43         self.sched.actions[n.id] = n
  44         self.sched.put_results(n)
  45         #Should have raised something like "Warning : the notification command 'BADCOMMAND' raised an error (exit code=2) : '[Errno 2] No such file or directory'"
  46         self.assert_(self.log_match(1, 'BADCOMMAND'))
  47
  48
  49 if __name__ == '__main__':
  50     unittest.main()
  51