Skip to content

Commit 935901f

Browse files
committed
[NDR-359] Correct DB name
1 parent 25c40ff commit 935901f

File tree

2 files changed

+25
-1
lines changed

2 files changed

+25
-1
lines changed

lambdas/services/document_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def delete_document_references(
186186
include={"doc_status", "deleted", "ttl"},
187187
)
188188
key_pair = {DocumentReferenceMetadataFields.ID.value: reference.id}
189-
if "CoreDocumentMetadata" in table_name:
189+
if "COREDocumentMetadata" in table_name:
190190
key_pair[DocumentReferenceMetadataFields.NHS_NUMBER.value] = (
191191
reference.nhs_number
192192
)

lambdas/tests/unit/services/test_document_service.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,30 @@ def test_delete_documents_soft_delete(mock_service, mock_dynamo_service):
203203
)
204204

205205

206+
@freeze_time("2023-10-1 13:00:00")
207+
def test_delete_documents_soft_delete_core(mock_service, mock_dynamo_service):
208+
test_doc_ref = DocumentReference.model_validate(MOCK_DOCUMENT)
209+
210+
test_date = datetime.now()
211+
ttl_date = test_date + timedelta(days=float(DocumentRetentionDays.SOFT_DELETE))
212+
213+
test_update_fields = {
214+
"Deleted": datetime.now().strftime("%Y-%m-%dT%H:%M:%S.%fZ"),
215+
"TTL": int(ttl_date.timestamp()),
216+
"DocStatus": "deprecated",
217+
}
218+
219+
mock_service.delete_document_references(
220+
"dev_COREDocument_metadata", [test_doc_ref], DocumentRetentionDays.SOFT_DELETE
221+
)
222+
223+
mock_dynamo_service.update_item.assert_called_once_with(
224+
table_name="dev_COREDocument_metadata",
225+
key_pair={"ID": test_doc_ref.id, "NhsNumber": test_doc_ref.nhs_number},
226+
updated_fields=test_update_fields,
227+
)
228+
229+
206230
@freeze_time("2023-10-1 13:00:00")
207231
def test_delete_documents_death_delete(mock_service, mock_dynamo_service):
208232
test_doc_ref = DocumentReference.model_validate(MOCK_DOCUMENT)

0 commit comments

Comments
 (0)