This revision was landed with ongoing or failed builds.
This revision was automatically updated to reflect the committed changes.
Closed by commit rGd3a6b93135cd: [lldb/test] Consolidate interactive scripted 
process debugging test (authored by mib).

Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D149179/new/

https://reviews.llvm.org/D149179

Files:
  
lldb/test/API/functionalities/interactive_scripted_process/TestInteractiveScriptedProcess.py

Index: lldb/test/API/functionalities/interactive_scripted_process/TestInteractiveScriptedProcess.py
===================================================================
--- lldb/test/API/functionalities/interactive_scripted_process/TestInteractiveScriptedProcess.py
+++ lldb/test/API/functionalities/interactive_scripted_process/TestInteractiveScriptedProcess.py
@@ -12,15 +12,185 @@
 
     NO_DEBUG_INFO_TESTCASE = True
 
-    def test_passthrough_launch(self):
-        """Test a simple pass-through process launch"""
+    def setUp(self):
+        # Call super's setUp().
+        TestBase.setUp(self)
+        # Build and load test program
         self.build()
         self.runCmd("file " + self.getBuildArtifact("a.out"), CURRENT_EXECUTABLE_SET)
         self.main_source_file = lldb.SBFileSpec("main.cpp")
         self.script_module = "interactive_scripted_process"
         self.script_file = self.script_module + ".py"
+
+    def test_passthrough_launch(self):
+        """Test a simple pass-through process launch"""
         self.passthrough_launch()
 
