helper: Modify set_counts to use index['unread_msgs'].

Per-message 'sender_id' is not available for stream messages. Due to
this, messages sent by user might cause undesirable behaviours.

Author: kaustubh-nair

tests/model/test_model.py

@@ -1277,7 +1277,7 @@ def test_update_read_status(self, mocker, model, event_op,
                     flags_before)
 
         if event_op == 'add':
-            set_count.assert_called_once_with(list(changed_ids),
+            set_count.assert_called_once_with(list(event_message_ids),
                                               self.controller, -1)
         elif event_op == 'remove':
             set_count.assert_not_called()

zulipterminal/helper.py

@@ -97,9 +97,8 @@ def _set_count_in_model(new_count: int, changed_messages: List[Message],
         if message['type'] == 'stream':
             key = (message['stream_id'], message['subject'])
             unreads = unread_counts['unread_topics']
-        # self-pm has only one display_recipient
-        # 1-1 pms have 2 display_recipient
-        elif len(message['display_recipient']) <= 2:
+        # 1-1 pms and self-pms
+        elif message['type'] == 'private' and 'sender_id' in message:
             key = message['sender_id']
             unreads = unread_counts['unread_pms']  # type: ignore
         else:  # If it's a group pm
@@ -136,13 +135,16 @@ def _set_count_in_view(controller: Any, new_count: int,
     all_msg = controller.view.home_button
     all_pm = controller.view.pm_button
     for message in changed_messages:
-        user_id = message['sender_id']
-
-        # If we sent this message, don't increase the count
-        if user_id == controller.model.user_id:
-            continue
 
         msg_type = message['type']
+        # FIXME no user_id for streams?
+        if msg_type != 'stream':
+            user_id = message['sender_id']
+
+            # If we sent this message, don't increase the count
+            if user_id == controller.model.user_id:
+                continue
+
         add_to_counts = True
         if msg_type == 'stream':
             stream_id = message['stream_id']
@@ -184,7 +186,7 @@ def set_count(id_list: List[int], controller: Any, new_count: int) -> None:
     # This method applies new_count for 'new message' (1) or 'read' (-1)
     # (we could ensure this in a different way by a different type)
     assert new_count == 1 or new_count == -1
-    messages = controller.model.index['messages']
+    messages = controller.model.index['unread_msgs']
     unread_counts = controller.model.unread_counts  # type: UnreadCounts
     changed_messages = [messages[id] for id in id_list]
     _set_count_in_model(new_count, changed_messages, unread_counts)

zulipterminal/model.py

@@ -840,8 +840,7 @@ def update_message_flag_status(self, event: Event) -> None:
             self.update_rendered_view(message_id)
 
         if event['operation'] == 'add' and flag_to_change == 'read':
-            set_count(list(message_ids_to_mark & indexed_message_ids),
-                      self.controller, -1)
+            set_count(list(message_ids_to_mark), self.controller, -1)
 
     def update_rendered_view(self, msg_id: int) -> None:
         # Update new content in the rendered view