173 lines
4.9 KiB
Python
173 lines
4.9 KiB
Python
from os import environ
|
|
from pathlib import Path
|
|
from typing import List
|
|
from unittest import mock
|
|
|
|
from pydantic import AnyHttpUrl, BaseModel
|
|
from pytest import mark, raises
|
|
|
|
from libretime_shared.config import (
|
|
BaseConfig,
|
|
DatabaseConfig,
|
|
RabbitMQConfig,
|
|
no_trailing_slash_validator,
|
|
)
|
|
|
|
|
|
# pylint: disable=too-few-public-methods
|
|
class FixtureConfig(BaseConfig):
|
|
public_url: AnyHttpUrl
|
|
api_key: str
|
|
allowed_hosts: List[str] = []
|
|
database: DatabaseConfig
|
|
rabbitmq: RabbitMQConfig = RabbitMQConfig()
|
|
|
|
# Validators
|
|
_public_url_no_trailing_slash = no_trailing_slash_validator("public_url")
|
|
|
|
|
|
FIXTURE_CONFIG_RAW = """
|
|
public_url: http://libretime.example.com/
|
|
api_key: "f3bf04fc"
|
|
allowed_hosts:
|
|
- example.com
|
|
- sub.example.com
|
|
|
|
# Comment !
|
|
database:
|
|
host: "localhost"
|
|
port: 5432
|
|
|
|
ignored: "ignored"
|
|
"""
|
|
|
|
|
|
def test_base_config(tmp_path: Path):
|
|
config_filepath = tmp_path / "config.yml"
|
|
config_filepath.write_text(FIXTURE_CONFIG_RAW)
|
|
|
|
with mock.patch.dict(
|
|
environ,
|
|
{
|
|
"LIBRETIME_API": "invalid",
|
|
"LIBRETIME_DATABASE_PORT": "8888",
|
|
"LIBRETIME_DATABASE": "invalid",
|
|
"LIBRETIME_RABBITMQ": "invalid",
|
|
"LIBRETIME_RABBITMQ_HOST": "changed",
|
|
"WRONGPREFIX_API_KEY": "invalid",
|
|
},
|
|
):
|
|
config = FixtureConfig(filepath=config_filepath)
|
|
|
|
assert config.public_url == "http://libretime.example.com"
|
|
assert config.api_key == "f3bf04fc"
|
|
assert config.allowed_hosts == ["example.com", "sub.example.com"]
|
|
assert config.database.host == "localhost"
|
|
assert config.database.port == 8888
|
|
assert config.rabbitmq.host == "changed"
|
|
assert config.rabbitmq.port == 5672
|
|
|
|
# Optional model: loading default values (rabbitmq)
|
|
with mock.patch.dict(environ, {}):
|
|
config = FixtureConfig(filepath=config_filepath)
|
|
assert config.allowed_hosts == ["example.com", "sub.example.com"]
|
|
assert config.rabbitmq.host == "localhost"
|
|
assert config.rabbitmq.port == 5672
|
|
|
|
# Optional model: overriding using environment (rabbitmq)
|
|
with mock.patch.dict(
|
|
environ,
|
|
{
|
|
"LIBRETIME_RABBITMQ_HOST": "changed",
|
|
"LIBRETIME_ALLOWED_HOSTS": "example.com, changed.example.com",
|
|
},
|
|
):
|
|
config = FixtureConfig(filepath=config_filepath)
|
|
assert config.allowed_hosts == ["example.com", "changed.example.com"]
|
|
assert config.rabbitmq.host == "changed"
|
|
assert config.rabbitmq.port == 5672
|
|
|
|
|
|
# pylint: disable=too-few-public-methods
|
|
class RequiredModel(BaseModel):
|
|
api_key: str
|
|
with_default: str = "original"
|
|
|
|
|
|
# pylint: disable=too-few-public-methods
|
|
class FixtureWithRequiredSubmodelConfig(BaseConfig):
|
|
required: RequiredModel
|
|
|
|
|
|
FIXTURE_WITH_REQUIRED_SUBMODEL_CONFIG_RAW = """
|
|
required:
|
|
api_key: "test_key"
|
|
"""
|
|
|
|
|
|
def test_base_config_required_submodel(tmp_path: Path):
|
|
config_filepath = tmp_path / "config.yml"
|
|
config_filepath.write_text(FIXTURE_WITH_REQUIRED_SUBMODEL_CONFIG_RAW)
|
|
|
|
# With config file
|
|
with mock.patch.dict(environ, {}):
|
|
config = FixtureWithRequiredSubmodelConfig(filepath=config_filepath)
|
|
assert config.required.api_key == "test_key"
|
|
assert config.required.with_default == "original"
|
|
|
|
# With env variables
|
|
with mock.patch.dict(environ, {"LIBRETIME_REQUIRED_API_KEY": "test_key"}):
|
|
config = FixtureWithRequiredSubmodelConfig(filepath=None)
|
|
assert config.required.api_key == "test_key"
|
|
assert config.required.with_default == "original"
|
|
|
|
# With env variables override
|
|
with mock.patch.dict(environ, {"LIBRETIME_REQUIRED_API_KEY": "changed"}):
|
|
config = FixtureWithRequiredSubmodelConfig(filepath=config_filepath)
|
|
assert config.required.api_key == "changed"
|
|
assert config.required.with_default == "original"
|
|
|
|
# With env variables default override
|
|
with mock.patch.dict(
|
|
environ,
|
|
{
|
|
"LIBRETIME_REQUIRED_API_KEY": "changed",
|
|
"LIBRETIME_REQUIRED_WITH_DEFAULT": "changed",
|
|
},
|
|
):
|
|
config = FixtureWithRequiredSubmodelConfig(filepath=config_filepath)
|
|
assert config.required.api_key == "changed"
|
|
assert config.required.with_default == "changed"
|
|
|
|
# Raise validation error
|
|
with mock.patch.dict(environ, {}):
|
|
with raises(SystemExit):
|
|
FixtureWithRequiredSubmodelConfig(filepath=None)
|
|
|
|
|
|
FIXTURE_CONFIG_RAW_MISSING = """
|
|
database:
|
|
host: "localhost"
|
|
"""
|
|
|
|
FIXTURE_CONFIG_RAW_INVALID = """
|
|
database
|
|
host: "localhost"
|
|
"""
|
|
|
|
|
|
@mark.parametrize(
|
|
"raw,exception",
|
|
[
|
|
(FIXTURE_CONFIG_RAW_INVALID, SystemExit),
|
|
(FIXTURE_CONFIG_RAW_MISSING, SystemExit),
|
|
],
|
|
)
|
|
def test_load_config_error(tmp_path: Path, raw, exception):
|
|
config_filepath = tmp_path / "config.yml"
|
|
config_filepath.write_text(raw)
|
|
|
|
with raises(exception):
|
|
with mock.patch.dict(environ, {}):
|
|
FixtureConfig(filepath=config_filepath)
|