dify/api/tests/unit_tests/test_constants.py

import importlib

import pytest

import constants
from configs import dify_config
from tests.unit_tests.config_override import apply_config_overrides


@pytest.mark.parametrize("etl_type", ["SelfHosted", "Unstructured"])
def test_document_extensions_include_odt_for_document_etl_modes(monkeypatch: pytest.MonkeyPatch, etl_type: str) -> None:
    original_etl_type = dify_config.ETL_TYPE
    original_unstructured_api_url = dify_config.UNSTRUCTURED_API_URL

    try:
        apply_config_overrides(monkeypatch, ETL_TYPE=etl_type, UNSTRUCTURED_API_URL=None)

        reloaded_constants = importlib.reload(constants)

        assert "odt" in reloaded_constants.DOCUMENT_EXTENSIONS
        assert "ODT" in reloaded_constants.DOCUMENT_EXTENSIONS
    finally:
        apply_config_overrides(
            monkeypatch,
            ETL_TYPE=original_etl_type,
            UNSTRUCTURED_API_URL=original_unstructured_api_url,
        )
        importlib.reload(constants)