test: added integration test for PilotAgentsDB summary table

Author: fstagni

src/DIRAC/WorkloadManagementSystem/DB/PilotAgentsDB.py

@@ -1099,7 +1099,6 @@ def getSummarySnapshot(self, requestedFields=False):
         return S_OK(((requestedFields + valueFields), result["Value"]))
 
 
-
 class PivotedPilotSummaryTable:
     """
     The class creates a 'pivoted' table by combining records with the same group

src/DIRAC/WorkloadManagementSystem/DB/PilotAgentsDB.sql

@@ -72,5 +72,5 @@ CREATE TABLE `PilotsHistorySummary` (
   `GridType` VARCHAR(32),
   `Status` VARCHAR(32),
   `PilotCount` INT,
-  PRIMARY KEY (`GridSite`,`GridType`,`Status`),
-) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
+  PRIMARY KEY (`GridSite`,`GridType`,`Status`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

tests/Integration/WorkloadManagementSystem/Test_PilotAgentsDB.py

@@ -5,6 +5,8 @@
 """
 # pylint: disable=wrong-import-position
 
+import csv
+
 from unittest.mock import patch
 
 import DIRAC
@@ -186,3 +188,19 @@ def test_PivotedPilotSummaryTable():
     assert "Total" in columns
 
     cleanUpPilots(pilotRef)
+
+
+def test_summarySnapshot():
+    # insert some predefined jobs to test the summary snapshot
+    with open("pilots.csv", newline="", encoding="utf-8") as csvfile:
+        csvreader = csv.reader(csvfile)
+        data = list(csvreader)
+        placeholders = ",".join(["%s"] * len(data[0]))
+        sql = f"INSERT INTO PilotAgents VALUES ({placeholders})"
+        res = paDB._updatemany(sql, data)
+        assert res["OK"], res["Message"]
+    # Act
+    res = paDB.fillPilotsHistorySummary()
+    assert res["OK"], res["Message"]
+    res = paDB.getSummarySnapshot()
+    assert res["OK"], res["Message"]