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)