snap-pac/tests.py
2021-02-14 16:41:31 -05:00

67 lines
2 KiB
Python

from configparser import ConfigParser
import tempfile
import os
import pytest
from snap_pac import create_snapper_cmd, get_snapper_configs, main, setup_config_parser
@pytest.fixture
def config():
config = ConfigParser()
config["DEFAULT"] = {
"snapshot": False,
"cleanup_algorithm": "number",
"pre_description": "\"foo\"",
"post_description": "\"bar\"",
"desc_limit": 72
}
config["root"] = {
"snapshot": True
}
config["home"] = {
"snapshot": True
}
return config
def test_create_snapper_pre_cmd(config):
snapper_cmd = create_snapper_cmd("pre", config, "root", "/tmp/somefile")
cmd = "snapper --config root create --type pre --cleanup-algorithm number --print-number --description \"foo\""
assert " ".join(snapper_cmd) == cmd
def test_create_snapper_post_cmd(config):
with tempfile.NamedTemporaryFile("w", delete=False) as f:
f.write("1234")
snapper_cmd = create_snapper_cmd("post", config, "root", f.name)
cmd = "snapper --config root create --type post --cleanup-algorithm number --print-number"
cmd += " --description \"bar\" --pre-number 1234"
assert " ".join(snapper_cmd) == cmd
def test_get_snapper_configs():
with tempfile.NamedTemporaryFile("w", delete=False) as f:
f.write("## Path: System/Snapper\n")
f.write("\n")
f.write("## Type: string\n")
f.write("## Default: \"\"\n")
f.write("# List of snapper configurations.\n")
f.write("SNAPPER_CONFIGS=\"home root foo bar\"\n")
name = f.name
assert get_snapper_configs(name) == ["home", "root", "foo", "bar"]
def test_skip_snap_pac():
os.environ["SNAP_PAC_SKIP"] = "y"
assert main("foo", "bar", "yep") is False
def test_setup_config_parser(config):
with tempfile.NamedTemporaryFile("w", delete=False) as f:
f.write("[home]\n")
f.write("snapshot = True")
name = f.name
config2 = setup_config_parser(name, "foo", "bar")
assert config == config2