@@ -1082,18 +1082,52 @@ def test_main_view_compact_output(
1082
1082
assert len (view_components ) == 1
1083
1083
assert isinstance (view_components [0 ], Padding )
1084
1084
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 ,
1087
1112
):
1088
1113
messages = messages_successful_response ["messages" ]
1089
1114
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
+ )
1091
1125
msg_box = MessageBox (message , self .model , message )
1092
1126
view_components = msg_box .main_view ()
1093
1127
1094
1128
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
1097
1131
1098
1132
@pytest .mark .parametrize (
1099
1133
"to_vary_in_last_message, update_required" ,
0 commit comments