mirror of
https://github.com/checktheroads/hyperglass
synced 2024-05-11 05:55:08 +00:00
59 lines
1.3 KiB
Python
59 lines
1.3 KiB
Python
"""Test configuration file collection."""
|
|
|
|
# Standard Library
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
# Project
|
|
from hyperglass.settings import Settings
|
|
|
|
# Local
|
|
from ..collect import load_config
|
|
|
|
TOML = """
|
|
test = "from toml"
|
|
"""
|
|
|
|
YAML = """
|
|
test: from yaml
|
|
"""
|
|
|
|
JSON = """
|
|
{"test": "from json"}
|
|
"""
|
|
|
|
PY_VARIABLE = """
|
|
MAIN = {'test': 'from python variable'}
|
|
"""
|
|
|
|
PY_FUNCTION = """
|
|
def main():
|
|
return {'test': 'from python function'}
|
|
"""
|
|
|
|
PY_COROUTINE = """
|
|
async def main():
|
|
return {'test': 'from python coroutine'}
|
|
"""
|
|
|
|
CASES = (
|
|
("test.toml", "from toml", TOML),
|
|
("test.yaml", "from yaml", YAML),
|
|
("test_py_variable.py", "from python variable", PY_VARIABLE),
|
|
("test_py_function.py", "from python function", PY_FUNCTION),
|
|
("test_py_coroutine.py", "from python coroutine", PY_COROUTINE),
|
|
)
|
|
|
|
|
|
def test_collect(monkeypatch):
|
|
with tempfile.TemporaryDirectory() as directory_name:
|
|
directory = Path(directory_name)
|
|
monkeypatch.setattr(Settings, "app_path", directory)
|
|
for name, value, data in CASES:
|
|
path = directory / Path(name)
|
|
with path.open("w") as p:
|
|
p.write(data)
|
|
loaded = load_config(path.stem, required=True)
|
|
assert loaded.get("test") is not None
|
|
assert loaded["test"] == value
|