mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2026-08-14 14:53:18 +00:00
More tests for remote_ocr_mode
This commit is contained in:
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user