test: Add more tests for getTransformationFiles

chrisburr · Oct 2, 2024 · 9078b1c · 9078b1c
1 parent dce673e
commit 9078b1c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/Integration/TransformationSystem/Test_Client_Transformation.py b/tests/Integration/TransformationSystem/Test_Client_Transformation.py
@@ -114,6 +114,17 @@ def test_addTasksAndFiles(self):
         for f in res["Value"]:
             self.assertEqual(f["Status"], TransformationFilesStatus.ASSIGNED)
 
+        # make sure we can selectively select LFNs
+        res = self.transClient.getTransformationFiles({"TransformationID": transID, "LFN": ["/aa/lfn.1.txt"]})
+        assert res["OK"], res
+        assert len(res["Value"]) == 1, res
+        assert "TargetSE" in res["Value"][0].keys(), res
+
+        # make sure we can selectively select columns
+        res = self.transClient.getTransformationFiles({"TransformationID": transID}, columns=["LFN", "Status"])
+        assert res["OK"], res
+        assert res["Value"][0].keys() == ["LFN", "Status"], res
+
         # now adding a new Transformation with new tasks, and introducing a mix of insertion,
         # to test that the trigger works as it should
         res = self.transClient.addTransformation(