feat (test): Added tests for pilot submission to monitoring

Author: rupozzi

tests/Integration/Monitoring/Test_MonitoringDB.py

@@ -1,12 +1,6 @@
 """ Test for MonitoringDB
 """
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
-__RCSID__ = "$Id$"
-
 import time
 import json
 import pytest

tests/Integration/Monitoring/Test_MonitoringReporter.py

@@ -54,9 +54,6 @@
 
 # pylint: disable=invalid-name,wrong-import-position
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 import unittest
 import sys
 
@@ -75,6 +72,7 @@ def setUp(self):
 
         self.wmsMonitoringReporter = MonitoringReporter(monitoringType="WMSHistory")
         self.componentMonitoringReporter = MonitoringReporter(monitoringType="ComponentMonitoring")
+        self.pilotMonitoringReporter = MonitoringReporter(monitoringType="PilotMonitoring")
 
         self.data = [
             {
@@ -838,6 +836,20 @@ def setUp(self):
                 u"timestamp": 1458226213,
             },
         ]
+        # Test data for the PilotMonitoring type
+        self.pilotMonitoringData = [
+            {
+                "HostName": "blabla",
+                "SiteDirector": "blabla",
+                "Site": "blabla",
+                "CE": "blabla",
+                "Queue": "blabla",
+                "Status": "blabla",
+                "NumTotal": "10",
+                "NumSucceded": "3",
+                "timestamp": 1458226213,
+            }
+        ]
 
     def tearDown(self):
         pass
@@ -858,6 +870,13 @@ def test_addComponentRecords(self):
         self.assertTrue(result["OK"])
         self.assertEqual(result["Value"], len(self.activityMonitoringData))
 
+    def test_addPilotRecords(self):
+        for record in self.pilotMonitoringData:
+            self.pilotMonitoringReporter.addRecord(record)
+        result = self.pilotMonitoringReporter.commit()
+        self.assertTrue(result["OK"])
+        self.assertEqual(result["Value"], len(self.pilotMonitoringData))
+
 
 if __name__ == "__main__":
     testSuite = unittest.defaultTestLoader.loadTestsFromTestCase(MonitoringTestCase)