Fix broken tests

Author: kaustubh-nair

tests/conftest.py

@@ -480,7 +480,8 @@ def empty_index():
             stream_msg_template['id']: stream_msg_template,
             pm_template['id']: pm_template,
             group_pm_template['id']: group_pm_template,
-        })
+        }),
+        'unread_msgs': set()
     })
 
 

zulipterminal/helper.py

@@ -37,8 +37,8 @@
     'search': Set[int],  # {message_id, ...}
     # Downloaded message data
     'messages': Dict[int, Message],  # message_id: Message
-    # unread message data
-    'unread_msgs': Set[int]  # message_ids
+    # unread message data; additional data in model.initial_data['unread_msgs']
+    'unread_msgs': Set[int]  # {message_ids, ...}
 })
 
 initial_index = Index(
@@ -369,6 +369,7 @@ def index_unread_messsages(unread_msgs: Dict[str, Any], index: Index) -> Index:
 
     return index
 
+
 def classify_unread_counts(model: Any) -> UnreadCounts:
     # TODO: support group pms
     unread_msg_counts = model.initial_data['unread_msgs']

zulipterminal/model.py

@@ -448,7 +448,9 @@ def _update_initial_data(self) -> None:
             self.user_email = self.initial_data['email']
             self.user_full_name = self.initial_data['full_name']
             self.server_name = self.initial_data['realm_name']
-            self.index = index_unread_messsages(self.initial_data['unread_msgs'], self.index)
+            unread_msgs = self.initial_data['unread_msgs']
+            self.index = index_unread_messsages(unread_msgs,
+                                                self.index)
         else:
             failures = [name for name, result in results.items() if not result]
             raise ServerConnectionFailure(", ".join(failures))