118 lines
4.0 KiB
Python
118 lines
4.0 KiB
Python
import io
|
|
import unittest
|
|
from contextlib import redirect_stdout
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import automtu.persist as persist
|
|
|
|
|
|
class TestPersist(unittest.TestCase):
|
|
def test_strip_persist_args_removes_flag_and_value(self) -> None:
|
|
argv = ["automtu", "--apply-wg-mtu", "--persist", "systemd"]
|
|
self.assertEqual(
|
|
persist._strip_persist_args(argv),
|
|
["automtu", "--apply-wg-mtu"],
|
|
)
|
|
|
|
def test_strip_persist_args_removes_equals_form(self) -> None:
|
|
argv = ["automtu", "--apply-wg-mtu", "--persist=systemd"]
|
|
self.assertEqual(
|
|
persist._strip_persist_args(argv),
|
|
["automtu", "--apply-wg-mtu"],
|
|
)
|
|
|
|
def test_strip_persist_args_removes_uninstall(self) -> None:
|
|
argv = ["automtu", "--persist", "docker", "--uninstall", "--apply-wg-mtu"]
|
|
self.assertEqual(
|
|
persist._strip_persist_args(argv),
|
|
["automtu", "--apply-wg-mtu"],
|
|
)
|
|
|
|
def test_persist_systemd_dry_run_prints_unit(self) -> None:
|
|
argv = [
|
|
"automtu",
|
|
"--auto-pmtu-from-wg",
|
|
"--apply-wg-mtu",
|
|
"--persist",
|
|
"systemd",
|
|
]
|
|
|
|
with (
|
|
patch("automtu.persist.shutil.which", return_value="/usr/bin/automtu"),
|
|
patch("automtu.persist._SYSTEMD_UNIT_PATH", Path("/tmp/automtu.service")),
|
|
):
|
|
out = io.StringIO()
|
|
with redirect_stdout(out):
|
|
persist.persist_systemd(argv, dry=True)
|
|
|
|
s = out.getvalue()
|
|
self.assertIn("DRY-RUN", s)
|
|
self.assertIn(
|
|
"ExecStart=/usr/bin/automtu --auto-pmtu-from-wg --apply-wg-mtu", s
|
|
)
|
|
|
|
def test_persist_systemd_adds_docker_ordering_if_apply_docker_mtu_present(
|
|
self,
|
|
) -> None:
|
|
argv = ["automtu", "--apply-docker-mtu", "--persist", "systemd"]
|
|
|
|
with (
|
|
patch("automtu.persist.shutil.which", return_value="/usr/bin/automtu"),
|
|
patch("automtu.persist._SYSTEMD_UNIT_PATH", Path("/tmp/automtu.service")),
|
|
):
|
|
out = io.StringIO()
|
|
with redirect_stdout(out):
|
|
persist.persist_systemd(argv, dry=True)
|
|
|
|
s = out.getvalue()
|
|
self.assertIn("After=network-online.target docker.service", s)
|
|
self.assertIn("Wants=network-online.target docker.service", s)
|
|
|
|
def test_persist_docker_dry_run_prints_unit_with_docker_ordering(self) -> None:
|
|
argv = ["automtu", "--dry-run", "--persist", "docker"]
|
|
|
|
with (
|
|
patch("automtu.persist.shutil.which", return_value="/usr/bin/automtu"),
|
|
patch(
|
|
"automtu.persist._DOCKER_SYSTEMD_UNIT_PATH",
|
|
Path("/tmp/automtu-docker.service"),
|
|
),
|
|
):
|
|
out = io.StringIO()
|
|
with redirect_stdout(out):
|
|
persist.persist_docker(argv, dry=True)
|
|
|
|
s = out.getvalue()
|
|
self.assertIn("DRY-RUN", s)
|
|
self.assertIn("After=network-online.target docker.service", s)
|
|
self.assertIn("Wants=network-online.target docker.service", s)
|
|
self.assertIn("ExecStart=/usr/bin/automtu --dry-run", s)
|
|
|
|
def test_uninstall_systemd_dry_run_prints_actions(self) -> None:
|
|
with patch("automtu.persist._SYSTEMD_UNIT_PATH", Path("/tmp/automtu.service")):
|
|
out = io.StringIO()
|
|
with redirect_stdout(out):
|
|
persist.uninstall_systemd(dry=True)
|
|
|
|
s = out.getvalue()
|
|
self.assertIn("DRY-RUN", s)
|
|
self.assertIn("systemctl disable", s)
|
|
|
|
def test_uninstall_docker_dry_run_prints_actions(self) -> None:
|
|
with patch(
|
|
"automtu.persist._DOCKER_SYSTEMD_UNIT_PATH",
|
|
Path("/tmp/automtu-docker.service"),
|
|
):
|
|
out = io.StringIO()
|
|
with redirect_stdout(out):
|
|
persist.uninstall_docker(dry=True)
|
|
|
|
s = out.getvalue()
|
|
self.assertIn("DRY-RUN", s)
|
|
self.assertIn("automtu-docker.service", s)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(verbosity=2)
|