78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
import shutil
|
|
import tempfile
|
|
from pathlib import Path
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
import yaml
|
|
|
|
from halig.config import get_config, Config, EncryptionKeysConfig
|
|
from halig.exceptions import ConfigFileDoesNotExist, ConfigFileIsInvalid
|
|
|
|
from ward import fixture, test, raises
|
|
|
|
|
|
@fixture()
|
|
def tmpfile():
|
|
_tmpfile = NamedTemporaryFile(delete=False)
|
|
with open(_tmpfile.name, "w") as file:
|
|
yield file
|
|
Path(_tmpfile.name).unlink()
|
|
|
|
|
|
@fixture()
|
|
def tmpdir():
|
|
tmpdir = Path(tempfile.mkdtemp())
|
|
yield tmpdir
|
|
shutil.rmtree(tmpdir)
|
|
|
|
|
|
@test("get config raises ConfigFileDoesNotExist")
|
|
def _():
|
|
with raises(ConfigFileDoesNotExist):
|
|
get_config(Path("/foobar"))
|
|
|
|
|
|
@test("get config raises ConfigFileIsInvalid")
|
|
def _():
|
|
with raises(ConfigFileIsInvalid):
|
|
with NamedTemporaryFile() as f:
|
|
get_config(Path(f.name))
|
|
|
|
|
|
@test("get config raises ConfigFileIsInvalid from invalid file contents")
|
|
def _(tmpfile=tmpfile):
|
|
tmpfile.write("foobar")
|
|
with raises(ConfigFileIsInvalid):
|
|
get_config(Path(tmpfile.name))
|
|
|
|
|
|
@test("get config raises ConfigFileIsInvalid from pydantic validation")
|
|
def _(tmpfile=tmpfile):
|
|
yaml.dump({"foo": "bar"}, tmpfile, Dumper=yaml.SafeDumper)
|
|
with raises(ConfigFileIsInvalid):
|
|
get_config(Path(tmpfile.name))
|
|
|
|
|
|
@test("get config returns a correct config instance")
|
|
def _(tmpdir=tmpdir):
|
|
notes_root_path = Path(tmpdir / "notes")
|
|
notes_root_path.mkdir(exist_ok=True)
|
|
age_binary_path = Path(tmpdir / "age")
|
|
age_binary_path.touch(exist_ok=True)
|
|
encryption_keys_root_path = Path(tmpdir / "encryption_keys")
|
|
encryption_keys_root_path.mkdir(exist_ok=True)
|
|
public_key_path = Path(encryption_keys_root_path / "public.key")
|
|
public_key_path.touch(exist_ok=True)
|
|
private_key_path = Path(encryption_keys_root_path / "private.key")
|
|
private_key_path.touch(exist_ok=True)
|
|
config = Config(
|
|
notes_root_path=notes_root_path,
|
|
age_binary_path=age_binary_path,
|
|
encryption_keys=EncryptionKeysConfig(
|
|
public_key_path=public_key_path, private_key_path=private_key_path
|
|
),
|
|
)
|
|
assert config.notes_root_path == notes_root_path
|
|
assert config.age_binary_path == age_binary_path
|
|
assert config.encryption_keys.public_key_path == public_key_path
|
|
assert config.encryption_keys.private_key_path == private_key_path
|