+        lldbutil.run_break_set_by_source_regexp(self, "also break here")
+        self.assertEqual(self.mux_target.GetNumBreakpoints(), 2)
+        error = self.mux_process.Continue()
+        self.assertSuccess(error, "Resuming multiplexer scripted process")
+        self.assertTrue(self.mux_process.IsValid(), "Got a valid process")
+
+        event = lldbutil.fetch_next_event(
+            self, self.mux_process_listener, self.mux_process.GetBroadcaster()
+        )
+        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning)
+        event = lldbutil.fetch_next_event(
+            self, self.mux_process_listener, self.mux_process.GetBroadcaster()
+        )
+        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped)
+
+    def test_multiplexed_launch(self):
+        """Test a multiple interactive scripted process debugging"""
+        self.passthrough_launch()
+        self.assertEqual(self.dbg.GetNumTargets(), 2)
+
+        driving_target = self.mux_process.GetScriptedImplementation().driving_target
+        self.assertTrue(driving_target.IsValid(), "Driving target is invalid")
+
+        # Create a target for the multiplexed even scripted process
+        even_target = self.duplicate_target(driving_target)
+        self.assertTrue(
+            even_target.IsValid(),
+            "Couldn't duplicate driving target to launch multiplexed even scripted process",
+        )
+
+        class_name = f"{self.script_module}.MultiplexedScriptedProcess"
+        dictionary = {"driving_target_idx": self.dbg.GetIndexOfTarget(self.mux_target)}
+
+        dictionary["parity"] = 0
+        muxed_launch_info = self.get_launch_info(class_name, dictionary)
+
+        # Launch Even Child Scripted Process
+        error = lldb.SBError()
+        even_process = even_target.Launch(muxed_launch_info, error)
+        self.assertTrue(
+            even_process, "Couldn't launch multiplexed even scripted process"
+        )
+        self.multiplex(even_process)
+
+        # Check that the even process started running
+        event = lldbutil.fetch_next_event(
+            self, self.dbg.GetListener(), even_process.GetBroadcaster()
+        )
+        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning)
+        # Check that the even process stopped
+        event = lldbutil.fetch_next_event(
+            self, self.dbg.GetListener(), even_process.GetBroadcaster()
+        )
+        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped)
+
+        self.assertTrue(even_process.IsValid(), "Got a valid process")
+        self.assertState(
+            even_process.GetState(), lldb.eStateStopped, "Process is stopped"
+        )
+
+        # Create a target for the multiplexed odd scripted process
+        odd_target = self.duplicate_target(driving_target)
+        self.assertTrue(
+            odd_target.IsValid(),
+            "Couldn't duplicate driving target to launch multiplexed odd scripted process",
+        )
+
+        dictionary["parity"] = 1
+        muxed_launch_info = self.get_launch_info(class_name, dictionary)
+
+        # Launch Odd Child Scripted Process
+        error = lldb.SBError()
+        odd_process = odd_target.Launch(muxed_launch_info, error)
+        self.assertTrue(odd_process, "Couldn't launch multiplexed odd scripted process")
+        self.multiplex(odd_process)
+
+        # Check that the odd process started running
+        event = lldbutil.fetch_next_event(
+            self, self.dbg.GetListener(), odd_process.GetBroadcaster()
+        )
+        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning)
+        # Check that the odd process stopped
+        event = lldbutil.fetch_next_event(
+            self, self.dbg.GetListener(), odd_process.GetBroadcaster()
+        )
+        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped)
+
+        self.assertTrue(odd_process.IsValid(), "Got a valid process")
+        self.assertState(
+            odd_process.GetState(), lldb.eStateStopped, "Process is stopped"
+        )
+
+        # Set a breakpoint on the odd child process
+        bkpt = odd_target.BreakpointCreateBySourceRegex(
+            "also break here", self.main_source_file
+        )
+        self.assertEqual(odd_target.GetNumBreakpoints(), 1)
+        self.assertTrue(bkpt, "Second breakpoint set on child scripted process")
+        self.assertEqual(bkpt.GetNumLocations(), 1, "Second breakpoint has 1 location")
+
+        # Verify that the breakpoint was also set on the multiplexer & real target
+        self.assertEqual(self.mux_target.GetNumBreakpoints(), 2)
+        bkpt = self.mux_target.GetBreakpointAtIndex(1)
+        self.assertEqual(
+            bkpt.GetNumLocations(), 1, "Second breakpoint set on mux scripted process"
+        )
+        self.assertTrue(bkpt.MatchesName("multiplexed_scripted_process_421"))
+
+        self.assertGreater(driving_target.GetNumBreakpoints(), 1)
+
+        # Resume execution on child process
+        error = odd_process.Continue()
+        self.assertSuccess(error, "Resuming odd child scripted process")
+        self.assertTrue(odd_process.IsValid(), "Got a valid process")
+
+        # Since all the execution is asynchronous, the order in which events
+        # arrive is non-deterministic, so we need a data structure to make sure
+        # we received both the running and stopped event for each target.
+
+        # Initialize the execution event "bingo book", that maps a process index
+        # to a dictionary that contains flags that are not set for the process
+        # events that we care about (running & stopped)
+
+        execution_events = {
+            1: {lldb.eStateRunning: False, lldb.eStateStopped: False},
+            2: {lldb.eStateRunning: False, lldb.eStateStopped: False},
+            3: {lldb.eStateRunning: False, lldb.eStateStopped: False},
+        }
+
+        def fetch_process_event(self, execution_events):
+            event = lldbutil.fetch_next_event(
+                self,
+                self.dbg.GetListener(),
+                lldb.SBProcess.GetBroadcasterClass(),
+                match_class=True,
+            )
+            state = lldb.SBProcess.GetStateFromEvent(event)
+            self.assertIn(state, [lldb.eStateRunning, lldb.eStateStopped])
+            event_process = lldb.SBProcess.GetProcessFromEvent(event)
+            self.assertTrue(event_process.IsValid())
+            event_target = event_process.GetTarget()
+            event_target_idx = self.dbg.GetIndexOfTarget(event_target)
+            self.assertFalse(
+                execution_events[event_target_idx][state],
+                "Event already received for this process",
+            )
+            execution_events[event_target_idx][state] = True
+
+        event = lldbutil.fetch_next_event(
+            self, self.mux_process_listener, self.mux_process.GetBroadcaster()
+        )
+        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning)
+
+        event = lldbutil.fetch_next_event(
+            self, self.mux_process_listener, self.mux_process.GetBroadcaster()
+        )
+        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped)
+
+        for _ in range((self.dbg.GetNumTargets() - 1) * 2):
+            fetch_process_event(self, execution_events)
+
+        for target_index, event_states in execution_events.items():
+            for state, is_set in event_states.items():
+                self.assertTrue(is_set, f"Target {target_index} has state {state} set")
+
     def duplicate_target(self, driving_target):
         exe = driving_target.executable.fullpath
         triple = driving_target.triple
@@ -36,6 +206,14 @@
         launch_info.SetScriptedProcessDictionary(structured_data)
         return launch_info
 
