65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from typing import List
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
|
||
|
|
class HardRestartArgTests(unittest.TestCase):
|
||
|
|
"""The hard-restart list defaults to empty (no compose down/up); callers
|
||
|
|
opt in per dir, e.g. compose hosts pass 'mailu' while swarm hosts, where
|
||
|
|
the dir is a stack whose overlay network collides with compose up, pass
|
||
|
|
nothing."""
|
||
|
|
|
||
|
|
def _parse(self, extra: List[str]):
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from baudolo.backup import cli
|
||
|
|
|
||
|
|
argv = [
|
||
|
|
"baudolo",
|
||
|
|
"--compose-dir",
|
||
|
|
"/tmp",
|
||
|
|
"--backups-dir",
|
||
|
|
"/tmp/backup",
|
||
|
|
"--database-containers",
|
||
|
|
"postgres",
|
||
|
|
"--images-no-stop-required",
|
||
|
|
"redis",
|
||
|
|
*extra,
|
||
|
|
]
|
||
|
|
with patch.object(sys, "argv", argv):
|
||
|
|
return cli.parse_args()
|
||
|
|
|
||
|
|
def test_default_is_empty(self) -> None:
|
||
|
|
args = self._parse([])
|
||
|
|
self.assertEqual(args.hard_restart_projects, [])
|
||
|
|
|
||
|
|
def test_empty_flag_stays_empty(self) -> None:
|
||
|
|
args = self._parse(["--hard-restart-projects"])
|
||
|
|
self.assertEqual(args.hard_restart_projects, [])
|
||
|
|
|
||
|
|
def test_explicit_names_preserved(self) -> None:
|
||
|
|
args = self._parse(["--hard-restart-projects", "mailu", "foo"])
|
||
|
|
self.assertEqual(args.hard_restart_projects, ["mailu", "foo"])
|
||
|
|
|
||
|
|
def test_backups_dir_is_required(self) -> None:
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from baudolo.backup import cli
|
||
|
|
|
||
|
|
argv = [
|
||
|
|
"baudolo",
|
||
|
|
"--compose-dir",
|
||
|
|
"/tmp",
|
||
|
|
"--database-containers",
|
||
|
|
"postgres",
|
||
|
|
"--images-no-stop-required",
|
||
|
|
"redis",
|
||
|
|
]
|
||
|
|
with patch.object(sys, "argv", argv), self.assertRaises(SystemExit):
|
||
|
|
cli.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main(verbosity=2)
|