SUMO-238205: Modified sumo metric query

Author: priyansh-sumo

EventHubs/tests/test_eventhub_metrics.py

@@ -60,11 +60,13 @@ def test_04_sumo_query_record_count(self):
         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 = len(result['response'][0]['results'][0]['datapoints']['value'])
+                record_count = result['response'][0]['results'][0]['datapoints']['value'][0]
         except Exception:
             record_count = 0
         self.assertTrue(record_count == expected_record_count,

test_utils/basetest.py

@@ -304,13 +304,12 @@ def fetch_sumo_query_results(cls, query='_sourceCategory="azure_br_logs" | count
     @classmethod
     def fetch_sumo_MetrixQuery_results(cls, query='_sourceCategory="azure_br_logs" | count', relative_time_in_minutes=15):
 
-        toTime = datetime.utcnow()
+        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')}")
 
-        # def search_metrics(self, query, fromTime=None, toTime=None, requestedDataPoints=600, maxDataPoints=800)
-        search_result = cls.sumologic_cli.search_metrics(query, fromTime, toTime)
+        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