More tests for remote_ocr_mode

This commit is contained in:
shamoon
2026-08-10 10:13:18 -07:00
parent a5d46a883e
commit cc8fee91c4
@@ -8,6 +8,7 @@ import pytest
from django.test import override_settings from django.test import override_settings
from paperless.config import RemoteOCRConfig from paperless.config import RemoteOCRConfig
from paperless.models import RemoteOCRMode
if TYPE_CHECKING: if TYPE_CHECKING:
from unittest.mock import MagicMock from unittest.mock import MagicMock
@@ -20,6 +21,7 @@ def null_app_config(mocker) -> MagicMock:
remote_ocr_engine=None, remote_ocr_engine=None,
remote_ocr_api_key=None, remote_ocr_api_key=None,
remote_ocr_endpoint=None, remote_ocr_endpoint=None,
remote_ocr_mode=None,
) )
@@ -47,10 +49,12 @@ class TestRemoteOCRConfig:
REMOTE_OCR_ENGINE="azureai", REMOTE_OCR_ENGINE="azureai",
REMOTE_OCR_API_KEY="env-key", REMOTE_OCR_API_KEY="env-key",
REMOTE_OCR_ENDPOINT="https://env.cognitiveservices.azure.com", REMOTE_OCR_ENDPOINT="https://env.cognitiveservices.azure.com",
REMOTE_OCR_MODE=RemoteOCRMode.WORKFLOW_ONLY,
) )
assert cfg.remote_ocr_engine == "azureai" assert cfg.remote_ocr_engine == "azureai"
assert cfg.remote_ocr_api_key == "env-key" assert cfg.remote_ocr_api_key == "env-key"
assert cfg.remote_ocr_endpoint == "https://env.cognitiveservices.azure.com" assert cfg.remote_ocr_endpoint == "https://env.cognitiveservices.azure.com"
assert cfg.remote_ocr_mode == RemoteOCRMode.WORKFLOW_ONLY
def test_app_config_takes_precedence( def test_app_config_takes_precedence(
self, self,
@@ -61,16 +65,19 @@ class TestRemoteOCRConfig:
remote_ocr_engine="azureai", remote_ocr_engine="azureai",
remote_ocr_api_key="db-key", remote_ocr_api_key="db-key",
remote_ocr_endpoint="https://db.cognitiveservices.azure.com", remote_ocr_endpoint="https://db.cognitiveservices.azure.com",
remote_ocr_mode=RemoteOCRMode.WORKFLOW_ONLY,
) )
cfg = make_remote_ocr_config( cfg = make_remote_ocr_config(
app_config, app_config,
REMOTE_OCR_ENGINE=None, REMOTE_OCR_ENGINE=None,
REMOTE_OCR_API_KEY="env-key", REMOTE_OCR_API_KEY="env-key",
REMOTE_OCR_ENDPOINT="https://env.cognitiveservices.azure.com", REMOTE_OCR_ENDPOINT="https://env.cognitiveservices.azure.com",
REMOTE_OCR_MODE=RemoteOCRMode.ALWAYS,
) )
assert cfg.remote_ocr_engine == "azureai" assert cfg.remote_ocr_engine == "azureai"
assert cfg.remote_ocr_api_key == "db-key" assert cfg.remote_ocr_api_key == "db-key"
assert cfg.remote_ocr_endpoint == "https://db.cognitiveservices.azure.com" assert cfg.remote_ocr_endpoint == "https://db.cognitiveservices.azure.com"
assert cfg.remote_ocr_mode == RemoteOCRMode.WORKFLOW_ONLY
def test_unset_everywhere( def test_unset_everywhere(
self, self,