Skip to content

Commit

Permalink
add key,value non equality check, the opposite of the -q and -Q
Browse files Browse the repository at this point in the history
  • Loading branch information
Robert Nemeti committed Jan 10, 2018
1 parent 89f42c1 commit d164a12
Showing 1 changed file with 27 additions and 1 deletion.
28 changes: 27 additions & 1 deletion check_http_json.py
Original file line number Diff line number Diff line change
Expand Up @@ -168,6 +168,15 @@ def checkEquality(self, equality_list):
failure += " Value for key %s did not match %s." % (alias, v)
return failure

def checkNonEquality(self, equality_list):
failure = ''
for kv in equality_list:
k, v = kv.split(',')
key, alias = _getKeyAlias(k)
if (self.helper.equals(key, v) == True):
failure += " Value for key %s matches %s." % (alias, v)
return failure

def checkThreshold(self, key, alias, r):
failure = ''
invert = False
Expand Down Expand Up @@ -214,6 +223,8 @@ def checkWarning(self):
failure += self.checkThresholds(self.rules.key_threshold_warning)
if self.rules.key_value_list != None:
failure += self.checkEquality(self.rules.key_value_list)
if self.rules.key_value_list_not != None:
failure += self.checkNonEquality(self.rules.key_value_list_not)
if self.rules.key_list != None:
failure += self.checkExists(self.rules.key_list)
return failure
Expand All @@ -224,6 +235,8 @@ def checkCritical(self):
failure += self.checkThresholds(self.rules.key_threshold_critical)
if self.rules.key_value_list_critical != None:
failure += self.checkEquality(self.rules.key_value_list_critical)
if self.rules.key_value_list_not_critical != None:
failure += self.checkNonEquality(self.rules.key_value_list_not_critical)
if self.rules.key_list_critical != None:
failure += self.checkExists(self.rules.key_list_critical)
return failure
Expand Down Expand Up @@ -296,6 +309,11 @@ def parseArgs():
Multiple key values can be delimited with colon (key,value1:value2). Return warning if equality check fails')
parser.add_argument('-Q', '--key_equals_critical', dest='key_value_list_critical', nargs='*',
help='Same as -q but return critical if equality check fails.')
parser.add_argument('-y', '--key_not_equals', dest='key_value_list_not', nargs='*',
help='Checks equality of these keys and values (key[>alias],value key2,value2) to determine status.\
Multiple key values can be delimited with colon (key,value1:value2). Return warning if equality check succeeds')
parser.add_argument('-Y', '--key_not_equals_critical', dest='key_value_list_not_critical', nargs='*',
help='Same as -q but return critical if equality check succeeds.')
parser.add_argument('-m', '--key_metric', dest='metric_list', nargs='*',
help='Gathers the values of these keys (key[>alias],UnitOfMeasure,WarnRange,CriticalRange,Min,Max) for Nagios performance data.\
More information about Range format and units of measure for nagios can be found at nagios-plugins.org/doc/guidelines.html\
Expand All @@ -315,7 +333,7 @@ def debugPrint(debug_flag, message, pretty_flag=False):
class RulesHelper:
separator = '.'
debug = False
key_threshold_warning,key_value_list,key_list,key_threshold_critical,key_value_list_critical,key_list_critical,metric_list = None, None, None, None, None, None, None
key_threshold_warning,key_value_list,key_value_list_not,key_list,key_threshold_critical,key_value_list_critical,key_value_list_not_critical,key_list_critical,metric_list = None, None, None, None, None, None, None, None, None
def dash_m(self, data):
self.metric_list = data; return self
def dash_e(self, data):
Expand All @@ -326,6 +344,10 @@ def dash_q(self, data):
self.key_value_list = data; return self
def dash_Q(self, data):
self.key_value_list_critical = data; return self
def dash_y(self, data):
self.key_value_list_not = data; return self
def dash_Y(self, data):
self.key_value_list_not_critical = data; return self
def dash_w(self, data):
self.key_threshold_warning = data; return self
def dash_c(self, data):
Expand Down Expand Up @@ -354,6 +376,10 @@ def test_equality(self):
self.check_data(RulesHelper().dash_q(['metric,6']), '{"metric": 5}', WARNING_CODE)
self.check_data(RulesHelper().dash_Q(['metric,6']), '{"metric": 5}', CRITICAL_CODE)
self.check_data(RulesHelper().dash_q(['metric,5']), '{"metric": 5}', OK_CODE)
def test_non_equality(self):
self.check_data(RulesHelper().dash_y(['metric,6']), '{"metric": 6}', WARNING_CODE)
self.check_data(RulesHelper().dash_Y(['metric,6']), '{"metric": 6}', CRITICAL_CODE)
self.check_data(RulesHelper().dash_y(['metric,5']), '{"metric": 6}', OK_CODE)
def test_warning_thresholds(self):
self.check_data(RulesHelper().dash_w(['metric,5']), '{"metric": 5}', OK_CODE)
self.check_data(RulesHelper().dash_w(['metric,5:']), '{"metric": 5}', OK_CODE)
Expand Down

0 comments on commit d164a12

Please sign in to comment.