79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
from pathlib import Path
|
|
from subprocess import CalledProcessError
|
|
|
|
import pytest
|
|
|
|
from halig.config import Config, EncryptionKeysConfig
|
|
from halig.edit import edit_note
|
|
from halig.exceptions import CouldNotEditTempfile
|
|
from tests.file_fixtures import tmpfile, tempfile # noqa: F401
|
|
|
|
from sh import age, ssh_keygen, yes
|
|
|
|
|
|
@pytest.fixture()
|
|
def notes_path(tmpdir): # noqa: F811
|
|
notes_path = Path(tmpdir.dirname) / "notes"
|
|
notes_path.mkdir(exist_ok=True)
|
|
|
|
yield notes_path
|
|
|
|
|
|
@pytest.fixture()
|
|
def config(notes_path): # noqa: F811
|
|
keys_path = Path(notes_path.parent / ".ssh")
|
|
keys_path.mkdir(exist_ok=True)
|
|
private_key_path = keys_path / "key"
|
|
public_key_path = keys_path / "key.pub"
|
|
|
|
ssh_keygen(yes(_piped=True), t="ed25519", f=str(private_key_path))
|
|
|
|
config = Config(
|
|
notes_root_path=notes_path,
|
|
encryption_keys=EncryptionKeysConfig(
|
|
public_key_path=public_key_path,
|
|
private_key_path=private_key_path,
|
|
),
|
|
)
|
|
|
|
note_path = notes_path / "note"
|
|
note_path.touch(exist_ok=True)
|
|
with open(note_path, "w") as f:
|
|
f.write("this is a test")
|
|
|
|
age(
|
|
"-R",
|
|
str(config.encryption_keys.public_key_path),
|
|
str(note_path),
|
|
_out=f"{note_path}.age",
|
|
)
|
|
|
|
yield config
|
|
|
|
|
|
def mock_subprocess(args: list):
|
|
with open(args[1], "w") as f:
|
|
f.write("mocked")
|
|
|
|
|
|
def test_edit_note(config: Config, mocker): # noqa: F811
|
|
mocker.patch("subprocess.call", side_effect=mock_subprocess)
|
|
edit_note(config.notes_root_path / "note.age", config)
|
|
assert (
|
|
age(
|
|
"-d",
|
|
"-i",
|
|
config.encryption_keys.private_key_path,
|
|
config.notes_root_path / "note.age",
|
|
)
|
|
== "mocked"
|
|
)
|
|
|
|
|
|
def test_edit_not_raises_could_not_edit_tempfile(config: Config, mocker): # noqa: F811
|
|
mocker.patch("subprocess.call", side_effect=CalledProcessError(
|
|
returncode=1,
|
|
cmd="foo",
|
|
output="mocked error"))
|
|
with pytest.raises(CouldNotEditTempfile):
|
|
edit_note(config.notes_root_path / "note.age", config)
|