add tests for utils.getNodesQueue

CMSCompOps · Sep 21, 2020 · 26063c3 · 26063c3
1 parent f7a7614
commit 26063c3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 6 deletions.
diff --git a/.github/workflows/run_test.yml b/.github/workflows/run_test.yml
@@ -3,7 +3,7 @@ name: Unit test
 on: [push, pull_request]
 
 jobs:
-  lint-python:
+  unit-test:
     runs-on: ['self-hosted', 'Linux', 'X64']
     steps:
     - name: Checkout

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -820,15 +820,24 @@ def getresponse(self):
                 response, [[{'name': 'someSite'}, {'name': 'someSite1'}]])
 
 
-class TestGetAllAgents(unittest.TestCase):
-    def test_getAllAgents(self):
-        from WmAgentScripts.utils import getBlockLocations as getAllAgents
+class TestGetBlockLocations(unittest.TestCase):
+    def test_getBlockLocations(self):
+        from WmAgentScripts.utils import getBlockLocations
 
         with patch('WmAgentScripts.utils.make_x509_conn', mock_make_x509_conn):
-            response = getAllAgents(url=reqmgr_url, dataset=test_dataset)
+            response = getBlockLocations(url=reqmgr_url, dataset=test_dataset)
             self.assertDictEqual(response, {})
 
 
+class TestGetNodesQueue(unittest.TestCase):
+    def test_getNodesQueue(self):
+        from WmAgentScripts.utils import getNodesQueue
+
+        with patch('WmAgentScripts.utils.make_x509_conn', mock_make_x509_conn):
+            response = getNodesQueue(url=reqmgr_url)
+            self.assertIn('T3_RU_MEPhI', response)
+
+
 class TestAgentSpeedDraining(unittest.TestCase):
     def test_agent_speed_draining(self):
         from WmAgentScripts.utils import agent_speed_draining
@@ -839,7 +848,7 @@ def test_agent_speed_draining(self):
 
 
 class TestBatchInfo(unittest.TestCase):
-    def test_content(self):
+    def test_batchInfo(self):
         from WmAgentScripts.utils import batchInfo
 
         with patch('WmAgentScripts.utils.mongo_client', mock_mongo_client):