mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
3027592e31
Vaults.add_keys and kv add skip null content keys instead of failing the batch; is_null_key moves to core.vault. A manifest snapshot restore has one 32 MiB inflate budget. Tests reset the speed limiter and config between runs.
55 lines
2.1 KiB
Python
55 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from click.testing import CliRunner
|
|
|
|
from unshackle.commands import kv as kv_module
|
|
from unshackle.vaults.SQLite import SQLite
|
|
|
|
KID = "0" * 31 + "1"
|
|
KEY = "a" * 32
|
|
|
|
|
|
def test_kv_copy_moves_a_plain_tag_table_from_a_file_into_the_vault_tag_table(
|
|
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
source = tmp_path / "key_store.db"
|
|
SQLite("seed", source).add_key("TESTSVC", KID, KEY)
|
|
dest = tmp_path / "local.db"
|
|
monkeypatch.setattr(kv_module.config, "key_vaults", [{"type": "SQLite", "name": "local", "path": str(dest)}])
|
|
monkeypatch.setattr(kv_module.Services, "get_vault_tag", lambda name: "Shared" if name == "TESTSVC" else name)
|
|
|
|
result = CliRunner().invoke(kv_module.kv, ["copy", "local", str(source), "-s", "TESTSVC"])
|
|
|
|
assert result.exit_code == 0, result.output
|
|
check = SQLite("check", dest)
|
|
assert check.get_key(KID, "Shared") == KEY
|
|
assert check.get_key(KID, "TESTSVC") is None
|
|
|
|
|
|
def test_kv_copy_rejects_a_name_that_is_neither_configured_nor_a_file(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(kv_module.config, "key_vaults", [])
|
|
|
|
result = CliRunner().invoke(kv_module.kv, ["copy", "local", "nowhere.db"])
|
|
|
|
assert result.exit_code != 0
|
|
assert "not defined in the config" in result.output
|
|
|
|
|
|
def test_kv_add_skips_a_null_key_line_and_adds_the_rest(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
keys = tmp_path / "keys.txt"
|
|
null_kid = "0" * 31 + "2"
|
|
keys.write_text(f"{KID}:{KEY}\n{null_kid}:{'0' * 32}\n", encoding="utf8")
|
|
dest = tmp_path / "local.db"
|
|
monkeypatch.setattr(kv_module.config, "key_vaults", [{"type": "SQLite", "name": "local", "path": str(dest)}])
|
|
monkeypatch.setattr(kv_module.Services, "get_vault_tag", lambda name: name)
|
|
|
|
result = CliRunner().invoke(kv_module.kv, ["add", str(keys), "TESTSVC", "local"])
|
|
|
|
assert result.exit_code == 0, result.output
|
|
check = SQLite("check", dest)
|
|
assert check.get_key(KID, "TESTSVC") == KEY
|
|
assert check.get_key(null_kid, "TESTSVC") is None
|