Chris CoutinhoandClaude Opus 4.6
c8e4cbe825
Replace ~80 manual os.getenv() calls in config.py with dynaconf-backed configuration, enabling TOML file-based config alongside existing env var support. Zero breaking changes — Settings dataclass interface and all consumers unchanged. Phase 1: Create settings.toml with all config keys and defaults, .secrets.toml.example template, update .gitignore, initialize Dynaconf instance with envvar_prefix=False and environment section switching. Phase 2: Wire adapter — replace os.getenv() with _dynaconf.get() in get_settings(), get_document_processor_config(), and deprecation/ dependency resolution helpers. Automatic type coercion eliminates ~30 manual int()/float()/.lower()=="true" patterns. Phase 3: Add 12 declarative validators for port ranges, positive integers, enum constraints, and float ranges. Remove redundant negative overlap check from Settings.__post_init__. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
133 lines
5.4 KiB
Python
133 lines
5.4 KiB
Python
"""Unit tests for document processor configuration."""
|
|
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from nextcloud_mcp_server.config import _reload_config, get_document_processor_config
|
|
|
|
pytestmark = pytest.mark.unit
|
|
|
|
|
|
class TestDocumentProcessorConfig:
|
|
"""Test document processor configuration system."""
|
|
|
|
def test_config_disabled_by_default(self):
|
|
"""Test that document processing is disabled by default."""
|
|
os.environ.pop("ENABLE_DOCUMENT_PROCESSING", None)
|
|
_reload_config()
|
|
config = get_document_processor_config()
|
|
assert config["enabled"] is False
|
|
|
|
def test_config_enabled(self):
|
|
"""Test enabling document processing."""
|
|
os.environ["ENABLE_DOCUMENT_PROCESSING"] = "true"
|
|
try:
|
|
_reload_config()
|
|
config = get_document_processor_config()
|
|
assert config["enabled"] is True
|
|
finally:
|
|
os.environ.pop("ENABLE_DOCUMENT_PROCESSING", None)
|
|
|
|
def test_unstructured_processor_config(self):
|
|
"""Test Unstructured processor configuration."""
|
|
os.environ["ENABLE_UNSTRUCTURED"] = "true"
|
|
os.environ["UNSTRUCTURED_API_URL"] = "http://test:8000"
|
|
os.environ["UNSTRUCTURED_STRATEGY"] = "hi_res"
|
|
os.environ["UNSTRUCTURED_LANGUAGES"] = "eng,fra"
|
|
os.environ["UNSTRUCTURED_TIMEOUT"] = "60"
|
|
|
|
try:
|
|
_reload_config()
|
|
config = get_document_processor_config()
|
|
assert "unstructured" in config["processors"]
|
|
unst_config = config["processors"]["unstructured"]
|
|
assert unst_config["api_url"] == "http://test:8000"
|
|
assert unst_config["strategy"] == "hi_res"
|
|
assert unst_config["languages"] == ["eng", "fra"]
|
|
assert unst_config["timeout"] == 60
|
|
finally:
|
|
os.environ.pop("ENABLE_UNSTRUCTURED", None)
|
|
os.environ.pop("UNSTRUCTURED_API_URL", None)
|
|
os.environ.pop("UNSTRUCTURED_STRATEGY", None)
|
|
os.environ.pop("UNSTRUCTURED_LANGUAGES", None)
|
|
os.environ.pop("UNSTRUCTURED_TIMEOUT", None)
|
|
|
|
def test_tesseract_processor_config(self):
|
|
"""Test Tesseract processor configuration."""
|
|
os.environ["ENABLE_TESSERACT"] = "true"
|
|
os.environ["TESSERACT_LANG"] = "eng+deu"
|
|
os.environ["TESSERACT_CMD"] = "/usr/local/bin/tesseract"
|
|
|
|
try:
|
|
_reload_config()
|
|
config = get_document_processor_config()
|
|
assert "tesseract" in config["processors"]
|
|
tess_config = config["processors"]["tesseract"]
|
|
assert tess_config["lang"] == "eng+deu"
|
|
assert tess_config["tesseract_cmd"] == "/usr/local/bin/tesseract"
|
|
finally:
|
|
os.environ.pop("ENABLE_TESSERACT", None)
|
|
os.environ.pop("TESSERACT_LANG", None)
|
|
os.environ.pop("TESSERACT_CMD", None)
|
|
|
|
def test_custom_processor_config(self):
|
|
"""Test custom processor configuration."""
|
|
os.environ["ENABLE_CUSTOM_PROCESSOR"] = "true"
|
|
os.environ["CUSTOM_PROCESSOR_NAME"] = "my_ocr"
|
|
os.environ["CUSTOM_PROCESSOR_URL"] = "http://localhost:9000/process"
|
|
os.environ["CUSTOM_PROCESSOR_API_KEY"] = "secret"
|
|
os.environ["CUSTOM_PROCESSOR_TIMEOUT"] = "30"
|
|
os.environ["CUSTOM_PROCESSOR_TYPES"] = "application/pdf,image/jpeg"
|
|
|
|
try:
|
|
_reload_config()
|
|
config = get_document_processor_config()
|
|
assert "custom" in config["processors"]
|
|
custom_config = config["processors"]["custom"]
|
|
assert custom_config["name"] == "my_ocr"
|
|
assert custom_config["api_url"] == "http://localhost:9000/process"
|
|
assert custom_config["api_key"] == "secret"
|
|
assert custom_config["timeout"] == 30
|
|
assert "application/pdf" in custom_config["supported_types"]
|
|
assert "image/jpeg" in custom_config["supported_types"]
|
|
finally:
|
|
os.environ.pop("ENABLE_CUSTOM_PROCESSOR", None)
|
|
os.environ.pop("CUSTOM_PROCESSOR_NAME", None)
|
|
os.environ.pop("CUSTOM_PROCESSOR_URL", None)
|
|
os.environ.pop("CUSTOM_PROCESSOR_API_KEY", None)
|
|
os.environ.pop("CUSTOM_PROCESSOR_TIMEOUT", None)
|
|
os.environ.pop("CUSTOM_PROCESSOR_TYPES", None)
|
|
|
|
def test_multiple_processors(self):
|
|
"""Test configuration with multiple processors enabled."""
|
|
os.environ["ENABLE_DOCUMENT_PROCESSING"] = "true"
|
|
os.environ["ENABLE_UNSTRUCTURED"] = "true"
|
|
os.environ["ENABLE_TESSERACT"] = "true"
|
|
|
|
try:
|
|
_reload_config()
|
|
config = get_document_processor_config()
|
|
assert config["enabled"] is True
|
|
assert "unstructured" in config["processors"]
|
|
assert "tesseract" in config["processors"]
|
|
finally:
|
|
os.environ.pop("ENABLE_DOCUMENT_PROCESSING", None)
|
|
os.environ.pop("ENABLE_UNSTRUCTURED", None)
|
|
os.environ.pop("ENABLE_TESSERACT", None)
|
|
|
|
def test_default_processor_selection(self):
|
|
"""Test default processor configuration."""
|
|
os.environ.pop("DOCUMENT_PROCESSOR", None)
|
|
_reload_config()
|
|
config = get_document_processor_config()
|
|
assert config["default_processor"] == "unstructured"
|
|
|
|
os.environ["DOCUMENT_PROCESSOR"] = "tesseract"
|
|
try:
|
|
_reload_config()
|
|
config = get_document_processor_config()
|
|
assert config["default_processor"] == "tesseract"
|
|
finally:
|
|
os.environ.pop("DOCUMENT_PROCESSOR", None)
|