Added test for curator_products wrester

apps/cli/utilities/aat_wrest/test/test_aat_wrest.py

@@ -52,6 +52,11 @@ args_carta = argparse.Namespace(
 )
 result_carta = '{"imageName": "VLASS1.1.ql.T29t04.J094850+743000.10.2048.v5"}'
 
+args_curator = argparse.Namespace(
+    curator="uid://evla/calibration/71d1f00c-a381-4688-9d5c-f528f43452f7", carta=None, stdcals=None, stdimg=None
+)
+result_curator = '{"filesList": ["weblog.tgz", "18B-265_2019_12_10_T01_24_58.690.tar", "ingestion_artifacts_2019_12_13_T23_31_42.382.tar"]}'
+
 
 def mock_wrester(args: argparse.Namespace) -> WrestWorkflowMetadata:
     with patch("psycopg2.connect") as mock_connect:
@@ -61,11 +66,14 @@ def mock_wrester(args: argparse.Namespace) -> WrestWorkflowMetadata:
             return WrestWorkflowMetadata(connection=mock_connect, sdm_id=args.stdimg)
         if args.carta is not None:
             return WrestWorkflowMetadata(connection=mock_connect, spl=args.carta)
+        if args.curator is not None:
+            return WrestWorkflowMetadata(connection=mock_connect, spl=args.curator)
 
 
 cal_wrester = mock_wrester(args_cal)
 img_wrester = mock_wrester(args_image)
 carta_wrester = mock_wrester(args_carta)
+curator_wrester = mock_wrester(args_curator)
 
 
 class TestAatWrest:
@@ -114,3 +122,17 @@ class TestAatWrest:
         assert args_carta.carta == "uid://evla/image/3d3db489-9331-4e61-aa80-002bc2989b1e"
         value = carta_wrester.wrest_image_name()
         assert value == '{"imageName": "VLASS1.1.ql.T29t04.J094850+743000.10.2048.v5"}'
+
+    @patch("json.dumps", MagicMock(return_value=result_curator))
+    def test_wrest_curator_products(self):
+        curator_wrester.conn.cursor.return_value.fetchall.return_value = [
+            "weblog.tgz",
+            "18B-265_2019_12_10_T01_24_58.690.tar",
+            "ingestion_artifacts_2019_12_13_T23_31_42.382.tar",
+        ]
+        assert args_curator.curator == "uid://evla/calibration/71d1f00c-a381-4688-9d5c-f528f43452f7"
+        value = curator_wrester.wrest_curator_products()
+        assert (
+            value
+            == '{"filesList": ["weblog.tgz", "18B-265_2019_12_10_T01_24_58.690.tar", "ingestion_artifacts_2019_12_13_T23_31_42.382.tar"]}'
+        )