diff --git a/src/paperless/tests/test_remote_ocr_config.py b/src/paperless/tests/test_remote_ocr_config.py index b331d41a9..bc04a043d 100644 --- a/src/paperless/tests/test_remote_ocr_config.py +++ b/src/paperless/tests/test_remote_ocr_config.py @@ -8,6 +8,7 @@ import pytest from django.test import override_settings from paperless.config import RemoteOCRConfig +from paperless.models import RemoteOCRMode if TYPE_CHECKING: from unittest.mock import MagicMock @@ -20,6 +21,7 @@ def null_app_config(mocker) -> MagicMock: remote_ocr_engine=None, remote_ocr_api_key=None, remote_ocr_endpoint=None, + remote_ocr_mode=None, ) @@ -47,10 +49,12 @@ class TestRemoteOCRConfig: REMOTE_OCR_ENGINE="azureai", REMOTE_OCR_API_KEY="env-key", REMOTE_OCR_ENDPOINT="https://env.cognitiveservices.azure.com", + REMOTE_OCR_MODE=RemoteOCRMode.WORKFLOW_ONLY, ) assert cfg.remote_ocr_engine == "azureai" assert cfg.remote_ocr_api_key == "env-key" assert cfg.remote_ocr_endpoint == "https://env.cognitiveservices.azure.com" + assert cfg.remote_ocr_mode == RemoteOCRMode.WORKFLOW_ONLY def test_app_config_takes_precedence( self, @@ -61,16 +65,19 @@ class TestRemoteOCRConfig: remote_ocr_engine="azureai", remote_ocr_api_key="db-key", remote_ocr_endpoint="https://db.cognitiveservices.azure.com", + remote_ocr_mode=RemoteOCRMode.WORKFLOW_ONLY, ) cfg = make_remote_ocr_config( app_config, REMOTE_OCR_ENGINE=None, REMOTE_OCR_API_KEY="env-key", REMOTE_OCR_ENDPOINT="https://env.cognitiveservices.azure.com", + REMOTE_OCR_MODE=RemoteOCRMode.ALWAYS, ) assert cfg.remote_ocr_engine == "azureai" assert cfg.remote_ocr_api_key == "db-key" assert cfg.remote_ocr_endpoint == "https://db.cognitiveservices.azure.com" + assert cfg.remote_ocr_mode == RemoteOCRMode.WORKFLOW_ONLY def test_unset_everywhere( self,