Cleaned Up tests

tests/e2e/test_branch_commands.py

@@ -0,0 +1,109 @@
+from __future__ import annotations
+
+import runpy
+import sys
+import unittest
+from unittest.mock import patch
+
+
+class TestIntegrationBranchCommands(unittest.TestCase):
+    """
+    Integration tests for the `pkgmgr branch` CLI wiring.
+
+    These tests execute the real entry point (main.py) and mock
+    the high-level helpers to ensure that argument parsing and
+    dispatch behave as expected.
+    """
+
+    def _run_pkgmgr(self, extra_args: list[str]) -> None:
+        """
+        Run the main entry point with the given extra args, as if called via:
+
+            pkgmgr <extra_args...>
+
+        We explicitly set sys.argv and execute main.py as __main__ using runpy.
+        """
+        original_argv = list(sys.argv)
+        try:
+            # argv[0] is the program name; the rest are CLI arguments.
+            sys.argv = ["pkgmgr"] + list(extra_args)
+            runpy.run_module("main", run_name="__main__")
+        finally:
+            sys.argv = original_argv
+
+    @patch("pkgmgr.cli.commands.branch.open_branch")
+    def test_branch_open_with_name_and_base(self, mock_open_branch) -> None:
+        """
+        `pkgmgr branch open feature/test --base develop` must forward
+        the name and base branch to open_branch() with cwd=".".
+        """
+        self._run_pkgmgr(
+            ["branch", "open", "feature/test", "--base", "develop"]
+        )
+
+        mock_open_branch.assert_called_once()
+        _, kwargs = mock_open_branch.call_args
+        self.assertEqual(kwargs.get("name"), "feature/test")
+        self.assertEqual(kwargs.get("base_branch"), "develop")
+        self.assertEqual(kwargs.get("cwd"), ".")
+
+    @patch("pkgmgr.cli.commands.branch.open_branch")
+    def test_branch_open_without_name_uses_default_base(
+        self,
+        mock_open_branch,
+    ) -> None:
+        """
+        `pkgmgr branch open` without a name must still call open_branch(),
+        passing name=None and the default base branch 'main'.
+        """
+        self._run_pkgmgr(["branch", "open"])
+
+        mock_open_branch.assert_called_once()
+        _, kwargs = mock_open_branch.call_args
+        self.assertIsNone(kwargs.get("name"))
+        self.assertEqual(kwargs.get("base_branch"), "main")
+        self.assertEqual(kwargs.get("cwd"), ".")
+
+    # ------------------------------------------------------------------
+    # close subcommand
+    # ------------------------------------------------------------------
+
+    @patch("pkgmgr.cli.commands.branch.close_branch")
+    def test_branch_close_with_name_and_base(self, mock_close_branch) -> None:
+        """
+        `pkgmgr branch close feature/test --base develop` must forward
+        the name and base branch to close_branch() with cwd=".".
+        """
+        self._run_pkgmgr(
+            ["branch", "close", "feature/test", "--base", "develop"]
+        )
+
+        mock_close_branch.assert_called_once()
+        _, kwargs = mock_close_branch.call_args
+        self.assertEqual(kwargs.get("name"), "feature/test")
+        self.assertEqual(kwargs.get("base_branch"), "develop")
+        self.assertEqual(kwargs.get("cwd"), ".")
+
+    @patch("pkgmgr.cli.commands.branch.close_branch")
+    def test_branch_close_without_name_uses_default_base(
+        self,
+        mock_close_branch,
+    ) -> None:
+        """
+        `pkgmgr branch close` without a name must still call close_branch(),
+        passing name=None and the default base branch 'main'.
+
+        The branch helper will then resolve the actual base (main/master)
+        internally.
+        """
+        self._run_pkgmgr(["branch", "close"])
+
+        mock_close_branch.assert_called_once()
+        _, kwargs = mock_close_branch.call_args
+        self.assertIsNone(kwargs.get("name"))
+        self.assertEqual(kwargs.get("base_branch"), "main")
+        self.assertEqual(kwargs.get("cwd"), ".")
+
+
+if __name__ == "__main__":
+    unittest.main()