Isolate the attack tests with setUp and tearDown methods

shreve · shreve · commit a011765b44fe · 2023-02-08T01:50:16.000-05:00
diff --git a/IPython/core/tests/test_interactiveshell.py b/IPython/core/tests/test_interactiveshell.py
@@ -1112,11 +1112,17 @@ def foo(*args, **kwargs):
     ip.Completer.custom_matchers.pop()
 
 
-class TestShowTracebacksAttack(unittest.TestCase):
+class TestShowTracebackAttack(unittest.TestCase):
     """Test that the interactive shell is resilient against the client attack of
     manipulating the showtracebacks method. These attacks shouldn't result in an
     unhandled exception in the kernel."""
 
+    def setUp(self):
+        self.orig_showtraceback = interactiveshell.InteractiveShell.showtraceback
+
+    def tearDown(self):
+        interactiveshell.InteractiveShell.showtraceback = self.orig_showtraceback
+
     def test_set_show_tracebacks_none(self):
         """Test the case of the client setting showtracebacks to None"""
 
