From d8ec12065f784f013825c5e2d0fbb91e215ed314 Mon Sep 17 00:00:00 2001 From: Daniel Roth Date: Tue, 2 Jun 2026 09:42:41 +0000 Subject: [PATCH] =?UTF-8?q?run()=20returns=20core=20and=20other=20file=20p?= =?UTF-8?q?aths=20=F0=9F=9F=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../tests/test_pashub_service.py | 24 +++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/backend/pashub_fetcher/tests/test_pashub_service.py b/backend/pashub_fetcher/tests/test_pashub_service.py index 6cf889a2..df77287d 100644 --- a/backend/pashub_fetcher/tests/test_pashub_service.py +++ b/backend/pashub_fetcher/tests/test_pashub_service.py @@ -73,6 +73,30 @@ def test_run_returns_file_paths() -> None: assert result == ["/tmp/a.pdf", "/tmp/b.pdf"] +# --------------------------------------------------------------------------- +# run(): returns core + other file paths when get_other_files=True +# --------------------------------------------------------------------------- + + +def test_run_returns_core_and_other_file_paths() -> None: + # Arrange + mock_client = MagicMock(spec=PashubClient) + mock_client.get_uprn_by_job_id.return_value = None + mock_client.get_evidence_files_by_job_id.return_value = make_downloaded( + core=["/tmp/core.pdf"], + other=["/tmp/other.pdf"], + ) + + service = make_service(pashub_client=mock_client) + + # Act + with patch("backend.pashub_fetcher.pashub_service.os.remove"): + result = service.run(make_request(get_other_files=True)) + + # Assert + assert result == ["/tmp/core.pdf", "/tmp/other.pdf"] + + # --------------------------------------------------------------------------- # run(): skips upload when neither uprn nor hubspot_deal_id # ---------------------------------------------------------------------------