fix: Fixed migration so that it works

Author: Robin-Van-de-Merghel

src/DIRAC/WorkloadManagementSystem/FutureClient/PilotManagerClient.py

@@ -6,7 +6,7 @@ class PilotManagerClient(FutureClient):
     @convertToReturnValue
     def addPilotReferences(self, pilot_references, VO, gridType="DIRAC", pilot_stamps_dict={}):
         with DiracXClient() as api:
-            pilot_stamps = list(pilot_stamps_dict.values())
+            pilot_stamps = [pilot_stamps_dict.get(ref, ref) for ref in pilot_references]
             pilot_ref_dict = dict(zip(pilot_stamps, pilot_references))
 
             # We will move toward a stamp as identifier for the pilot

tests/Integration/WorkloadManagementSystem/Test_PilotsClient.py

@@ -30,9 +30,9 @@ def test_PilotsDB():
     for pilot_ref in ["aPilot", "anotherPilot"]:
         res = pilots.deletePilot(pilot_ref)
 
-    res = pilots.addPilotReferences(["aPilot"], "VO")
+    res = pilots.addPilotReferences(["aPilot"], "vo")
     assert res["OK"], res["Message"]
-    res = pilots.addPilotReferences(["aPilot"], "VO")
+    res = pilots.addPilotReferences(["aPilot"], "vo")
 
     # Duplicates to see if we have a conflict
     # If supports diracx, then it should be detected
@@ -44,7 +44,7 @@ def test_PilotsDB():
     res = pilots.deletePilots("aPilot")
     assert res["OK"], res["Message"]
 
-    res = pilots.addPilotReferences(["anotherPilot"], "VO")
+    res = pilots.addPilotReferences(["anotherPilot"], "vo")
     assert res["OK"], res["Message"]
 
     res = pilots.getPilotInfo("anotherPilot")