keyosk/tests/test_config.py

59 lines
1.7 KiB
Python

import os
from pathlib import Path
import toml
from keyosk import config
from keyosk import constants
from keyosk import datatypes
DEMO_CONFIG = {
"storage": {
"backend": "maria",
"sqlite": {
"path": "/foo/bar/baz.db",
"pragmas": {"foo": 1, "bar": "buzz", "baz": ["dog", "cat"]},
},
"maria": {
"schema": "authentifiy",
"host": "10.20.30.40",
"port": 6033,
"username": "qwerty",
"password": "uiop",
},
}
}
def test_default():
assert config.load() == config.KeyoskConfig()
def test_roundtrip():
serializer = config.ConfigSerializer()
loaded = serializer.load(DEMO_CONFIG)
assert DEMO_CONFIG == serializer.dump(loaded)
assert loaded == serializer.load(serializer.dump(loaded))
def test_settings():
loaded = config.ConfigSerializer().load(DEMO_CONFIG)
assert loaded.storage.backend == datatypes.StorageBackend.MARIA
assert loaded.storage.sqlite.path == Path(DEMO_CONFIG["storage"]["sqlite"]["path"])
assert loaded.storage.sqlite.pragmas == DEMO_CONFIG["storage"]["sqlite"]["pragmas"]
for key, value in DEMO_CONFIG["storage"]["maria"].items():
assert getattr(loaded.storage.maria, key) == value
def test_filepath(tmp_path):
tmp_file = Path(tmp_path, "conf.toml")
with tmp_file.open("w+") as outfile:
toml.dump(DEMO_CONFIG, outfile)
assert config.load(tmp_file) == config.ConfigSerializer().load(DEMO_CONFIG)
os.environ[constants.ENV_CONFIG_PATH] = str(tmp_file)
assert config.load() == config.ConfigSerializer().load(DEMO_CONFIG)
tmp_file.unlink()
assert config.load(tmp_file) == config.KeyoskConfig()