diff --git a/backend/pashub_fetcher/tests/test_pashub_service.py b/backend/pashub_fetcher/tests/test_pashub_service.py index cf1c489a..1f750117 100644 --- a/backend/pashub_fetcher/tests/test_pashub_service.py +++ b/backend/pashub_fetcher/tests/test_pashub_service.py @@ -148,10 +148,11 @@ def test_run_persists_uploaded_file_records_to_db() -> None: service.run(make_request(uprn="12345")) fake_session.add_all.assert_called_once() - added: list = fake_session.add_all.call_args[0][0] + added: list[Any] = fake_session.add_all.call_args[0][0] assert len(added) == 1 assert added[0].s3_file_bucket == "test-bucket" assert added[0].uprn == 12345 + assert added[0].file_source == FileSourceEnum.PAS_HUB.value # ---------------------------------------------------------------------------