test: Add more tests for getTransformationFiles

tests/Integration/TransformationSystem/Test_Client_Transformation.py

@@ -114,6 +114,21 @@ 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 res["ParameterNames"][0] == ["LFN"], res
+        assert "TargetSE" in res["ParameterNames"][0], res
+        assert isinstance(res["Records"][0], list), 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
+        assert res["ParameterNames"] == ["LFN", "Status"], res
+        assert isinstance(res["Records"][0], list), 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(