mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
2e75cc35a1
A kv vault argument can be the path to an SQLite vault file. With -s, kv copy reads the VAULT_TAG table and the plain-tag table into the VAULT_TAG table.
39 lines
1.3 KiB
Python
39 lines
1.3 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
|