model: Add muted_users to intial_data_to_fetch.

The muted users are stored in muted_users list of Model class.
Tests adapted.

Co-authored by: Subhasish-Behera <greendoor3000@gmail.com>

Author: rsashank

tests/conftest.py

@@ -903,6 +903,7 @@ def initial_data(
             }
         ],
         "result": "success",
+        "muted_users": {},
         "queue_id": "1522420755:786",
         "realm_users": users_fixture,
         "cross_realm_bots": [

tests/model/test_model.py

@@ -190,6 +190,36 @@ def test_init_muted_topics(
 
         assert model._muted_topics == locally_processed_data
 
+    @pytest.mark.parametrize(
+        "server_response, ids, zulip_feature_level",
+        [
+            (
+                [
+                    {"id": 32323, "timestamp": 1726810359},
+                    {"id": 37372, "timestamp": 214214214},
+                ],
+                {32323, 37372},
+                48,
+            ),
+            ([], set(), 0),
+        ],
+        ids=[
+            "zulip_feature_level:48",
+            "zulip_feature_level:0",
+        ],
+    )
+    def test_init_muted_users(
+        self, mocker, initial_data, server_response, ids, zulip_feature_level
+    ):
+        mocker.patch(MODEL + ".get_messages", return_value="")
+        initial_data["zulip_feature_level"] = zulip_feature_level
+        initial_data["muted_users"] = server_response
+        self.client.register = mocker.Mock(return_value=initial_data)
+
+        model = Model(self.controller)
+
+        assert model._muted_users == ids
+
     def test_init_InvalidAPIKey_response(self, mocker, initial_data):
         # Both network calls indicate the same response
         mocker.patch(MODEL + ".get_messages", return_value="Invalid API key")
@@ -262,6 +292,7 @@ def test_register_initial_desired_events(self, mocker, initial_data):
             "realm_emoji",
             "custom_profile_fields",
             "zulip_version",
+            "muted_users",
         ]
         model.client.register.assert_called_once_with(
             event_types=event_types,

zulipterminal/model.py

@@ -145,6 +145,7 @@ def __init__(self, controller: Any) -> None:
             # zulip_version and zulip_feature_level are always returned in
             # POST /register from Feature level 3.
             "zulip_version",
+            "muted_users",
         ]
 
         # Events desired with their corresponding callback
@@ -208,6 +209,11 @@ def __init__(self, controller: Any) -> None:
             )
             for stream_name, topic, *date_muted in muted_topics
         }
+        # NOTE: muted_users also contains timestamps, but we only store the user IDs
+        # muted_users was added in ZFL 48, Zulip 4.0
+        self._muted_users: Set[int] = set()
+        if self.server_feature_level >= 48:
+            self._update_muted_users(self.initial_data["muted_users"])
 
         groups = self.initial_data["realm_user_groups"]
         self.user_group_by_id: Dict[int, Dict[str, Any]] = {}
@@ -1204,6 +1210,9 @@ def get_user_info(self, user_id: int) -> Optional[TidiedUserInfo]:
 
         return user_info
 
+    def _update_muted_users(self, muted_users: List[Dict[int, int]]) -> None:
+        self._muted_users = {muted_user["id"] for muted_user in muted_users}
+
     def _update_users_data_from_initial_data(self) -> None:
         # Dict which stores the active/idle status of users (by email)
         presences = self.initial_data["presences"]