Use save_identifiers on document not identifiers.save()

nationalarchives · Dec 12, 2024 · 5f83679 · 5f83679
1 parent 6635cdb
commit 5f83679
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ds-caselaw-ingester/lambda_function.py b/ds-caselaw-ingester/lambda_function.py
@@ -456,7 +456,7 @@ def set_document_identifiers(self) -> None:
 
         if ncn:
             doc.identifiers.add(NeutralCitationNumber(ncn))
-            doc.identifiers.save(doc)
+            doc.save_identifiers(doc)
             logger.info(f"Ingested document had NCN {ncn}")
         else:
             logger.info(f"Ingested document had NCN (NOT FOUND)")

diff --git a/ds-caselaw-ingester/tests.py b/ds-caselaw-ingester/tests.py
@@ -215,7 +215,7 @@ def test_handler_messages_s3(
         )
         assert annotation.call_count == 2
         assert doc.identifiers.add.call_args_list[0].args[0].value == "[2012] UKUT 82 (IAC)"
-        doc.identifiers.save.assert_called()
+        doc.save_identifiers.assert_called()
 
 
 class TestLambda: