unit tests for MSTransferor

dmwm · Oct 28, 2021 · 4acc8cb · 4acc8cb
1 parent c322bdf
commit 4acc8cb
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test/python/WMCore_t/MicroService_t/Tools_t/PycurlRucio_t.py b/test/python/WMCore_t/MicroService_t/Tools_t/PycurlRucio_t.py
@@ -10,7 +10,7 @@
 from Utils.PythonVersion import PY3
 
 from WMCore.MicroService.Tools.PycurlRucio import (getRucioToken, parseNewLineJson,
-                                                   getPileupContainerSizesRucio, listReplicationRules,
+                                                   getPileupContainerSizes, listReplicationRules,
                                                    getBlocksAndSizeRucio, stringDateToEpoch)
 
 CONT1 = "/NoBPTX/Run2018D-12Nov2019_UL2018-v1/MINIAOD"
@@ -46,21 +46,21 @@ def testParseNewLineJson(self):
         for num, item in enumerate(parseNewLineJson(dataStream)):
             self.assertItemsEqual(item, validate[str(num + 1)])
 
-    def testGetPileupContainerSizesRucio(self):
+    def testGetPileupContainerSizes(self):
         """
         Test the getPileupContainerSizesRucio function, which fetches the container
         total bytes.
         """
         self.rucioToken, self.tokenValidity = getRucioToken(self.rucioAuthUrl, self.rucioAccount)
         # Test 1: no DIDs provided as input
-        resp = getPileupContainerSizesRucio([], self.rucioUrl,
-                                            self.rucioToken, scope=self.rucioScope)
+        resp = getPileupContainerSizes([], self.rucioUrl,
+                                       self.rucioToken, scope=self.rucioScope)
         self.assertEqual(resp, {})
 
         # Test 2: multiple valid/invalid DIDs provided as input
         containers = [CONT1, PU_CONT, self.badDID]
-        resp = getPileupContainerSizesRucio(containers, self.rucioUrl,
-                                            self.rucioToken, scope=self.rucioScope)
+        resp = getPileupContainerSizes(containers, self.rucioUrl,
+                                       self.rucioToken, scope=self.rucioScope)
         self.assertTrue(len(resp) == 3)
         self.assertTrue(resp[PU_CONT] > 0)
         self.assertIsNone(resp[self.badDID])