yadunut/clan-core
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

pkgs/generate-test-vars: speed up vars generation by precaching

Browse Source
This commit is contained in:
Johannes Kirschbauer
2025-04-22 16:43:04 +02:00
parent b47f2b6870
commit e417f79527
1 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pkgs/generate-test-vars/generate_test_vars/cli.py
View File

@@ -22,7 +22,7 @@ sops_priv_key = (
sops_pub_key = "age1qm0p4vf9jvcnn43s6l4prk8zn6cx0ep9gzvevxecv729xz540v8qa742eg"
def machine_names(repo_root: Path, check_attr: str) -> list[str]:
def get_machine_names(repo_root: Path, check_attr: str) -> list[str]:
"""
Get the machine names from the test flake
"""
@@ -133,12 +133,22 @@ def main() -> None:
shutil.rmtree(test_dir / "sops", ignore_errors=True)
flake = Flake(str(opts.repo_root))
machine_names = get_machine_names(
opts.repo_root,
opts.check_attr,
)
config = nix_config()
system = config["system"]
flake.precache(
[
f"checks.{system}.{opts.check_attr}.nodes.{{{','.join(machine_names)}}}.config.clan.core.vars.generators.*.validationHash",
f"checks.{system}.{opts.check_attr}.nodes.{{{','.join(machine_names)}}}.config.system.clan.deployment.file",
]
)
machines = [
TestMachine(name, flake, test_dir, opts.check_attr)
for name in machine_names(
opts.repo_root,
opts.check_attr,
)
TestMachine(name, flake, test_dir, opts.check_attr) for name in machine_names
]
user = "admin"
admin_key_path = Path(test_dir.resolve() / "sops" / "users" / user / "key.json")