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