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: sd_notify("STATUS=%s" % status) if virtWho: virtWho.terminate() queueLogger = log.getQueueLogger() if queueLogger: queueLogger.terminate() sys.exit(code)