Added support for reporting boolean for matching status code

Author: toshke

README.md

@@ -34,6 +34,8 @@ their namespace, defaults to 'HttpCheck'
 metric to be published as well, with value determined by success of matching response body against
 regular expression contained within this option
 
+`STATUS_CODE_MATCH` - report whether http status code is equal to given status code or not. If this option
+is not present, it won't be reported upon. Defaults to empty
 
 ## Outputs
 
@@ -50,6 +52,9 @@ is 2 minutes for http requests.
 
 `ResponseBody` - Optional, by default this won't be reported
 
+`ResponseBodyRegexMatch` - Optional, if `BODY_REGEX_MATCH` option is provided
+
+`StatusCodeMatch` - Optional, if `STATUS_CODE_MATCH` options is provided
 
 ## Dependencies
 
@@ -71,6 +76,9 @@ server timeout
 - `ResponseBodyRegexMatch` - **optional** this will report 1 or 0 if `BODY_REGEX_MATCH` option is specified. 1 is reported
  if response body matches regex provided, or 0 otherwise. 
 
+- `StatusCodeMatch` - **optional*& this will report 1 or 0 if `STATUS_CODE_MATCH` options is specified. 1 is reported
+ if response status code matches code provided, or 0 otherwise
+
 ## Deployment
 
 You can either deploy Lambda manually, or through [serverless](serverless.com) project.

handler.py

@@ -22,6 +22,7 @@ class Config:
     CW_METRICS_NAMESPACE = 'CW_METRICS_NAMESPACE'
     CW_METRICS_METRIC_NAME = 'CW_METRICS_METRIC_NAME'
     BODY_REGEX_MATCH = 'BODY_REGEX_MATCH'
+    STATUS_CODE_MATCH = 'STATUS_CODE_MATCH'
 
     def __init__(self, event):
         self.event = event
@@ -34,7 +35,8 @@ def __init__(self, event):
             self.REPORT_AS_CW_METRICS: '1',
             self.CW_METRICS_NAMESPACE: 'HttpCheck',
             self.HEADERS: '',
-            self.BODY_REGEX_MATCH: None
+            self.BODY_REGEX_MATCH: None,
+            self.STATUS_CODE_MATCH: None
         }
 
     def __get_property(self, property_name):
@@ -83,6 +85,10 @@ def headers(self):
     @property
     def bodyregexmatch(self):
         return self.__get_property(self.BODY_REGEX_MATCH)
+
+    @property
+    def statuscodematch(self):
+        return self.__get_property(self.STATUS_CODE_MATCH)
 
     @property
     def cwoptions(self):
@@ -102,6 +108,7 @@ def __init__(self, config):
         self.payload = config.payload
         self.headers = config.headers
         self.bodyregexmatch = config.bodyregexmatch
+        self.statuscodematch = config.statuscodematch
 
     def execute(self):
         url = urlparse(self.endpoint)
@@ -145,6 +152,9 @@ def execute(self):
                 regex = re.compile(self.bodyregexmatch)
                 value = 1 if regex.match(response_body) else 0
                 result['ResponseBodyRegexMatch'] = value
+
+            if self.statuscodematch is not None:
+                result['StatusCodeMatch'] = int(int(response_data.status) == int(self.statuscodematch))
 
             # return structure with data
             return result
@@ -189,15 +199,16 @@ def report(self, result):
                         'Unit': 'None',
                         'Value': int(result['StatusCode'])
                     })
-                    if 'ResponseBodyRegexMatch' in result:
-                        metric_data.append({
-                            'MetricName': 'ResponseBodyRegexMatch',
-                            'Dimensions': [
-                                {'Name': 'Endpoint', 'Value': self.endpoint}
-                            ],
-                            'Unit': 'None',
-                            'Value': int(result['ResponseBodyRegexMatch'])
-                        })
+                    for additional_metric in ['ResponseBodyRegexMatch', 'StatusCodeMatch']:
+                        if additional_metric in result:
+                            metric_data.append({
+                                'MetricName': additional_metric,
+                                'Dimensions': [
+                                    {'Name': 'Endpoint', 'Value': self.endpoint}
+                                ],
+                                'Unit': 'None',
+                                'Value': int(result[additional_metric])
+                            })
 
                 result = cloudwatch.put_metric_data(
                     MetricData=metric_data,

test/ipify.json

@@ -5,5 +5,6 @@
   "REPORT_RESPONSE_BODY":"1",
   "REPORT_AS_CW_METRICS":"1",
   "CW_METRICS_NAMESPACE":"HttpCheckTestNamespace1",
-  "BODY_REGEX_MATCH":"\\{\"ip\":(.*)\\}"
+  "BODY_REGEX_MATCH":"\\{\"ip\":(.*)\\}",
+  "STATUS_CODE_MATCH": 200
 }

test/postRequest.json

@@ -5,5 +5,6 @@
   "PAYLOAD":"{\"title\":\"foo\",\"body\":\"bar\",\"userId\":1}",
   "REPORT_RESPONSE_BODY":"1",
   "REPORT_AS_CW_METRICS":"1",
-  "CW_METRICS_NAMESPACE":"HttpCheckTestNamespace"
+  "CW_METRICS_NAMESPACE":"HttpCheckTestNamespace",
+  "STATUS_CODE_MATCH": 201
 }