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()