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)
