def test_disabled_changes_is_valid(self):
manager = server_job.warning_manager()
self.assertEqual(manager.is_valid(15, "MSGTYPE"), True)
manager.disable_warnings("MSGTYPE", lambda: 10)
self.assertEqual(manager.is_valid(15, "MSGTYPE"), False)
def test_multiple_disabled_calls(self):
manager = server_job.warning_manager()
manager.disable_warnings("MSGTYPE", lambda: 10)
manager.disable_warnings("MSGTYPE", lambda: 20)
manager.enable_warnings("MSGTYPE", lambda: 30)
self.assertEqual(manager.is_valid(15, "MSGTYPE"), False)
self.assertEqual(manager.is_valid(25, "MSGTYPE"), False)
self.assertEqual(manager.is_valid(35, "MSGTYPE"), True)
def test_multiple_types(self):
manager = server_job.warning_manager()
manager.disable_warnings("MSGTYPE1", lambda: 10)
manager.disable_warnings("MSGTYPE2", lambda: 20)
manager.enable_warnings("MSGTYPE2", lambda: 30)
self.assertEqual(manager.is_valid(15, "MSGTYPE1"), False)
self.assertEqual(manager.is_valid(15, "MSGTYPE2"), True)
self.assertEqual(manager.is_valid(25, "MSGTYPE1"), False)
self.assertEqual(manager.is_valid(25, "MSGTYPE2"), False)
self.assertEqual(manager.is_valid(35, "MSGTYPE1"), False)
self.assertEqual(manager.is_valid(35, "MSGTYPE2"), True)
if __name__ == "__main__":
unittest.main()
# Open it again to ensure the original gets renamed.
fun_log = console._open_logfile(path)
fun_log.write("it's better than bad, it's good!\n")
fun_log.close()
log_file_list = os.listdir(self.tempdir)
self.assertEqual(2, len(log_file_list))
for name in log_file_list:
self.assertTrue(name.startswith('its-log-log.'),
'unexpected name %s' % name)
self.assertTrue(name.endswith('.gz'), 'unexpected name %s' % name)
def test_logfile_close_signal_handler(self):
self.god.stub_function(os, 'exit')
os.exit.expect_call(1)
logfile = StringIO.StringIO()
console._set_logfile_close_signal_handler(logfile)
try:
self.assertFalse(logfile.closed)
os.kill(os.getpid(), signal.SIGTERM)
finally:
console._unset_signal_handler()
self.god.check_playback()
self.assertTrue(logfile.closed)
if __name__ == '__main__':
unittest.main()
