mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-06-04 11:14:10 +08:00
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
from textwrap import dedent
|
|
|
|
import pytest
|
|
from flask import Flask
|
|
|
|
from configs.app_config import DifyConfig
|
|
|
|
EXAMPLE_ENV_FILENAME = '.env'
|
|
|
|
|
|
@pytest.fixture
|
|
def example_env_file(tmp_path, monkeypatch) -> str:
|
|
monkeypatch.chdir(tmp_path)
|
|
file_path = tmp_path.joinpath(EXAMPLE_ENV_FILENAME)
|
|
file_path.write_text(dedent(
|
|
"""
|
|
CONSOLE_API_URL=https://example.com
|
|
"""))
|
|
return str(file_path)
|
|
|
|
|
|
def test_dify_config_undefined_entry(example_env_file):
|
|
# load dotenv file with pydantic-settings
|
|
config = DifyConfig(_env_file=example_env_file)
|
|
|
|
# entries not defined in app settings
|
|
with pytest.raises(TypeError):
|
|
# TypeError: 'AppSettings' object is not subscriptable
|
|
assert config['LOG_LEVEL'] == 'INFO'
|
|
|
|
|
|
def test_dify_config(example_env_file):
|
|
# load dotenv file with pydantic-settings
|
|
config = DifyConfig(_env_file=example_env_file)
|
|
|
|
# constant values
|
|
assert config.COMMIT_SHA == ''
|
|
|
|
# default values
|
|
assert config.EDITION == 'SELF_HOSTED'
|
|
assert config.API_COMPRESSION_ENABLED is False
|
|
assert config.SENTRY_TRACES_SAMPLE_RATE == 1.0
|
|
|
|
|
|
def test_flask_configs(example_env_file):
|
|
flask_app = Flask('app')
|
|
flask_app.config.from_mapping(DifyConfig(_env_file=example_env_file).model_dump())
|
|
config = flask_app.config
|
|
|
|
# configs read from dotenv directly
|
|
assert config['LOG_LEVEL'] == 'INFO'
|
|
|
|
# configs read from pydantic-settings
|
|
assert config['COMMIT_SHA'] == ''
|
|
assert config['EDITION'] == 'SELF_HOSTED'
|
|
assert config['API_COMPRESSION_ENABLED'] is False
|
|
assert config['SENTRY_TRACES_SAMPLE_RATE'] == 1.0
|
|
|
|
# value from env file
|
|
assert config['CONSOLE_API_URL'] == 'https://example.com'
|
|
# fallback to alias choices value as CONSOLE_API_URL
|
|
assert config['FILES_URL'] == 'https://example.com'
|
|
|
|
assert config['SQLALCHEMY_DATABASE_URI'] == 'postgresql://postgres:@localhost:5432/dify'
|
|
assert config['SQLALCHEMY_ENGINE_OPTIONS'] == {
|
|
'connect_args': {
|
|
'options': '-c timezone=UTC',
|
|
},
|
|
'max_overflow': 10,
|
|
'pool_pre_ping': False,
|
|
'pool_recycle': 3600,
|
|
'pool_size': 30,
|
|
}
|