bugfix: ui: Focus on compose area when draft is opened from side-panels.

This commit fixes a bug that caused the focus to not move to the central
area if the draft was opened from the side-panels. This is a minor bugfix
that helps redirect the focus to the compose area in such cases.

The recently added tests for OPEN_DRAFT are amended in this commit to
match the added bugfix.

Author: zee-bit

tests/ui/test_ui.py

@@ -344,6 +344,7 @@ def test_keypress_autohide_streams(self, view, mocker, autohide, key, widget_siz
     )
     @pytest.mark.parametrize("key", keys_for_command("OPEN_DRAFT"))
     def test_keypress_OPEN_DRAFT(self, view, mocker, draft, key, widget_size):
+        view.body = mocker.Mock()
         view.middle_column = mocker.Mock()
         view.set_footer_text = mocker.Mock()
         view.controller.is_in_editor_mode = lambda: False
@@ -367,6 +368,7 @@ def test_keypress_OPEN_DRAFT(self, view, mocker, draft, key, widget_size):
                     emails=draft["to"], recipient_user_ids=[1, 2]
                 )
 
+            assert view.body.focus_col == 1
             assert view.write_box.msg_write_box.edit_text == draft["content"]
             assert view.write_box.msg_write_box.edit_pos == len(draft["content"])
             view.middle_column.set_focus.assert_called_once_with("footer")

zulipterminal/ui.py

@@ -262,6 +262,7 @@ def keypress(self, size: Tuple[int, int], key: str) -> Optional[str]:
                 content = saved_draft["content"]
                 self.write_box.msg_write_box.edit_text = content
                 self.write_box.msg_write_box.edit_pos = len(content)
+                self.body.focus_col = 1
                 self.middle_column.set_focus("footer")
             else:
                 self.set_footer_text("No draft message was saved in this session.", 3)