Add tests

Author: kaustubh-nair

tests/conftest.py

@@ -481,7 +481,7 @@ def empty_index():
             pm_template['id']: pm_template,
             group_pm_template['id']: group_pm_template,
         }),
-        'unread_msgs': set()
+        'unread_msg_ids': set()
     })
 
 
@@ -526,6 +526,11 @@ def index_user(empty_index):
     return dict(empty_index, **diff)
 
 
+@pytest.fixture
+def index_unread_message_ids():
+    return {1, 2, 3, 4, 5, 6, 7, 11, 12, 13, 101, 102}
+
+
 @pytest.fixture
 def index_user_multiple(empty_index):
     """

tests/helper/test_helper.py

@@ -179,14 +179,17 @@ def test_powerset(iterable, map_func, expected_powerset):
 )
 def test_classify_unread_counts(mocker, initial_data, stream_dict,
                                 classified_unread_counts, muted_topics,
-                                muted_streams, vary_in_unreads):
+                                muted_streams, vary_in_unreads, initial_index,
+                                index_unread_message_ids):
     model = mocker.Mock()
     model.stream_dict = stream_dict
     model.initial_data = initial_data
     model.muted_topics = muted_topics
     model.muted_streams = muted_streams
-    assert classify_unread_counts(model) == dict(classified_unread_counts,
-                                                 **vary_in_unreads)
+    model.index = initial_index
+    assert classify_unread_counts(model) == dict(
+            classified_unread_counts, **vary_in_unreads)
+    assert model.index['unread_msg_ids'] == index_unread_message_ids
 
 
 @pytest.mark.parametrize('color', [