model/boxes/buttons/views: Add get_stream_name_from_id method to model.

This commit introduces the get_stream_name_from_id method in
order to limit the direct accessing of stream_dict outside the
model. The commit also replaces the access of the name attribute
from stream_dict with the new method.

Tests added and updated.

Author: theViz343

tests/model/test_model.py

@@ -1819,6 +1819,36 @@ def test__get_all_stream_ids(
         model._never_subscribed_streams = never_subscribed_streams_fixture
         assert model._get_all_stream_ids() == expected_value
 
+    @pytest.mark.parametrize(
+        "stream_id, expected_stream_name",
+        [
+            case(
+                1000,
+                "Some general stream",
+                id="Subcribed stream",
+            ),
+            case(
+                3,
+                "Stream 3",
+                id="Unsubcribed stream",
+            ),
+            case(
+                5,
+                "Stream 5",
+                id="Never subcribed stream",
+            ),
+        ],
+    )
+    def test_get_stream_name_from_id(
+        self, mocker, model, stream_id, expected_stream_name
+    ):
+        mocker.patch(
+            MODEL + "._get_stream_from_id", return_value={"name": expected_stream_name}
+        )
+
+        assert model.get_stream_name_from_id(stream_id) == expected_stream_name
+        model._get_stream_from_id.assert_called_once()
+
     @pytest.mark.parametrize("muted", powerset([99, 1000]))
     @pytest.mark.parametrize("visual_notification_enabled", powerset([99, 1000]))
     def test__subscribe_to_streams(

tests/ui/test_ui_tools.py

@@ -891,6 +891,7 @@ def test_keypress_NEXT_UNREAD_TOPIC_stream(
 
         mid_col_view.model.stream_dict = {1: {"name": "stream"}}
         mid_col_view.model.next_unread_topic_from_message_id.return_value = (1, "topic")
+        mid_col_view.model.get_stream_name_from_id.return_value = "stream"
 
         return_value = mid_col_view.keypress(size, key)
 

tests/ui_tools/test_boxes.py

@@ -957,6 +957,9 @@ def test_generic_autocomplete_streams(
             for stream in stream_dict.values()
             if stream["name"] in stream_categories.get("muted", set())
         }
+        write_box.model.get_stream_name_from_id.side_effect = lambda x: stream_dict[x][
+            "name"
+        ]
         states = state_and_required_typeahead.keys()
         required_typeaheads = list(state_and_required_typeahead.values())
         typeahead_strings = [

tests/ui_tools/test_buttons.py

@@ -444,6 +444,8 @@ def test_init_calls_top_button(
         top_button = mocker.patch(MODULE + ".TopButton.__init__")
         params = dict(controller=controller, count=count)
 
+        controller.model.get_stream_name_from_id.return_value = stream_name
+
         topic_button = TopicButton(
             stream_id=stream_id, topic=title, view=view, **params
         )
@@ -922,6 +924,7 @@ def test__validate_narrow_link(
             "is_user_subscribed_to_stream",
             "is_valid_stream",
             "stream_id_from_name_return_value",
+            "get_stream_name_from_id_return_value",
             "expected_parsed_link",
             "expected_error",
         ],
@@ -933,6 +936,7 @@ def test__validate_narrow_link(
                 True,
                 None,
                 None,
+                "Stream 1",
                 ParsedNarrowLink(
                     stream=DecodedStream(stream_id=1, stream_name="Stream 1")
                 ),
@@ -945,6 +949,7 @@ def test__validate_narrow_link(
                 False,
                 None,
                 None,
+                None,
                 ParsedNarrowLink(stream=DecodedStream(stream_id=462, stream_name=None)),
                 "The stream seems to be either unknown or unsubscribed",
             ),
@@ -955,6 +960,7 @@ def test__validate_narrow_link(
                 None,
                 True,
                 1,
+                None,
                 ParsedNarrowLink(
                     stream=DecodedStream(stream_id=1, stream_name="Stream 1")
                 ),
@@ -967,6 +973,7 @@ def test__validate_narrow_link(
                 None,
                 False,
                 None,
+                "foo",
                 ParsedNarrowLink(
                     stream=DecodedStream(stream_id=None, stream_name="foo")
                 ),
@@ -987,6 +994,7 @@ def test__validate_and_patch_stream_data(
         is_user_subscribed_to_stream: Optional[bool],
         is_valid_stream: Optional[bool],
         stream_id_from_name_return_value: Optional[int],
+        get_stream_name_from_id_return_value: Optional[str],
         expected_parsed_link: ParsedNarrowLink,
         expected_error: str,
     ) -> None:
@@ -1000,6 +1008,10 @@ def test__validate_and_patch_stream_data(
         self.controller.model.is_valid_stream.return_value = is_valid_stream
         mocked_button = self.message_link_button()
 
+        mocked_button.model.get_stream_name_from_id.return_value = (
+            get_stream_name_from_id_return_value
+        )
+
         error = mocked_button._validate_and_patch_stream_data(parsed_link)
 
         assert parsed_link == expected_parsed_link

zulipterminal/model.py

@@ -890,7 +890,7 @@ def is_muted_topic(self, stream_id: int, topic: str) -> bool:
         """
         Returns True if topic is muted via muted_topics.
         """
-        stream_name = self.stream_dict[stream_id]["name"]
+        stream_name = self.get_stream_name_from_id(stream_id)
         topic_to_search = (stream_name, topic)
         return topic_to_search in self._muted_topics
 
@@ -1297,6 +1297,10 @@ def _get_all_stream_ids(self) -> List[int]:
         id_list.extend(stream_id for stream_id in self._never_subscribed_streams)
         return id_list
 
+    def get_stream_name_from_id(self, stream_id: int) -> str:
+        stream = self._get_stream_from_id(stream_id)
+        return stream["name"]
+
     def _subscribe_to_streams(self, subscriptions: List[Subscription]) -> None:
         def make_reduced_stream_data(stream: Subscription) -> StreamData:
             # stream_id has been changed to id.

zulipterminal/ui_tools/boxes.py

@@ -618,7 +618,7 @@ def autocomplete_streams(
         )
 
         muted_streams = [
-            self.model.stream_dict[stream_id]["name"]
+            self.model.get_stream_name_from_id(stream_id)
             for stream_id in self.model.muted_streams
         ]
         matching_muted_streams = [
@@ -637,9 +637,8 @@ def autocomplete_streams(
             else:
                 matched_streams.append(matching_muted_stream)
 
-        current_stream = self.model.stream_dict.get(self.stream_id, None)
-        if current_stream is not None:
-            current_stream_name = current_stream["name"]
+        if self.stream_id in self.model.stream_dict:
+            current_stream_name = self.model.get_stream_name_from_id(self.stream_id)
             if current_stream_name in matched_streams:
                 matched_streams.remove(current_stream_name)
                 matched_streams.insert(0, current_stream_name)

zulipterminal/ui_tools/buttons.py

@@ -306,7 +306,7 @@ def __init__(
         view: Any,
         count: int,
     ) -> None:
-        self.stream_name = controller.model.stream_dict[stream_id]["name"]
+        self.stream_name = controller.model.get_stream_name_from_id(stream_id)
         self.topic_name = topic
         self.stream_id = stream_id
         self.model = controller.model
@@ -574,7 +574,7 @@ def _validate_and_patch_stream_data(self, parsed_link: ParsedNarrowLink) -> str:
             stream_id = cast(int, model.stream_id_from_name(stream_name))
             parsed_link["stream"]["stream_id"] = stream_id
         else:
-            stream_name = cast(str, model.stream_dict[stream_id]["name"])
+            stream_name = cast(str, model.get_stream_name_from_id(stream_id))
             parsed_link["stream"]["stream_name"] = stream_name
 
         return ""

zulipterminal/ui_tools/views.py

@@ -578,8 +578,8 @@ def keypress(self, size: urwid_Size, key: str) -> Optional[str]:
             # For new streams with no previous conversation.
             if self.footer.focus is None:
                 stream_id = self.model.stream_id
-                stream_dict = self.model.stream_dict
-                self.footer.stream_box_view(caption=stream_dict[stream_id]["name"])
+                stream_name = self.model.get_stream_name_from_id(stream_id)
+                self.footer.stream_box_view(caption=stream_name)
             self.set_focus("footer")
             self.footer.focus_position = 0
             return key
@@ -609,7 +609,7 @@ def keypress(self, size: urwid_Size, key: str) -> Optional[str]:
 
             stream_id, topic = stream_topic
             self.controller.narrow_to_topic(
-                stream_name=self.model.stream_dict[stream_id]["name"],
+                stream_name=self.model.get_stream_name_from_id(stream_id),
                 topic_name=topic,
             )
             return key