Backend for manual tempalte choice

src/documents/tests/test_bulk_edit.py

@@ -21,6 +21,7 @@ from documents.models import CustomField
 from documents.models import CustomFieldInstance
 from documents.models import Document
 from documents.models import DocumentType
+from documents.models import OcrTemplate
 from documents.models import StoragePath
 from documents.models import Tag
 from documents.permissions import set_permissions_for_objects
@@ -1999,9 +2000,31 @@ class TestBulkEditRunZoneOcr:
 
         assert result == "OK"
         assert [
-            call.kwargs["kwargs"]["document_id"]
-            for call in mock_task.apply_async.call_args_list
-        ] == [doc.id for doc in docs]
+            call.kwargs["kwargs"] for call in mock_task.apply_async.call_args_list
+        ] == [{"document_id": doc.id, "template_id": None} for doc in docs]
+
+    def test_queues_the_chosen_template(self, mock_task: mock.MagicMock) -> None:
+        """
+        GIVEN:
+            - A chosen OCR template
+        WHEN:
+            - run_zone_ocr is called with it
+        THEN:
+            - The template is queued with the document
+        """
+        doc = DocumentFactory()
+        template = OcrTemplate.objects.create(
+            name="Bills",
+            source_width=100,
+            source_height=100,
+        )
+
+        bulk_edit.run_zone_ocr([doc.id], template=template)
+
+        assert mock_task.apply_async.call_args.kwargs["kwargs"] == {
+            "document_id": doc.id,
+            "template_id": template.pk,
+        }
 
     def test_task_runs_extraction(self, mocker: MockerFixture) -> None:
         """
@@ -2019,6 +2042,43 @@ class TestBulkEditRunZoneOcr:
 
         extract.assert_called_once_with(doc, None)
 
+    def test_task_runs_the_chosen_template(self, mocker: MockerFixture) -> None:
+        """
+        GIVEN:
+            - A queued zone OCR task with a chosen template
+        WHEN:
+            - The task runs
+        THEN:
+            - Zone extraction runs with only that template
+        """
+        extract = mocker.patch("documents.tasks.run_zone_extraction")
+        doc = DocumentFactory()
+        template = OcrTemplate.objects.create(
+            name="Bills",
+            source_width=100,
+            source_height=100,
+        )
+
+        extract_zones(doc.id, template.pk)
+
+        extract.assert_called_once_with(doc, None, templates=[template])
+
+    def test_task_skips_a_deleted_template(self, mocker: MockerFixture) -> None:
+        """
+        GIVEN:
+            - A template deleted between queueing and running
+        WHEN:
+            - The task runs
+        THEN:
+            - Nothing is extracted
+        """
+        extract = mocker.patch("documents.tasks.run_zone_extraction")
+        doc = DocumentFactory()
+
+        extract_zones(doc.id, 999)
+
+        extract.assert_not_called()
+
 
 @pytest.mark.django_db
 class TestBulkEditReprocess: