Adjust return value for new changes to main

sul-dlss · May 16, 2024 · 9f39576 · 9f39576
1 parent fcae103
commit 9f39576
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/libsys_airflow/plugins/data_exports/transmission_tasks.py b/libsys_airflow/plugins/data_exports/transmission_tasks.py
@@ -140,4 +140,4 @@ def is_production():
 
 
 def return_success_test_instance(files) -> dict:
-    return {"success": files, "failures": []}
+    return {"success": files["file_list"], "failures": []}
diff --git a/tests/data_exports/test_transmission_tasks.py b/tests/data_exports/test_transmission_tasks.py
@@ -114,12 +114,13 @@ def test_transmit_data_ftp_task(
         return_value=mock_ftphook_connection,
     )
 
-    transmit_data = transmit_data_ftp_task.function("ftp-example.com", mock_marc_files)
-    assert len(transmit_data["success"]) == 3
-    assert "Start transmission of file" in caplog.text
-    assert ftp_hook.store_file.called_with(
-        "/remote/path/dir/2024022914.mrc", "2024022914.mrc"
-    )
+    with mock.patch.dict("os.environ", AIRFLOW_VAR_OKAPI_URL="http://okapi-prod"):
+        transmit_data = transmit_data_ftp_task.function("ftp-example.com", mock_marc_files)
+        assert len(transmit_data["success"]) == 3
+        assert "Start transmission of file" in caplog.text
+        assert ftp_hook.store_file.called_with(
+            "/remote/path/dir/2024022914.mrc", "2024022914.mrc"
+        )
 
 
 def test_transmit_data_task(