Fix tests

Author: pablogsal

Lib/profiling/sampling/cli.py

@@ -390,13 +390,17 @@ def _validate_args(args, parser):
             "Live mode requires the curses module, which is not available."
         )
 
-    # Async-aware mode is incompatible with --native and --gc/--no-gc
+    # Async-aware mode is incompatible with --native, --no-gc, --mode, and --all-threads
     if args.async_aware is not None:
         issues = []
         if args.native:
             issues.append("--native")
         if not args.gc:
             issues.append("--no-gc")
+        if hasattr(args, 'mode') and args.mode != "wall":
+            issues.append(f"--mode={args.mode}")
+        if hasattr(args, 'all_threads') and args.all_threads:
+            issues.append("--all-threads")
         if issues:
             parser.error(
                 f"Options {', '.join(issues)} are incompatible with --async-aware. "

Lib/test/test_profiling/test_sampling_profiler/test_cli.py

@@ -659,3 +659,37 @@ def test_async_aware_incompatible_with_both_native_and_no_gc(self):
         self.assertIn("--native", error_msg)
         self.assertIn("--no-gc", error_msg)
         self.assertIn("incompatible with --async-aware", error_msg)
+
+    def test_async_aware_incompatible_with_mode(self):
+        """Test --async-aware is incompatible with --mode (non-wall)."""
+        test_args = ["profiling.sampling.cli", "attach", "12345", "--async-aware", "all", "--mode", "cpu"]
+
+        with (
+            mock.patch("sys.argv", test_args),
+            mock.patch("sys.stderr", io.StringIO()) as mock_stderr,
+            self.assertRaises(SystemExit) as cm,
+        ):
+            from profiling.sampling.cli import main
+            main()
+
+        self.assertEqual(cm.exception.code, 2)  # argparse error
+        error_msg = mock_stderr.getvalue()
+        self.assertIn("--mode=cpu", error_msg)
+        self.assertIn("incompatible with --async-aware", error_msg)
+
+    def test_async_aware_incompatible_with_all_threads(self):
+        """Test --async-aware is incompatible with --all-threads."""
+        test_args = ["profiling.sampling.cli", "attach", "12345", "--async-aware", "running", "--all-threads"]
+
+        with (
+            mock.patch("sys.argv", test_args),
+            mock.patch("sys.stderr", io.StringIO()) as mock_stderr,
+            self.assertRaises(SystemExit) as cm,
+        ):
+            from profiling.sampling.cli import main
+            main()
+
+        self.assertEqual(cm.exception.code, 2)  # argparse error
+        error_msg = mock_stderr.getvalue()
+        self.assertIn("--all-threads", error_msg)
+        self.assertIn("incompatible with --async-aware", error_msg)

Lib/test/test_profiling/test_sampling_profiler/test_integration.py

@@ -909,4 +909,6 @@ def test_async_aware_running_sees_only_cpu_task(self):
         # Should see the tree structure (supervisor in the parent chain)
         self.assertIn("supervisor", output)
         # Should see task boundary markers
-        self.assertIn("<task>", output)
+        self.assertIn("<task>", output)
+        # async_aware="running" should NOT see sleeping tasks
+        self.assertNotIn("sleeping_leaf", output)