def test_log_ctrlc_and_exit(capfd, caplog):
# GIVEN opslog invocation
# WHEN the ops_utils.log_ctrlc_and_exit function is invoked and
# the output is captured
with pytest.raises(SystemExit) as e:
ops_utils.log_ctrlc_and_exit()
out, err = capfd.readouterr()
# THEN a SystemExit exception should be raised and
# the exit status should be 130 and
# there should be only a single newline written to stderr and
# there should be a single log message and
# the log level of the message should be INFO and
# the log message should match the expected text
assert int(str(e.value)) == 130
assert err == "\n"
assert len(caplog.records) == 1
assert caplog.records[0].levelno == logging.INFO
assert caplog.records[0].message == "(130) Halted via KeyboardInterrupt."
def test_log_ctrlc_and_exit(capfd, caplog):
# GIVEN opslog invocation
# WHEN the ops_utils.log_ctrlc_and_exit function is invoked and
# the output is captured
with pytest.raises(SystemExit) as e:
ops_utils.log_ctrlc_and_exit()
out, err = capfd.readouterr()
# THEN a SystemExit exception should be raised and
# the exit status should be 130 and
# there should be only a single newline written to stderr and
# there should be a single log message and
# the log level of the message should be INFO and
# the log message should match the expected text
assert int(str(e.value)) == 130
assert err == "\n"
assert len(caplog.records) == 1
assert caplog.records[0].levelno == logging.INFO
assert caplog.records[0].message == "(130) Halted via KeyboardInterrupt."
def test_log_ctrlc_and_exit__without_logging(capfd):
# GIVEN the caplog logging handler is removed and
# the expected output to stderr
root_logger = logging.getLogger()
handler = root_logger.handlers[0]
root_logger.removeHandler(handler)
expected_err = ("\nCRITICAL No handlers could be found for logger \"{0}\""
"\nINFO (130) Halted via KeyboardInterrupt.\n"
.format(MODULE))
# WHEN the ops_utils.log_ctrlc_and_exit function is invoked and
# the output is captured
with pytest.raises(SystemExit) as e:
ops_utils.log_ctrlc_and_exit()
out, err = capfd.readouterr()
# THEN a SystemExit exception should be raised and
# the exit status should be 130 and
# stderr should match the expected error message
assert int(str(e.value)) == 130
assert err == expected_err
def test_log_ctrlc_and_exit__without_logging(capfd):
# GIVEN the caplog logging handler is removed and
# the expected output to stderr
root_logger = logging.getLogger()
handler = root_logger.handlers[0]
root_logger.removeHandler(handler)
expected_err = ("\nCRITICAL No handlers could be found for logger \"{0}\""
"\nINFO (130) Halted via KeyboardInterrupt.\n"
.format(MODULE))
# WHEN the ops_utils.log_ctrlc_and_exit function is invoked and
# the output is captured
with pytest.raises(SystemExit) as e:
ops_utils.log_ctrlc_and_exit()
out, err = capfd.readouterr()
# THEN a SystemExit exception should be raised and
# the exit status should be 130 and
# stderr should match the expected error message
assert int(str(e.value)) == 130
assert err == expected_err
"""
add_args = {"config": True, "EMAIL": True, "dryrun": True,
"verbosity": True}
cap = ops_config.OpsConfigArgParse(description=__doc__, add_args=add_args)
logger = ops_logging.OpScriptsLogging(cap.prog)
args = ops_config.parse_args(cap)
logger.set_log_level(args.verbosity)
logger.dryrun(args.dryrun)
return args
def main():
args = setup()
subject = "Test - disregard"
body = "This is a test message"
message = ops_notify_email.Message(args.program_name, subject, body)
message.send(args.email_from, args.email_to, dryrun=args.dryrun)
if __name__ == "__main__":
try:
main()
except SystemExit as e:
sys.exit(e.code)
except KeyboardInterrupt:
ops_utils.log_ctrlc_and_exit()
except ops_utils.Fatal:
ops_utils.log_fatal_and_exit()
except:
ops_utils.log_exception_and_exit()
"dryrun": True,
"verbosity": True
}
cap = ops_config.OpsConfigArgParse(description=__doc__, add_args=add_args)
logger = ops_logging.OpScriptsLogging(cap.prog)
args = ops_config.parse_args(cap)
logger.set_log_level(args.verbosity)
logger.dryrun(args.dryrun)
return args
def main():
args = setup()
subject = "Test - disregard"
body = "This is a test message"
message = ops_notify_email.Message(args.program_name, subject, body)
message.send(args.email_from, args.email_to, dryrun=args.dryrun)
if __name__ == "__main__":
try:
main()
except SystemExit as e:
sys.exit(e.code)
except KeyboardInterrupt:
ops_utils.log_ctrlc_and_exit()
except ops_utils.Fatal:
ops_utils.log_fatal_and_exit()
except:
ops_utils.log_exception_and_exit()
