test: Added tests for pilot history monitoring

Author: rupozzi

src/DIRAC/WorkloadManagementSystem/Agent/PilotsHistoryAgent.py

@@ -37,6 +37,8 @@ def execute(self):
         for record in values:
             record = record[1:]
             rD = {}
+            for iP in range(len(self.__summaryKeyFieldsMapping)):
+                rD[self.__summaryKeyFieldsMapping[iP]] = record[iP]
             record = record[len(self.__summaryKeyFieldsMapping) :]
             for iP in range(len(self.__summaryValueFieldsMapping)):
                 rD[self.__summaryValueFieldsMapping[iP]] = int(record[iP])

tests/Integration/Monitoring/Test_MonitoringReporter.py

@@ -63,7 +63,7 @@
 wmsMonitoringReporter = MonitoringReporter(monitoringType="WMSHistory")
 componentMonitoringReporter = MonitoringReporter(monitoringType="ComponentMonitoring")
 pilotMonitoringReporter = MonitoringReporter(monitoringType="PilotSubmissionMonitoring")
-
+pilotsHistoryReporter = MonitoringReporter(monitoringType="PilotsHistory")
 data = [
     {
         "Status": "Waiting",
@@ -852,6 +852,25 @@
     },
 ]
 
+pilotsHistoryData = [
+    {
+        "TaskQueueID": "1240451",
+        "GridSite": "LCG.CNAF.it",
+        "GridType": "",
+        "Status": "failed",
+        "Pilots": "7",
+        "timestamp": 1649161714,
+    },
+    {
+        "TaskQueueID": "12401",
+        "GridSite": "LCG.CNAF.it",
+        "GridType": "",
+        "Status": "failed",
+        "Pilots": "7",
+        "timestamp": 1649161714,
+    },
+]
+
 
 def test_addWMSRecords():
     for record in data:
@@ -875,3 +894,11 @@ def test_addPilotSubmissionRecords():
     result = pilotMonitoringReporter.commit()
     assert result["OK"]
     assert result["Value"] == len(pilotMonitoringData)
+
+
+def test_addPilotHistoryRecords():
+    for record in pilotsHistoryData:
+        pilotsHistoryReporter.addRecord(record)
+    result = pilotsHistoryReporter.commit()
+    assert result["OK"]
+    assert result["Value"] == len(pilotsHistoryData)