generated from Luis/nextjs-python-web-template
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
import subprocess
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from cli import Cli
|
|
|
|
from clan_cli.machines.facts import machine_get_fact
|
|
from clan_cli.nix import nix_shell
|
|
from clan_cli.ssh import HostGroup
|
|
|
|
|
|
@pytest.mark.impure
|
|
def test_upload_secret(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
test_flake_with_core_and_pass: Path,
|
|
temporary_dir: Path,
|
|
host_group: HostGroup,
|
|
) -> None:
|
|
monkeypatch.chdir(test_flake_with_core_and_pass)
|
|
gnupghome = temporary_dir / "gpg"
|
|
gnupghome.mkdir(mode=0o700)
|
|
monkeypatch.setenv("GNUPGHOME", str(gnupghome))
|
|
monkeypatch.setenv("PASSWORD_STORE_DIR", str(temporary_dir / "pass"))
|
|
gpg_key_spec = temporary_dir / "gpg_key_spec"
|
|
gpg_key_spec.write_text(
|
|
"""
|
|
Key-Type: 1
|
|
Key-Length: 1024
|
|
Name-Real: Root Superuser
|
|
Name-Email: test@local
|
|
Expire-Date: 0
|
|
%no-protection
|
|
"""
|
|
)
|
|
cli = Cli()
|
|
subprocess.run(
|
|
nix_shell(["gnupg"], ["gpg", "--batch", "--gen-key", str(gpg_key_spec)]),
|
|
check=True,
|
|
)
|
|
subprocess.run(nix_shell(["pass"], ["pass", "init", "test@local"]), check=True)
|
|
cli.run(["secrets", "generate", "vm1"])
|
|
network_id = machine_get_fact("vm1", "zerotier-network-id")
|
|
assert len(network_id) == 16
|
|
identity_secret = (
|
|
temporary_dir / "pass" / "machines" / "vm1" / "zerotier-identity-secret.gpg"
|
|
)
|
|
secret1_mtime = identity_secret.lstat().st_mtime_ns
|
|
|
|
# test idempotency
|
|
cli.run(["secrets", "generate", "vm1"])
|
|
assert identity_secret.lstat().st_mtime_ns == secret1_mtime
|
|
|
|
flake = test_flake_with_core_and_pass.joinpath("flake.nix")
|
|
host = host_group.hosts[0]
|
|
addr = f"{host.user}@{host.host}:{host.port}?StrictHostKeyChecking=no&UserKnownHostsFile=/dev/null&IdentityFile={host.key}"
|
|
new_text = flake.read_text().replace("__CLAN_DEPLOYMENT_ADDRESS__", addr)
|
|
flake.write_text(new_text)
|
|
cli.run(["secrets", "upload", "vm1"])
|
|
zerotier_identity_secret = (
|
|
test_flake_with_core_and_pass / "secrets" / "zerotier-identity-secret"
|
|
)
|
|
assert zerotier_identity_secret.exists()
|