def test_config_local_override():
local_config = {
"app": {
"values": {
"a": 1
}
},
"level1": {
"level2": {
"level3": {
"level4": "new value"
}
}
},
}
with mock_config_file(mock_default_config, local_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
assert str(config) == "Config of fake-tool"
assert config.app.api_key == mock_default_config["app"]["api_key"]
assert config.level1.level2.level3.level4 == "new value"
def test_warn_on_extra_key():
os.environ["FAKE_TOOL_APP_SOME_UNKNOWN_KEY"] = "unknown"
with warnings.catch_warnings(record=True) as ws:
with mock_config_file(mock_default_config,
local_values=mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
assert any(issubclass(w.category, UserWarning) for w in ws)
warning_messages = [str(w.message) for w in ws]
assert (
"Environment variable [FAKE_TOOL_APP_SOME_UNKNOWN_KEY] does not match any "
"possible setting, ignoring." in warning_messages)
local_config = copy.deepcopy(mock_default_config)
local_config["app"]["some_unknown_key"] = "unknown"
with warnings.catch_warnings(record=True) as ws:
with mock_config_file(mock_default_config, local_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
assert any(issubclass(w.category, UserWarning) for w in ws)
warning_messages = [str(w.message) for w in ws]
assert any((
warning_message for warning_message in warning_messages if
"Override [app.some_unknown_key] not found in base config, ignoring."
in warning_message))
def test_raise_on_missing_default_file():
with mock_config_file({}):
with pytest.raises(FileNotFoundError):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
def test_config():
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
assert str(config) == "Config of fake-tool"
assert config.app.api_key == mock_default_config["app"]["api_key"]
def test_config_with_logging_config():
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=True,
reload_config=True)
assert logging.StreamHandler in [
type(handler) for handler in logging.getLogger().handlers
]
def test_config_env_var_override():
os.environ["FAKE_TOOL_APP_API_KEY"] = "do_not_override"
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
assert config.app.api_key == "do_not_override"
def test_json_format() -> None:
mock_conf = mock_default_config.copy()
mock_conf["logging"]["json_format"] = True
with mock_config_file(mock_conf):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=True,
reload_config=True)
root_logger = logging.getLogger()
assert isinstance(root_logger.handlers[0].formatter.processor,
structlog.processors.JSONRenderer)
def test_raise_on_all_missing_variables():
with mock_config_file({"foo": {"bar": None, "baz": None}}, {}):
with pytest.raises(CriticalSettingException) as exception_info:
setup_config("fake-tool",
"project.config",
critical_settings=True,
setup_logging=False,
reload_config=True)
assert all(x in exception_info.value.args[0] for x in [
"section 'foo' and setting 'bar'",
"section 'foo' and setting 'baz'"
])
def test_config_no_reload():
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
assert str(config) == "Config of fake-tool"
setup_config("fakez-toolz",
"project.config",
critical_settings=False,
setup_logging=False)
assert str(config) != "Config of fakez-toolz"
def test_snake_config_section_env_var_override():
os.environ["FAKE_TOOL_SNAKE_CASE_SECTION_VALUE"] = "new value"
os.environ["FAKE_TOOL_SNAKE_CASE_SECTION_SPLIT_VALUE"] = "new split value"
with mock_config_file(mock_default_config), warnings.catch_warnings(
record=True) as ws:
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
assert config.snake_case_section.value == "new value"
assert config.snake_case_section.split_value == "new split value"
for w in ws:
assert "FAKE_TOOL_SNAKE_CASE_SECTION_VALUE" not in str(w.message)
assert "FAKE_TOOL_SNAKE_CASE_SECTION_SPLIT_VALUE" not in str(w.message)
def test_load_logging_config_with_custom_processor():
def custom_processor_1(_, __, e):
return e
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
logger = load_logging_config(custom_processors=[custom_processor_1])
_handlers = [
handler for handler in logger.handlers
if type(handler) is logging.StreamHandler
]
stream_handler = _handlers[0]
assert custom_processor_1 in stream_handler.formatter.foreign_pre_chain
def test_load_logging_config():
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
logger = load_logging_config()
stream_handler = [
handler for handler in logger.handlers
if type(handler) is logging.StreamHandler
][0]
assert type(
stream_handler.formatter) is structlog.stdlib.ProcessorFormatter
assert type(stream_handler.formatter.processor
) is structlog.dev.ConsoleRenderer
assert logger.level == logging.DEBUG
def test_load_logging_config_with_custom_handler():
class CustomHandler(logging.Handler):
def __init__(self):
logging.Handler.__init__(self)
def emit(self, record):
print(record)
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
logger = load_logging_config(custom_handlers=[CustomHandler()])
assert len(logger.handlers) == 2
assert type(logger.handlers[0]) == logging.StreamHandler
assert type(logger.handlers[1]) == CustomHandler
def test_critical_settings():
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
local_config = copy.deepcopy(mock_default_config)
del local_config["app"]["no_default"]
with mock_config_file(mock_default_config, local_config):
with pytest.raises(CriticalSettingException):
setup_config("fake-tool",
"project.config",
critical_settings=True,
setup_logging=False,
reload_config=True)
def test_mock_config():
overwrite_api_key_value = "bin kin mon coquin toi"
def external_function():
return config.app.api_key
# Setup config
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=True,
reload_config=True)
# Verifying that the original value is in config.app.api_key
assert config.app.api_key == mock_default_config["app"]["api_key"]
# Verifying that mock_config will change the value of config.app.api_key
with mock_config({"app": {"api_key": overwrite_api_key_value}}):
assert external_function() == overwrite_api_key_value
def test_empty_string_setting():
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=False,
setup_logging=False,
reload_config=True)
assert config.app.no_default is None, "expected the setting to be None"
local_config = copy.deepcopy(mock_default_config)
local_config["app"]["no_default"] = ""
local_config["app"]["v"] = {"a": 1}
with mock_config_file(mock_default_config, local_config):
setup_config("fake-tool",
"project.config",
critical_settings=True,
setup_logging=False,
reload_config=True)
assert config.app.no_default == "", "expected the setting to be overwritten by local.yml to ''"
assert config.app.v == {
"a": 1
}, 'expected the setting to be overwritten by local.yml to "{"a": 1}"'
os.environ["FAKE_TOOL_APP_NO_DEFAULT"] = ""
os.environ["FAKE_TOOL_APP_V"] = '{"a":1}'
with mock_config_file(mock_default_config):
setup_config("fake-tool",
"project.config",
critical_settings=True,
setup_logging=False,
reload_config=True)
assert config.app.no_default == "", "expected the setting to be overwritten by env var to ''"
assert config.app.v == {
"a": 1
}, 'expected the setting to be overwritten by env var to "{"a": 1}"'
import typer
from opset import setup_config
from tbot import bot
setup_config("tbot", "tbot.config")
cli = typer.Typer()
@cli.command()
def main():
typer.secho("Starting Twitter bot", fg="blue")
bot.bot()
if __name__ == '__main__':
cli()
