Update the sidebar after moving the prompt

Author: serhiy-storchaka

Lib/idlelib/idle_test/test_pyshell.py

@@ -106,6 +106,7 @@ def test_output_at_prompt(self):
         self.assertEqual(text.get('iomark-6c', 'end-1c'), 'hello\na = (')
         self.assertIn('console', text.tag_names('iomark-1c'))
         self.assertNotIn('console', text.tag_names('iomark-7c'))
+        self.assertEqual(shell.shell_sidebar.line_prompts, {3: '>>>'})
 
 
 class PyShellRemoveLastNewlineAndSurroundingWhitespaceTest(unittest.TestCase):

Lib/idlelib/pyshell.py

@@ -1449,6 +1449,7 @@ def write(self, s, tags=()):
                            # let's find out what they are and be specific.
         if at_prompt and s.endswith('\n'):
             text.tag_add("console", "iomark-1c")
+            self.shell_sidebar.update_sidebar()
         if self.canceled:
             self.canceled = False
             if not use_subprocess: