125 lines
3.4 KiB
Python
125 lines
3.4 KiB
Python
import json
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
import yaml
|
|
from pyrage.ssh import Identity, Recipient
|
|
|
|
from halig.encryption import Encryptor
|
|
from halig.settings import Settings
|
|
|
|
|
|
@pytest.fixture()
|
|
def halig_ssh_public_key():
|
|
return (
|
|
"ssh-ed25519 "
|
|
"AAAAC3NzaC1lZDI1NTE5AAAAIGjHhIF/DlVCb2dRFMlKia7nij1Aq+zRDCaMIwe/VKDh"
|
|
" foo@bar"
|
|
)
|
|
|
|
|
|
@pytest.fixture()
|
|
def halig_ssh_private_key():
|
|
return """-----BEGIN OPENSSH PRIVATE KEY-----
|
|
b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW
|
|
QyNTUxOQAAACBox4SBfw5VQm9nURTJSomu54o9QKvs0QwmjCMHv1Sg4QAAAJhvD2Jxbw9i
|
|
cQAAAAtzc2gtZWQyNTUxOQAAACBox4SBfw5VQm9nURTJSomu54o9QKvs0QwmjCMHv1Sg4Q
|
|
AAAEAZANW15ieou1ds73BlM1nqzyZ2A0454JnB3QirZycGv2jHhIF/DlVCb2dRFMlKia7n
|
|
ij1Aq+zRDCaMIwe/VKDhAAAAEXJvb3RANGNjNWUxOWYyYThiAQIDBA==
|
|
-----END OPENSSH PRIVATE KEY-----
|
|
"""
|
|
|
|
|
|
@pytest.fixture()
|
|
def ssh_identity(halig_ssh_private_key: str) -> Identity:
|
|
return Identity.from_buffer(halig_ssh_private_key.encode())
|
|
|
|
|
|
@pytest.fixture()
|
|
def ssh_recipient(halig_ssh_public_key: str) -> Recipient:
|
|
return Recipient.from_str(halig_ssh_public_key)
|
|
|
|
|
|
# @pytest.fixture()
|
|
# def halig_path(fs, halig_ssh_public_key, halig_ssh_private_key) -> Path:
|
|
# fs.add_real_paths(["/etc/localtime"])
|
|
# ssh_path = Path("~/.ssh").expanduser()
|
|
# ssh_path.mkdir(parents=True)
|
|
#
|
|
# with (ssh_path / "id_ed25519").open("w") as f:
|
|
# f.write(halig_ssh_private_key)
|
|
#
|
|
# with (ssh_path / "id_ed25519.pub").open("w") as f:
|
|
# f.write(halig_ssh_public_key)
|
|
#
|
|
# halig_path = Path("~/.config/halig").expanduser()
|
|
# halig_path.mkdir(parents=True)
|
|
# return halig_path
|
|
|
|
@pytest.fixture()
|
|
def halig_ssh_path(tmp_path: Path, halig_ssh_public_key, halig_ssh_private_key) -> Path:
|
|
ssh_path = tmp_path / ".ssh"
|
|
ssh_path.mkdir()
|
|
|
|
with (ssh_path / "id_ed25519").open("w") as f:
|
|
f.write(halig_ssh_private_key)
|
|
|
|
with (ssh_path / "id_ed25519.pub").open("w") as f:
|
|
f.write(halig_ssh_public_key)
|
|
|
|
return ssh_path
|
|
|
|
|
|
@pytest.fixture()
|
|
def halig_config_path(tmp_path: Path):
|
|
halig_path = tmp_path / ".config/halig"
|
|
halig_path.mkdir(parents=True)
|
|
return halig_path
|
|
|
|
|
|
@pytest.fixture()
|
|
def notebooks_path(tmp_path) -> Path:
|
|
notebooks_path = tmp_path / "Notebooks"
|
|
notebooks_path.mkdir()
|
|
return notebooks_path
|
|
|
|
|
|
@pytest.fixture()
|
|
def settings(notebooks_path: Path, halig_ssh_path) -> Settings:
|
|
return Settings(
|
|
notebooks_root_path=notebooks_path,
|
|
identity_paths=[halig_ssh_path / "id_ed25519"],
|
|
recipient_paths=[halig_ssh_path / "id_ed25519.pub"]
|
|
)
|
|
|
|
|
|
@pytest.fixture()
|
|
def settings_file_path(halig_config_path: Path, notebooks_path: Path) -> Path:
|
|
yaml_file = halig_config_path / "halig.yml"
|
|
yaml_file.touch()
|
|
s = Settings(notebooks_root_path=notebooks_path)
|
|
# `.dict()` doesn't serialize some fields that yaml doesn't understand
|
|
serialized = json.loads(s.model_dump_json())
|
|
with yaml_file.open("w") as f:
|
|
yaml.safe_dump(serialized, f)
|
|
return yaml_file
|
|
|
|
|
|
@pytest.fixture()
|
|
def empty_file_path(halig_config_path: Path) -> Path:
|
|
empty_path = halig_config_path / "empty"
|
|
empty_path.touch()
|
|
return empty_path
|
|
|
|
|
|
@pytest.fixture()
|
|
def notebooks_root_path_envvar(notebooks_path: Path):
|
|
os.environ["HALIG_NOTEBOOKS_ROOT_PATH"] = str(notebooks_path)
|
|
yield notebooks_path
|
|
del os.environ["HALIG_NOTEBOOKS_ROOT_PATH"]
|
|
|
|
|
|
@pytest.fixture()
|
|
def encryptor(settings: Settings) -> Encryptor:
|
|
return Encryptor(settings)
|