tests: helper: Test set_count for stream messages.

Add test for verifying unread_counts when marking streams as read/unread.

Added new index fixture that contains a large number of messages.

Author: kaustubh-nair

tests/conftest.py

@@ -4,6 +4,7 @@
 
 import pytest
 
+from zulipterminal.helper import UnreadCounts
 from zulipterminal.helper import initial_index as helper_initial_index
 from zulipterminal.ui_tools.boxes import MessageBox
 from zulipterminal.ui_tools.buttons import (
@@ -743,3 +744,37 @@ def classified_unread_counts():
             99: 1
         }
     }
+
+
+@pytest.fixture
+def initial_unread_counts():
+    return UnreadCounts(all_msg=0, all_pms=0,
+                        unread_topics=dict(), unread_pms=dict(),
+                        unread_huddles=dict(), streams=dict())
+
+
+@pytest.fixture
+def index_multiple_messages():
+    """
+    Index fixture filled with multiple entries for messages
+    """
+    return dict(helper_initial_index, **{'messages': {
+        1: {'id': 1, 'type': 'stream', 'subject': 'Topic 1',
+            'sender_id': 99, 'stream_id': 1001, 'display_recipient': []},
+        2: {'id': 2, 'type': 'stream', 'subject': 'Topic 1',
+            'sender_id': 100, 'stream_id': 1001, 'display_recipient': []},
+        3: {'id': 3, 'type': 'stream', 'subject': 'Topic 2',
+            'sender_id': 98, 'stream_id': 1002, 'display_recipient': []},
+        4: {'id': 4, 'type': 'stream', 'subject': 'Topic 3',
+            'sender_id': 101, 'stream_id': 1002, 'display_recipient': []},
+        5: {'id': 5, 'type': 'stream', 'subject': 'Topic 3',
+            'sender_id': 101, 'stream_id': 1002, 'display_recipient': []},
+        6: {'id': 6, 'type': 'stream', 'subject': 'Topic 1',
+            'sender_id': 101, 'stream_id': 1001, 'display_recipient': []},
+        7: {'id': 7, 'type': 'stream', 'subject': 'Topic 1',
+            'sender_id': 101, 'stream_id': 1001, 'display_recipient': []},
+        8: {'id': 8, 'type': 'stream', 'subject': 'Topic 100',
+            'sender_id': 101, 'stream_id': 1003, 'display_recipient': []},
+        9: {'id': 9, 'type': 'stream', 'subject': 'Topic 100',
+            'sender_id': 101, 'stream_id': 1003, 'display_recipient': []},
+        }})

tests/helper/test_helper.py

@@ -1,13 +1,55 @@
+from copy import deepcopy
 from typing import Any
 
 import pytest
 
 from zulipterminal.helper import (
     canonicalize_color, classify_unread_counts, index_messages, notify,
-    powerset,
+    powerset, set_count,
 )
 
 
+@pytest.mark.parametrize('all_msg, unread_topics', [
+       (6, {(1001, 'Topic 1'): 3, (1002, 'Topic 2'): 1, (1002, 'Topic 3'): 2}),
+       ])
+@pytest.mark.parametrize(
+        'new_count, id_list, expected_all_msg, expected_unread_topics', [
+         (1, [], 6, {(1001, 'Topic 1'): 3, (1002, 'Topic 2'): 1,
+                     (1002, 'Topic 3'): 2}),
+         (1, [7], 7, {(1001, 'Topic 1'): 4, (1002, 'Topic 2'): 1,
+                      (1002, 'Topic 3'): 2}),
+         (1, [7, 8, 9], 9, {(1001, 'Topic 1'): 4, (1002, 'Topic 2'): 1,
+                            (1002, 'Topic 3'): 2, (1003, 'Topic 100'): 2}),
+         (-1, [], 6, {(1001, 'Topic 1'): 3, (1002, 'Topic 2'): 1,
+                      (1002, 'Topic 3'): 2}),
+         (-1, [5], 5, {(1001, 'Topic 1'): 3, (1002, 'Topic 2'): 1,
+                       (1002, 'Topic 3'): 1}),
+         (-1, [1, 2, 4, 6], 2, {(1002, 'Topic 2'): 1, (1002, 'Topic 3'): 1}),
+         (-1, [1, 2, 3, 4, 5, 6], 0, {}),
+         ])
+def test_set_count_stream(mocker, initial_unread_counts,
+                          unread_topics, all_msg, id_list, new_count,
+                          expected_unread_topics, expected_all_msg,
+                          index_multiple_messages):
+    controller = mocker.patch('zulipterminal.core.Controller.__init__')
+    controller.model.is_muted_stream = mocker.patch(
+            'zulipterminal.model.Model.is_muted_stream', return_value=False)
+
+    unread_counts = deepcopy(dict(initial_unread_counts,
+                                  **{'all_msg': all_msg,
+                                     'unread_topics': unread_topics}))
+
+    controller.model.index = index_multiple_messages
+    controller.model.unread_counts = unread_counts
+    set_count(id_list, controller, new_count)
+
+    expected_unread_counts = dict(initial_unread_counts,
+                                  **{'all_msg': expected_all_msg,
+                                     'unread_topics': expected_unread_topics})
+
+    assert controller.model.unread_counts == expected_unread_counts
+
+
 def test_index_messages_narrow_all_messages(mocker,
                                             messages_successful_response,
                                             index_all_messages,