def test_no_checks(self):
service_checks.CRITICAL_CHECKS = ()
service_checks.WARNING_CHECKS = ()
service_status, check_info = service_checks.run_checks()
self.assertEqual(service_status, service_checks.ServiceStatus.OK)
self.assertEqual(check_info, {})
def test_warning_one_pass_one_fail(self):
def warn_pass():
return (True, {"info": "warn pass"})
def warn_fail():
return (False, {"info": "warn fail"})
service_checks.CRITICAL_CHECKS = ()
service_checks.WARNING_CHECKS = (warn_pass, warn_fail)
service_status, check_info = service_checks.run_checks()
self.assertEqual(service_status, service_checks.ServiceStatus.WARNING)
self.assertEqual(
{
"warn_fail": {
"info": "warn fail"
},
"warn_pass": {
"info": "warn pass"
},
},
check_info,
)
def test_critical_one_pass_one_fail(self):
def crit_pass():
return (True, {"info": "crit pass"})
def crit_fail():
return (False, {"info": "crit fail"})
service_checks.CRITICAL_CHECKS = (crit_fail, crit_pass)
service_checks.WARNING_CHECKS = ()
service_status, check_info = service_checks.run_checks()
self.assertEqual(service_status, service_checks.ServiceStatus.CRITICAL)
self.assertEqual(
{
"crit_fail": {
"info": "crit fail"
},
"crit_pass": {
"info": "crit pass"
},
},
check_info,
)
def test_critical_and_warning_fail(self):
def crit_fail():
return (False, {"info": "crit fail"})
def warn_fail():
return (False, {"info": "warn fail"})
service_checks.CRITICAL_CHECKS = (crit_fail, )
service_checks.WARNING_CHECKS = (warn_fail, )
service_status, check_info = service_checks.run_checks()
self.assertEqual(service_status, service_checks.ServiceStatus.CRITICAL)
self.assertEqual(
{
"crit_fail": {
"info": "crit fail"
},
"warn_fail": {
"info": "warn fail"
},
},
check_info,
)
def test_critical_and_warning_pass(self):
def crit_pass():
return (True, {"info": "crit pass"})
def warn_pass():
return (True, {"info": "warn pass"})
service_checks.CRITICAL_CHECKS = (crit_pass, )
service_checks.WARNING_CHECKS = (warn_pass, )
service_status, check_info = service_checks.run_checks()
self.assertEqual(service_status, service_checks.ServiceStatus.OK)
self.assertEqual(
{
"crit_pass": {
"info": "crit pass"
},
"warn_pass": {
"info": "warn pass"
},
},
check_info,
)