Merge pull request #94 from SumoLogic/SUMO-238205-SumoMetricsQuery

SUMO-238205: Added Metric Sumo Query and Unit test

Author: himanshu219

EventHubs/tests/test_eventhub_metrics.py

@@ -52,6 +52,26 @@ def test_03_func_logs(self):
         self.assertFalse(self.filter_logs(captured_output, 'severityLevel', '2'),
                          "Warning messages found in azure function logs")
 
+    def test_04_sumo_query_record_count(self):
+        self.logger.info("fetching metrix data count from sumo")
+        query = f'_sourceCategory="{self.source_category}" | count'
+        relative_time_in_minutes = 30
+        expected_record_count = 10
+        result = self.fetch_sumo_MetrixQuery_results(query, relative_time_in_minutes)
+        #sample: {"error":false,"errorMessage":null,"errorInstanceId":null,"errorKey":null,"keyedErrors":[],"response":[{"rowId":"A","results":[{"metric":{"dimensions":[{"key":"metric","value":"count"}],"algoId":1},"horAggs":{"min":1.0,"max":17.0,"avg":2.0,"sum":32.0,"count":16,"latest":1.0},"datapoints":{"timestamp":[],"value":[],"outlierParams":[],"max":[],"min":[],"avg":[],"count":[],"isFilled":[]}}]}],"queryInfo":{"startTime":1711710360000,"endTime":1711710460000,"desiredQuantizationInSecs":{"empty":false,"defined":true},"actualQuantizationInSecs":1,"sessionIdStr":""}}
+        self.assertFalse(result['error'],
+                        f"Metrix sumo query failed with error message {result['errorMessage']}")
+        try:
+            if result['error']:
+                record_count = 0
+            else:
+                record_count = result['response'][0]['results'][0]['datapoints']['value'][0]
+        except Exception:
+            record_count = 0
+        self.assertTrue(record_count == expected_record_count,
+                        f"Metrix record count: {record_count} differs from expected count {expected_record_count} in sumo '{self.source_category}'")
+
+
     def insert_mock_metrics_in_EventHub(self, filename):
         self.logger.info("inserting fake metrics in EventHub")
 

test_utils/basetest.py

@@ -300,3 +300,16 @@ def fetch_sumo_query_results(cls, query='_sourceCategory="azure_br_logs" | count
             cls.logger.info(f"source result: {result}")
             return result
         return
+
+    @classmethod
+    def fetch_sumo_MetrixQuery_results(cls, query='_sourceCategory="azure_br_logs" | count', relative_time_in_minutes=15):
+
+        toTime = datetime.now()
+        fromTime = toTime + timedelta(minutes=-1*relative_time_in_minutes)
+
+        cls.logger.info(
+            f"query: {query}, fromTime: {fromTime.isoformat(timespec='seconds')}, toTime: {toTime.isoformat(timespec='seconds')}")
+
+        search_result = cls.sumologic_cli.search_metrics(query, int(fromTime.timestamp())*1000, int(toTime.timestamp())*1000)
+        cls.logger.info(f"source result: {search_result}")
+        return search_result