model: Update index['unread_msgs'] when appending messages.

Entries are added to unread message datastructure when new messsages are
added, so that unread count can be updated.

Tests amended.

Author: kaustubh-nair

tests/model/test_model.py

@@ -650,7 +650,7 @@ def test_append_message_with_Falsey_log(self, mocker, model,
         model.found_newest = True
         mocker.patch('zulipterminal.model.Model.update_topic_index')
         index_msg = mocker.patch('zulipterminal.model.index_messages',
-                                 return_value={})
+                                 return_value=initial_index)
         model.msg_list = mocker.Mock()
         create_msg_box_list = mocker.patch('zulipterminal.model.'
                                            'create_msg_box_list',
@@ -672,7 +672,7 @@ def test_append_message_with_valid_log(self, mocker, model,
         model.found_newest = True
         mocker.patch('zulipterminal.model.Model.update_topic_index')
         index_msg = mocker.patch('zulipterminal.model.index_messages',
-                                 return_value={})
+                                 return_value=initial_index)
         model.msg_list = mocker.Mock()
         create_msg_box_list = mocker.patch('zulipterminal.model.'
                                            'create_msg_box_list',
@@ -691,11 +691,36 @@ def test_append_message_with_valid_log(self, mocker, model,
          assert_called_once_with(model, [message_fixture['id']],
                                  last_message=expected_last_msg))
 
+    @pytest.mark.parameter('unread_msgs', [
+        ({'type': 'stream', 'stream_id': 5140,
+          'subject': 'Test', 'display_recipient': 'PTEST'}),
+        ({'type': 'private', 'sender_id': '5140'}),
+        ({'type': 'private', 'display_recipient':
+            [{
+                'id': 5179,
+                'is_mirror_dummy': False,
+                'full_name': 'Boo Boo',
+                'short_name': 'boo',
+                'email': 'boo@zulip.com',
+            }, {
+                'short_name': 'foo',
+                'id': 5140,
+                'is_mirror_dummy': False,
+                'full_name': 'Foo Foo',
+                'email': 'foo@zulip.com',
+            }, {
+                'short_name': 'bar',
+                'id': 5180,
+                'is_mirror_dummy': False,
+                'full_name': 'Bar Bar',
+                'email': 'bar@zulip.com',
+            }]}),
+        ])
     def test_append_message_event_flags(self, mocker, model, message_fixture):
         model.found_newest = True
         mocker.patch('zulipterminal.model.Model.update_topic_index')
         index_msg = mocker.patch('zulipterminal.model.index_messages',
-                                 return_value={})
+                                 return_value=initial_index)
         model.msg_list = mocker.Mock()
         create_msg_box_list = mocker.patch('zulipterminal.model.'
                                            'create_msg_box_list',
@@ -721,8 +746,9 @@ def test_append_message_event_flags(self, mocker, model, message_fixture):
 
     @pytest.mark.parametrize('response, narrow, recipients, log', [
         ({'type': 'stream', 'stream_id': 1, 'subject': 'FOO',
-          'id': 1}, [], frozenset(), ['msg_w']),
-        ({'type': 'private', 'id': 1},
+          'id': 1, 'display_recipient': 'a'}, [], frozenset(), ['msg_w']),
+        ({'type': 'private', 'id': 1, 'sender_id': 1,
+          'display_recipient': []},
          [['is', 'private']], frozenset(), ['msg_w']),
         ({'type': 'stream', 'id': 1, 'stream_id': 1, 'subject': 'FOO',
           'display_recipient': 'a'},
@@ -735,14 +761,15 @@ def test_append_message_event_flags(self, mocker, model, message_fixture):
           'display_recipient': 'a'},
          [['stream', 'c'], ['topic', 'b']],
          frozenset(), []),
-        ({'type': 'private', 'id': 1,
+        ({'type': 'private', 'id': 1, 'sender_id': 5827,
           'display_recipient': [{'id': 5827}, {'id': 5}]},
          [['pm_with', 'notification-bot@zulip.com']],
          frozenset({5827, 5}), ['msg_w']),
-        ({'type': 'private', 'id': 1},
+        ({'type': 'private', 'id': 1, 'sender_id': 5827,
+          'display_recipient': []},
          [['is', 'search']],
          frozenset(), []),
-        ({'type': 'private', 'id': 1,
+        ({'type': 'private', 'id': 1, 'sender_id': 5827,
           'display_recipient': [{'id': 5827}, {'id': 3212}]},
          [['pm_with', 'notification-bot@zulip.com']],
          frozenset({5827, 5}), []),
@@ -756,7 +783,7 @@ def test_append_message(self, mocker, user_profile, response,
         model.found_newest = True
         mocker.patch('zulipterminal.model.Model.update_topic_index')
         index_msg = mocker.patch('zulipterminal.model.index_messages',
-                                 return_value={})
+                                 return_value=initial_index)
         create_msg_box_list = mocker.patch('zulipterminal.model.'
                                            'create_msg_box_list',
                                            return_value=["msg_w"])

zulipterminal/model.py

@@ -746,6 +746,23 @@ def append_message(self, event: Event) -> None:
                 if narrow_recipients == message_recipients:
                     self.msg_list.log.append(msg_w)
             if 'read' not in response['flags']:
+                if response['type'] == 'stream':
+                    unread_data = {'type': 'stream', 'display_recipient':
+                                   response['display_recipient'],
+                                   'stream_id': response['stream_id'],
+                                   'subject': response['subject']}
+                elif len(response['display_recipient']) <= 2:
+                    # pm and self-pm
+                    unread_data = {'type': 'private',
+                                   'sender_id': response['sender_id']}
+                else:
+                    # huddles
+                    message_recipients = frozenset(
+                        [user['id'] for user in response['display_recipient']])
+                    unread_data = {'type': 'private',
+                                   'display_recipient': message_recipients}
+                self.index['unread_msgs'].update(
+                                {int(response['id']): unread_data})
                 set_count([response['id']], self.controller, 1)
             self.controller.update_screen()