From fbcaaf2e9b419df94a169273a4ca59e0b2425bc6 Mon Sep 17 00:00:00 2001 From: shamoon <4887959+shamoon@users.noreply.github.com> Date: Thu, 29 Jan 2026 11:45:14 -0800 Subject: [PATCH] Also in the tests --- src/documents/tests/utils.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/documents/tests/utils.py b/src/documents/tests/utils.py index 88dddc557..4a7f1db6d 100644 --- a/src/documents/tests/utils.py +++ b/src/documents/tests/utils.py @@ -20,6 +20,7 @@ from django.db.migrations.executor import MigrationExecutor from django.test import TransactionTestCase from django.test import override_settings +from documents.consumer import AsnCheckPlugin from documents.consumer import ConsumerPlugin from documents.consumer import ConsumerPreflightPlugin from documents.data_models import ConsumableDocument @@ -371,6 +372,14 @@ class GetConsumerMixin: "task-id", ) preflight_plugin.setup() + asncheck_plugin = AsnCheckPlugin( + doc, + overrides or DocumentMetadataOverrides(), + self.status, # type: ignore + self.dirs.scratch_dir, + "task-id", + ) + asncheck_plugin.setup() reader = ConsumerPlugin( doc, overrides or DocumentMetadataOverrides(), @@ -381,6 +390,7 @@ class GetConsumerMixin: reader.setup() try: preflight_plugin.run() + asncheck_plugin.run() yield reader finally: reader.cleanup()