def test_get_queue_logger(self, queueloggerClass): queueloggerClass.return_value = Mock() # These two should be the same Mock object defaultQueueLogger = log.getQueueLogger() self.addCleanup(defaultQueueLogger.terminate()) secondQueueLogger = log.getQueueLogger() self.assertTrue(defaultQueueLogger == secondQueueLogger) defaultQueueLogger.start_logging.assert_any_call()
def exit(code, status=None): """ exits with the code provided, properly disposing of resources If status is not None, use sd_notify to report the status to systemd """ if status is not None and status != "": sd_notify("STATUS=%s" % status) if executor: executor.logger.debug(str(status)) try: executor.terminate() except KeyboardInterrupt: signal.signal(signal.SIGINT, signal.SIG_IGN) for v in executor.virts: v.stop() if v.ident: v.join() for d in executor.destinations: d.stop() if d.ident: d.join() if log.hasQueueLogger(): queueLogger = log.getQueueLogger() queueLogger.terminate() sys.exit(code)
def exit(code, status=None): """ exits with the code provided, properly disposing of resources If status is not None, use sd_notify to report the status to systemd """ if status is not None and status != "": sd_notify("STATUS=%s" % status) if executor: try: executor.terminate() except KeyboardInterrupt: signal.signal(signal.SIGINT, signal.SIG_IGN) for v in executor.virts: v.stop() if v.ident: v.join() for d in executor.destinations: d.stop() if d.ident: d.join() if log.hasQueueLogger(): queueLogger = log.getQueueLogger() queueLogger.terminate() sys.exit(code)