✏️ fix sdtout bug when user select not to enter debugger

Ackee-Blockchain · Sep 17, 2024 · 46d4adc · 46d4adc
1 parent dcd9346
commit 46d4adc
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/wake/testing/custom_pdb.py b/wake/testing/custom_pdb.py
@@ -31,6 +31,4 @@ def cleanup_before_exit(self):
         """
         This function performs the cleanup before exiting the debugger.
         """
-        print("Performing cleanup before exiting the debugger...")
-        self._program_instance._setup_stdio()
-        self._program_instance._conn.send(("breakpoint_handled",))
+        self._program_instance.exit_debugger()
diff --git a/wake/testing/pytest_plugin_multiprocess.py b/wake/testing/pytest_plugin_multiprocess.py
@@ -147,6 +147,10 @@ def pytest_exception_interact(self, node, call, report):
             self._exception_handler(
                 call.excinfo.type, call.excinfo.value, call.excinfo.tb
             )
+
+    def exit_debugger(self):
+        self._setup_stdio()
+        self._conn.send(("breakpoint_handled",))
 
     def pytest_runtestloop(self, session: Session):
         if (
@@ -192,7 +196,7 @@ def custom_debugger():
                 p.set_trace(frame)
             else:
                 # trace nothing, same as continue
-                self._conn.send(("breakpoint_handled",))  
+                self.exit_debugger()
 
         sys.breakpointhook = custom_debugger
 

diff --git a/wake/testing/pytest_plugin_multiprocess_server.py b/wake/testing/pytest_plugin_multiprocess_server.py
@@ -254,7 +254,7 @@ def pytest_runtestloop(self, session: pytest.Session):
                             attach = None
                             while attach is None:
                                 response = input(
-                                    "Would you like to attach the debugger? [y/n] ('n' to continue operations)"
+                                    "Would you like to attach the debugger? [y/n] "
                                 )
                                 if response == "y":
                                     attach = True