Update index when marking message as read

Author: kaustubh-nair

tests/model/test_model.py

@@ -563,6 +563,7 @@ def test_toggle_message_star_status(self, mocker, model, flags_before,
         model.client.update_message_flags.assert_called_once_with(request)
 
     def test_mark_message_ids_as_read(self, model, mocker: Any) -> None:
+
         mock_api_query = mocker.patch('zulipterminal.core.Controller'
                                       '.client.update_message_flags')
 
@@ -572,6 +573,14 @@ def test_mark_message_ids_as_read(self, model, mocker: Any) -> None:
             {'flag': 'read', 'messages': [1, 2], 'op': 'add'},
         )
 
+    def test_mark_message_ids_as_read_index_updated(self, model,
+                                                    classified_unread_counts,
+                                                    mocker: Any) -> None:
+        model.index['unread_msg_ids'] = classified_unread_counts[1]
+        model.mark_message_ids_as_read([1, 2])
+        expected_msg_ids = {3, 4, 5, 6, 7, 11, 12, 13, 101, 102}
+        assert model.index['unread_msg_ids'] == expected_msg_ids
+
     def test_mark_message_ids_as_read_empty_msg_list(self, model) -> None:
         assert model.mark_message_ids_as_read([]) is None
 

zulipterminal/helper.py

@@ -357,7 +357,7 @@ def index_messages(messages: List[Message],
 def classify_unread_counts(model: Any) -> Tuple[UnreadCounts, Set[int]]:
     # TODO: support group pms
     unread_msg_counts = model.initial_data['unread_msgs']
-    unread_msg_ids: Set[int] = set()
+    unread_msg_ids = set()  # type: Set[int]
 
     unread_counts = UnreadCounts(
         all_msg=0,

zulipterminal/model.py

@@ -281,6 +281,7 @@ def toggle_message_star_status(self, message: Message) -> None:
     def mark_message_ids_as_read(self, id_list: List[int]) -> None:
         if not id_list:
             return
+        self.index['unread_msg_ids'].difference_update(id_list)
         self.client.update_message_flags({
             'messages': id_list,
             'flag': 'read',