5
5
import pytest
6
6
from pytest import param as case
7
7
from pytest_mock import MockerFixture
8
- from urwid import Widget
8
+ from urwid import Text , Widget
9
9
from urwid_readline import ReadlineEdit
10
10
11
11
from zulipterminal .api_types import (
@@ -1704,7 +1704,10 @@ def test_keypress_CYCLE_COMPOSE_FOCUS(
1704
1704
else :
1705
1705
write_box .stream_box_view (stream_id )
1706
1706
else :
1707
- write_box .private_box_view ()
1707
+ if message_being_edited :
1708
+ write_box .private_box_edit_view ()
1709
+ else :
1710
+ write_box .private_box_view ()
1708
1711
size = widget_size (write_box )
1709
1712
1710
1713
def focus_val (x : str ) -> int :
@@ -1730,6 +1733,7 @@ def focus_val(x: str) -> int:
1730
1733
expected_focus_col_name
1731
1734
)
1732
1735
1736
+ @pytest .mark .parametrize ("message_being_edited" , [(True ), (False )])
1733
1737
@pytest .mark .parametrize (
1734
1738
"recipient_ids, expected_recipient_text" ,
1735
1739
[
@@ -1749,14 +1753,19 @@ def test_private_box_recipient_editing(
1749
1753
user_id_email_dict : Dict [int , str ],
1750
1754
recipient_ids : List [int ],
1751
1755
expected_recipient_text : str ,
1756
+ message_being_edited : bool ,
1752
1757
) -> None :
1753
1758
write_box .model .user_id_email_dict = user_id_email_dict
1754
1759
write_box .model .user_dict = user_dict
1755
1760
mocker .patch ("urwid.connect_signal" )
1756
1761
1757
- write_box .private_box_view (recipient_user_ids = recipient_ids )
1762
+ if message_being_edited :
1763
+ write_box .private_box_edit_view (recipient_user_ids = recipient_ids )
1764
+ assert isinstance (write_box .to_write_box , Text )
1765
+ else :
1766
+ write_box .private_box_view (recipient_user_ids = recipient_ids )
1767
+ assert isinstance (write_box .to_write_box , ReadlineEdit )
1758
1768
1759
- assert isinstance (write_box .to_write_box , ReadlineEdit )
1760
1769
assert write_box .to_write_box .text == expected_recipient_text
1761
1770
1762
1771
@pytest .mark .parametrize ("key" , keys_for_command ("MARKDOWN_HELP" ))
0 commit comments