panda-server-source 0.8.0__py3-none-any.whl

pandaserver/api/v1/tests/pilot_api_tests.py

@@ -0,0 +1,154 @@
+# Description: Unit tests for the Pilot API methods
+import os
+import unittest
+from datetime import datetime, timedelta
+
+from pandaserver.api.v1.http_client import HttpClient, api_url_ssl
+
+
+class TestPilotAPI(unittest.TestCase):
+    def setUp(self):
+        self.http_client = HttpClient()
+
+    def test_acquire_jobs(self):
+        url = f"{api_url_ssl}/pilot/acquire_jobs"
+        print(f"Testing URL: {url}")
+        data = {
+            "site_name": "CERN",
+            "timeout": 60,
+            "memory": 999999999,
+            "disk_space": 999999999,
+            "prod_source_label": "managed",
+            "node": "aipanda120.cern.ch",
+            "computing_element": "CERN",
+            "prod_user_id": None,
+            "get_proxy_key": None,
+            "task_id": None,
+            "n_jobs": 1,
+            "background": False,
+            "resource_type": "SCORE",
+            "harvester_id": "imaginary_harvester",
+            "worker_id": 12345,
+            "scheduler_id": "imaginary_scheduler",
+            "job_type": "user",
+            "via_topic": False,
+        }
+
+        status, output = self.http_client.post(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": False, "data": 2, "message": ""}
+        self.assertEqual(output, expected_response)
+
+    def test_acquire_jobs(self):
+        url = f"{api_url_ssl}/pilot/acquire_jobs"
+        print(f"Testing URL: {url}")
+        data = {
+            "site_name": "CERN",
+            "timeout": 60,
+            "memory": 999999999,
+            "disk_space": 999999999,
+            "prod_source_label": "managed",
+            "node": "aipanda120.cern.ch",
+            "computing_element": "CERN",
+            "prod_user_id": None,
+            "get_proxy_key": None,
+            "task_id": None,
+            "n_jobs": 1,
+            "background": False,
+            "resource_type": "SCORE",
+            "harvester_id": "imaginary_harvester",
+            "worker_id": 12345,
+            "scheduler_id": "imaginary_scheduler",
+            "job_type": "user",
+            "via_topic": False,
+        }
+
+        status, output = self.http_client.post(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": False, "data": 2, "message": ""}
+        self.assertEqual(output, expected_response)
+
+    def test_get_job_status(self):
+        url = f"{api_url_ssl}/pilot/get_job_status"
+        print(f"Testing URL: {url}")
+        data = {
+            "job_ids": "4674379299",
+            "timeout": 60,
+        }
+
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": False, "data": 2, "message": ""}
+        self.assertEqual(output, expected_response)
+
+    def test_update_job(self):
+        url = f"{api_url_ssl}/pilot/update_job"
+        print(f"Testing URL: {url}")
+        data = {"job_id": 4674379299, "job_status": "starting"}
+
+        status, output = self.http_client.post(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": False, "data": 2, "message": ""}
+        self.assertEqual(output, expected_response)
+
+    def test_update_worker_node(self):
+        url = f"{api_url_ssl}/pilot/update_worker_node"
+        print(f"Testing URL: {url}")
+        data = {
+            "site": "CERN",
+            "panda_queue": "CERN",
+            "host_name": "slot1@wn1.cern.ch",
+            "cpu_model": "AMD EPYC 7B12",
+            "n_logical_cpus": 64,
+            "n_sockets": 2,
+            "cores_per_socket": 8,
+            "threads_per_core": 2,
+            "cpu_architecture": "x86_64",
+            "cpu_architecture_level": "x86_64-v3",
+            "clock_speed": 2.7,
+            "total_memory": 3350,
+            "total_local_disk": 75,
+        }
+
+        status, output = self.http_client.post(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": True, "data": None, "message": "Inserted new worker node."}
+        self.assertEqual(output, expected_response)
+
+    def test_update_worker_node_gpu(self):
+        url = f"{api_url_ssl}/pilot/update_worker_node_gpu"
+        print(f"Testing URL: {url}")
+        data = {
+            "site": "CERN",
+            "host_name": "slot1@lxplus940.cern.ch",
+            "vendor": "NVIDIA",
+            "model": "Tesla T4",
+            "architecture": "Turing",
+            "vram": 15360,
+            "framework": "CUDA",
+            "framework_version": "12.9",
+            "driver_version": "575.51.03",
+            "count": 1,
+        }
+
+        status, output = self.http_client.post(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": True, "data": None, "message": "Inserted new worker node GPU."}
+        self.assertEqual(output, expected_response)
+
+
+# Run tests
+if __name__ == "__main__":
+    unittest.main()

pandaserver/api/v1/tests/statistics_api_tests.py

@@ -0,0 +1,104 @@
+# Description: Unit tests for the Statistics API methods
+import unittest
+
+from pandaserver.api.v1.http_client import HttpClient, api_url_ssl
+
+
+class TestTaskAPI(unittest.TestCase):
+    def setUp(self):
+        self.http_client = HttpClient()
+
+    def test_job_stats_by_cloud(self):
+        url = f"{api_url_ssl}/statistics/job_stats_by_cloud"
+        print(f"Testing URL: {url}")
+        data = {"type": "production"}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        # Remove the data field from the response for comparison
+        del output["data"]
+
+        expected_response = {"success": True, "message": "", "status": 0}
+
+        self.assertEqual(output, expected_response)
+
+    def test_production_job_stats_by_cloud_and_processing_type(self):
+        url = f"{api_url_ssl}/statistics/production_job_stats_by_cloud_and_processing_type"
+        print(f"Testing URL: {url}")
+        data = {}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        # Remove the data field from the response for comparison
+        del output["data"]
+
+        expected_response = {"success": True, "message": "", "status": 0}
+
+        self.assertEqual(output, expected_response)
+
+    def test_active_job_stats_by_site(self):
+        url = f"{api_url_ssl}/statistics/active_job_stats_by_site"
+        print(f"Testing URL: {url}")
+        data = {}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        # Remove the data field from the response for comparison
+        del output["data"]
+
+        expected_response = {"success": True, "message": "", "status": 0}
+
+        self.assertEqual(output, expected_response)
+
+    def test_active_job_detailed_stats_by_site(self):
+        url = f"{api_url_ssl}/statistics/active_job_detailed_stats_by_site"
+        print(f"Testing URL: {url}")
+        data = {}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        # Remove the data field from the response for comparison
+        del output["data"]
+
+        expected_response = {"success": True, "message": "", "status": 0}
+
+        self.assertEqual(output, expected_response)
+
+    def test_job_stats_by_site_and_resource_type(self):
+        url = f"{api_url_ssl}/statistics/job_stats_by_site_and_resource_type"
+        print(f"Testing URL: {url}")
+        data = {"time_window": 12 * 60}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        # Remove the data field from the response for comparison
+        del output["data"]
+
+        expected_response = {"success": True, "message": "", "status": 0}
+
+        self.assertEqual(output, expected_response)
+
+    def test_job_stats_by_site_share_and_resource_type(self):
+        url = f"{api_url_ssl}/statistics/job_stats_by_site_share_and_resource_type"
+        print(f"Testing URL: {url}")
+        data = {"time_window": 12 * 60}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        # Remove the data field from the response for comparison
+        del output["data"]
+
+        expected_response = {"success": True, "message": "", "status": 0}
+
+        self.assertEqual(output, expected_response)
+
+
+# Run tests
+if __name__ == "__main__":
+    unittest.main()

pandaserver/api/v1/tests/system_api_tests.py

@@ -0,0 +1,58 @@
+# Description: Unit tests for the System API methods
+import unittest
+
+from pandaserver.api.v1.http_client import HttpClient, api_url_ssl
+
+
+class TestSystemAPI(unittest.TestCase):
+    def setUp(self):
+        self.http_client = HttpClient()
+
+    def test_get_attributes(self):
+        url = f"{api_url_ssl}/system/get_attributes"
+        print(f"Testing URL: {url}")
+        data = {"test_key_string": "test_value", "test_key_int": 12345}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": True}
+        self.assertEqual(output, expected_response)
+
+    def test_get_user_attributes(self):
+        url = f"{api_url_ssl}/system/get_user_attributes"
+        print(f"Testing URL: {url}")
+        data = {}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": True}
+        self.assertEqual(output, expected_response)
+
+    def test_get_voms_attributes(self):
+        url = f"{api_url_ssl}/system/get_voms_attributes"
+        print(f"Testing URL: {url}")
+        data = {}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": True}
+        self.assertEqual(output, expected_response)
+
+    def test_is_alive(self):
+        url = f"{api_url_ssl}/system/is_alive"
+        print(f"Testing URL: {url}")
+        data = {}
+        status, output = self.http_client.get(url, data)
+        print(output)
+        output["status"] = status
+
+        expected_response = {"status": 0, "success": True}
+        self.assertEqual(output, expected_response)
+
+
+# Run tests
+if __name__ == "__main__":
+    unittest.main()