fix bpython#450 - run startupfile on undo

thomasballinger · thomasballinger · commit a679338ac2af · 2015-01-22T13:38:25.000-05:00
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -1306,6 +1306,7 @@ def reevaluate(self, insert_into_history=False):
         self.display_buffer = []
         self.highlighted_paren = None
 
+        self.process_event(bpythonevents.RunStartupFileEvent())
         self.reevaluating = True
         sys.stdin = ReevaluateFakeStdin(self.stdin, self)
         for line in old_logical_lines:
diff --git a/bpython/test/test_curtsies_repl.py b/bpython/test/test_curtsies_repl.py
@@ -257,5 +257,17 @@ def test_complex(self):
         self.assertEqual(self.repl.predicted_indent('reduce(asdfasdf,'), 7)
 
 
+class TestCurtsiesReevaluate(unittest.TestCase):
+    def setUp(self):
+        self.repl = create_repl()
+
+    def test_variable_is_cleared(self):
+        self.repl._current_line = 'b = 10'
+        self.repl.on_enter()
+        self.assertIn('b', self.repl.interp.locals)
+        self.repl.undo()
+        self.assertNotIn('b', self.repl.interp.locals)
+
+
 if __name__ == '__main__':
     unittest.main()