+    def multiplex(self, muxed_process):
+        muxed_process.GetScriptedImplementation().multiplexer = (
+            self.mux_process.GetScriptedImplementation()
+        )
+        self.mux_process.GetScriptedImplementation().multiplexed_processes[
+            muxed_process.GetProcessID()
+        ] = muxed_process
+
     def passthrough_launch(self):
         """Test that a simple passthrough wrapper functions correctly"""
         # First build the real target:
@@ -49,42 +227,42 @@
         script_path = os.path.join(self.getSourceDir(), self.script_file)
         self.runCmd(f"command script import '{script_path}'")
 
-        mux_target = self.duplicate_target(real_target)
-        self.assertTrue(mux_target.IsValid(), "duplicate target succeeded")
+        self.mux_target = self.duplicate_target(real_target)
+        self.assertTrue(self.mux_target.IsValid(), "duplicate target succeeded")
 
         mux_class = f"{self.script_module}.MultiplexerScriptedProcess"
         script_dict = {"driving_target_idx": real_target_id}
         mux_launch_info = self.get_launch_info(mux_class, script_dict)
-        mux_process_listener = lldb.SBListener(
+        self.mux_process_listener = lldb.SBListener(
             "lldb.test.interactive-scripted-process.listener"
         )
-        mux_launch_info.SetPassthroughListener(mux_process_listener)
+        mux_launch_info.SetShadowListener(self.mux_process_listener)
 
         self.dbg.SetAsync(True)
         error = lldb.SBError()
-        mux_process = mux_target.Launch(mux_launch_info, error)
+        self.mux_process = self.mux_target.Launch(mux_launch_info, error)
         self.assertSuccess(error, "Launched multiplexer scripted process")
-        self.assertTrue(mux_process.IsValid(), "Got a valid process")
+        self.assertTrue(self.mux_process.IsValid(), "Got a valid process")
 
         # Check that the mux process started running
         event = lldbutil.fetch_next_event(
-            self, mux_process_listener, mux_process.GetBroadcaster(), timeout=60 * 5
+            self, self.mux_process_listener, self.mux_process.GetBroadcaster()
         )
         self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning)
         # Check that the real process started running
         event = lldbutil.fetch_next_event(
-            self, self.dbg.GetListener(), mux_process.GetBroadcaster()
+            self, self.dbg.GetListener(), self.mux_process.GetBroadcaster()
         )
         self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning)
 
         # Check that the real process stopped
         event = lldbutil.fetch_next_event(
-            self, self.dbg.GetListener(), mux_process.GetBroadcaster(), timeout=60 * 5
+            self, self.dbg.GetListener(), self.mux_process.GetBroadcaster()
         )
         self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped)
         # Check that the mux process stopped
         event = lldbutil.fetch_next_event(
-            self, mux_process_listener, mux_process.GetBroadcaster(), timeout=60 * 5
+            self, self.mux_process_listener, self.mux_process.GetBroadcaster()
         )
         self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped)
 
@@ -98,25 +276,10 @@
         # Check that we got the right threads:
         self.assertEqual(
             len(real_process.threads),
-            len(mux_process.threads),
+            len(self.mux_process.threads),
             "Same number of threads",
         )
         for id in range(len(real_process.threads)):
             real_pc = real_process.threads[id].frame[0].pc
-            mux_pc = mux_process.threads[id].frame[0].pc
+            mux_pc = self.mux_process.threads[id].frame[0].pc
             self.assertEqual(real_pc, mux_pc, f"PC's equal for {id}")
-
-        lldbutil.run_break_set_by_source_regexp(self, "also break here")
-        self.assertEqual(mux_target.GetNumBreakpoints(), 2)
-        error = mux_process.Continue()
-        self.assertSuccess(error, "Resuming multiplexer scripted process")
-        self.assertTrue(mux_process.IsValid(), "Got a valid process")
-
-        event = lldbutil.fetch_next_event(
-            self, mux_process_listener, mux_process.GetBroadcaster()
-        )
-        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning)
-        event = lldbutil.fetch_next_event(
-            self, mux_process_listener, mux_process.GetBroadcaster()
-        )
-        self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped)
_______________________________________________
lldb-commits mailing list
lldb-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/lldb-commits

Reply via email to