+cli tests

2026-06-22 11:43:00 +08:00
parent 0b97846d77
commit 48f6d8a7f0
22 changed files with 2930 additions and 42 deletions
@@ -0,0 +1,107 @@
+"""Tests for cli.taskkill module."""
+
+from __future__ import annotations
+
+from unittest.mock import patch
+
+import pytest
+
+import pyflowx as px
+from pyflowx.cli import taskkill
+from pyflowx.conditions import Constants
+
+
+# ---------------------------------------------------------------------- #
+# main function
+# ---------------------------------------------------------------------- #
+class TestMain:
+    """Test main function."""
+
+    def test_main_with_single_process(self) -> None:
+        """main() should handle single process argument."""
+        with patch("sys.argv", ["taskkill", "chrome.exe"]), \
+             patch.object(px, "run") as mock_run:
+            taskkill.main()
+            assert mock_run.called
+            graph = mock_run.call_args[0][0]
+            assert isinstance(graph, px.Graph)
+
+    def test_main_with_multiple_processes(self) -> None:
+        """main() should handle multiple process arguments."""
+        with patch("sys.argv", ["taskkill", "chrome.exe", "python.exe", "node.exe"]), \
+             patch.object(px, "run") as mock_run:
+            taskkill.main()
+            assert mock_run.called
+            graph = mock_run.call_args[0][0]
+            assert isinstance(graph, px.Graph)
+
+    def test_main_with_no_args_shows_help(self) -> None:
+        """main() with no args should show help and exit."""
+        with patch("sys.argv", ["taskkill"]), pytest.raises(SystemExit) as exc_info:
+            taskkill.main()
+        assert exc_info.value.code == 2
+
+    def test_main_creates_task_specs_with_correct_names(self) -> None:
+        """main() should create TaskSpecs with correct names."""
+        with patch("sys.argv", ["taskkill", "chrome.exe", "python.exe"]), \
+             patch.object(px, "run") as mock_run:
+            taskkill.main()
+            graph = mock_run.call_args[0][0]
+            task_names = list(graph.all_specs().keys())
+            assert "kill_chrome.exe" in task_names
+            assert "kill_python.exe" in task_names
+
+    def test_main_uses_thread_strategy(self) -> None:
+        """main() should use thread strategy."""
+        with patch("sys.argv", ["taskkill", "chrome.exe"]), \
+             patch.object(px, "run") as mock_run:
+            taskkill.main()
+            assert mock_run.call_args[1]["strategy"] == "thread"
+
+    def test_main_windows_command_format(self) -> None:
+        """main() should use Windows command format on Windows."""
+        if Constants.IS_WINDOWS:
+            with patch("sys.argv", ["taskkill", "chrome.exe"]), \
+                 patch.object(px, "run") as mock_run:
+                taskkill.main()
+                graph = mock_run.call_args[0][0]
+                specs = graph.all_specs()
+                # Check that command includes Windows taskkill format
+                for spec in specs.values():
+                    assert spec.cmd[0] == "taskkill"
+                    assert spec.cmd[1] == "/f"
+                    assert spec.cmd[2] == "/im"
+
+    def test_main_linux_command_format(self) -> None:
+        """main() should use Linux command format on Linux."""
+        with patch.object(Constants, "IS_WINDOWS", False), \
+             patch("sys.argv", ["taskkill", "chrome.exe"]), \
+             patch.object(px, "run") as mock_run:
+            taskkill.main()
+            graph = mock_run.call_args[0][0]
+            specs = graph.all_specs()
+            # Check that command includes Linux pkill format
+            for spec in specs.values():
+                assert spec.cmd[0] == "pkill"
+                assert spec.cmd[1] == "-f"
+
+    def test_main_tasks_have_verbose_true(self) -> None:
+        """main() should create tasks with verbose=True."""
+        with patch("sys.argv", ["taskkill", "chrome.exe"]), \
+             patch.object(px, "run") as mock_run:
+            taskkill.main()
+            graph = mock_run.call_args[0][0]
+            specs = graph.all_specs()
+            for spec in specs.values():
+                assert spec.verbose is True
+
+    def test_main_adds_wildcard_to_process_name(self) -> None:
+        """main() should add wildcard to process name."""
+        with patch("sys.argv", ["taskkill", "chrome.exe"]), \
+             patch.object(px, "run") as mock_run:
+            taskkill.main()
+            graph = mock_run.call_args[0][0]
+            specs = graph.all_specs()
+            # Check that wildcard is added
+            for spec in specs.values():
+                assert spec.cmd[-1].endswith("*")