[NDR-314] Change tests to reflect changes introduced by another ticket

Author: jameslinnell

lambdas/tests/unit/services/test_pdm_upload_document_reference_service.py

@@ -167,7 +167,7 @@ def test_fetch_preliminary_document_reference_success(
 
     assert result == mock_pdm_document_reference
     pdm_service.document_service.fetch_documents_from_table.assert_called_once_with(
-        table=MOCK_PDM_TABLE_NAME,
+        table_name=MOCK_PDM_TABLE_NAME,
         search_condition=document_key,
         search_key="ID",
         query_filter=PreliminaryStatus,
@@ -412,7 +412,7 @@ def test_update_dynamo_table_clean_scan_result(
 
     pdm_service.document_service.update_document.assert_called_once_with(
         table_name=MOCK_PDM_TABLE_NAME,
-        document_reference=mock_pdm_document_reference,
+        document=mock_pdm_document_reference,
         update_fields_name={
             "virus_scanner_result",
             "doc_status",