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

@@ -5,6 +5,7 @@
 import pytest
 
 from zulipterminal.config.keys import keys_for_command
+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 (
@@ -758,6 +759,39 @@ def classified_unread_counts():
         }
     }
 
+
+@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': 'stream 1'},
+        2: {'id': 2, 'type': 'stream', 'subject': 'Topic 1', 'sender_id': 100,
+            'stream_id': 1001, 'display_recipient': 'stream 1'},
+        3: {'id': 3, 'type': 'stream', 'subject': 'Topic 2', 'sender_id': 98,
+            'stream_id': 1002, 'display_recipient': 'stream 2'},
+        4: {'id': 4, 'type': 'stream', 'subject': 'Topic 3', 'sender_id': 101,
+            'stream_id': 1002, 'display_recipient': 'stream 3'},
+        5: {'id': 5, 'type': 'stream', 'subject': 'Topic 3', 'sender_id': 101,
+            'stream_id': 1002, 'display_recipient': 'stream 3'},
+        6: {'id': 6, 'type': 'stream', 'subject': 'Topic 1', 'sender_id': 101,
+            'stream_id': 1001, 'display_recipient': 'stream 1'},
+        7: {'id': 7, 'type': 'stream', 'subject': 'Topic 1', 'sender_id': 101,
+            'stream_id': 1001, 'display_recipient': 'stream 1'},
+        8: {'id': 8, 'type': 'stream', 'subject': 'Topic 10', 'sender_id': 101,
+            'stream_id': 1003, 'display_recipient': 'stream 10'},
+        9: {'id': 9, 'type': 'stream', 'subject': 'Topic 10', 'sender_id': 101,
+            'stream_id': 1003, 'display_recipient': 'stream 10'},
+        }})
 # --------------- UI Fixtures -----------------------------------------
 
 

tests/helper/test_helper.py

@@ -1,13 +1,52 @@
+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 10'): 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.index = index_multiple_messages
+    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}))
+    expected_unread_counts = dict(initial_unread_counts,
+                                  **{'all_msg': expected_all_msg,
+                                     'unread_topics': expected_unread_topics})
+    controller.model.unread_counts = unread_counts
+    set_count(id_list, controller, new_count)
+    assert controller.model.unread_counts == expected_unread_counts
+
+
 def test_index_messages_narrow_all_messages(mocker,
                                             messages_successful_response,
                                             index_all_messages,