nextcloud-mcp-server/tests/unit/test_document_processor_config.py

"""Unit tests for document processor configuration."""

import os

import pytest

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."""
        from nextcloud_mcp_server.config import get_document_processor_config

        os.environ.pop("ENABLE_DOCUMENT_PROCESSING", None)
        config = get_document_processor_config()
        assert config["enabled"] is False

    def test_config_enabled(self):
        """Test enabling document processing."""
        from nextcloud_mcp_server.config import get_document_processor_config

        os.environ["ENABLE_DOCUMENT_PROCESSING"] = "true"
        try:
            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."""
        from nextcloud_mcp_server.config import get_document_processor_config

        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:
            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."""
        from nextcloud_mcp_server.config import get_document_processor_config

        os.environ["ENABLE_TESSERACT"] = "true"
        os.environ["TESSERACT_LANG"] = "eng+deu"
        os.environ["TESSERACT_CMD"] = "/usr/local/bin/tesseract"

        try:
            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."""
        from nextcloud_mcp_server.config import get_document_processor_config

        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:
            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."""
        from nextcloud_mcp_server.config import get_document_processor_config

        os.environ["ENABLE_DOCUMENT_PROCESSING"] = "true"
        os.environ["ENABLE_UNSTRUCTURED"] = "true"
        os.environ["ENABLE_TESSERACT"] = "true"

        try:
            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."""
        from nextcloud_mcp_server.config import get_document_processor_config

        os.environ.pop("DOCUMENT_PROCESSOR", None)
        config = get_document_processor_config()
        assert config["default_processor"] == "unstructured"

        os.environ["DOCUMENT_PROCESSOR"] = "tesseract"
        try:
            config = get_document_processor_config()
            assert config["default_processor"] == "tesseract"
        finally:
            os.environ.pop("DOCUMENT_PROCESSOR", None)