Skip to content

Commit dff8de8

Browse files
author
Subhasish-Behera
committed
test_messages: Add and modify test for message labels.
Tests the conditions for moved and edited label along wth UI differences. Fixes #1253
1 parent e74353b commit dff8de8

File tree

1 file changed

+39
-5
lines changed

1 file changed

+39
-5
lines changed

tests/ui_tools/test_messages.py

Lines changed: 39 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1082,18 +1082,52 @@ def test_main_view_compact_output(
10821082
assert len(view_components) == 1
10831083
assert isinstance(view_components[0], Padding)
10841084

1085-
def test_main_view_generates_EDITED_label(
1086-
self, mocker, messages_successful_response
1085+
@pytest.mark.parametrize(
1086+
[
1087+
"message_edited",
1088+
"message_moved",
1089+
"expected_left_padding",
1090+
"expected_label_text",
1091+
],
1092+
[
1093+
case(True, False, 7, "EDITED", id="message_indexed_to_edited_messages"),
1094+
case(False, True, 6, "MOVED", id="message_indexed_to_moved_messages"),
1095+
case(
1096+
False,
1097+
False,
1098+
0,
1099+
"EDITED",
1100+
id="message_neither_index_to_edited_messages_or_moved_messages",
1101+
),
1102+
],
1103+
)
1104+
def test_main_view_generates_EDITED_or_MOVED_label(
1105+
self,
1106+
mocker,
1107+
messages_successful_response,
1108+
message_edited,
1109+
message_moved,
1110+
expected_left_padding,
1111+
expected_label_text,
10871112
):
10881113
messages = messages_successful_response["messages"]
10891114
for message in messages:
1090-
self.model.index["edited_messages"].add(message["id"])
1115+
msg_id = message["id"]
1116+
message_index = {
1117+
"edited_messages": {msg_id} if message_edited else set(),
1118+
"moved_messages": {msg_id} if message_moved else set(),
1119+
}
1120+
self.model.index = dict(
1121+
self.model.index,
1122+
edited_messages=message_index["edited_messages"],
1123+
moved_messages=message_index["moved_messages"],
1124+
)
10911125
msg_box = MessageBox(message, self.model, message)
10921126
view_components = msg_box.main_view()
10931127

10941128
label = view_components[0].original_widget.contents[0]
1095-
assert label[0].text == "EDITED"
1096-
assert label[1][1] == 7
1129+
assert label[0].text == expected_label_text
1130+
assert label[1][1] == expected_left_padding
10971131

10981132
@pytest.mark.parametrize(
10991133
"to_vary_in_last_message, update_required",

0 commit comments

Comments
 (0)