Add assign_roles support (#85)

Author: Anatoly Rugalev

stream_chat/async_chat/channel.py

@@ -147,16 +147,28 @@ async def truncate(self):
         """
         return await self.client.post(f"{self.url}/truncate")
 
-    async def add_members(self, user_ids, message=None):
+    async def add_members(self, members, message=None):
         """
         Adds members to the channel
 
-        :param user_ids: user IDs to add as members
+        :param members: member objects to add
         :param message: An optional to show
         :return:
         """
         return await self.client.post(
-            self.url, data={"add_members": user_ids, "message": message}
+            self.url, data={"add_members": members, "message": message}
+        )
+
+    async def assign_roles(self, members, message=None):
+        """
+        Assigns new roles to specified channel members
+
+        :param members: member objects with role information
+        :param message: An optional to show
+        :return:
+        """
+        return await self.client.post(
+            self.url, data={"assign_roles": members, "message": message}
         )
 
     async def invite_members(self, user_ids, message=None):

stream_chat/base/channel.py

@@ -148,11 +148,22 @@ def truncate(self):
         pass
 
     @abc.abstractmethod
-    def add_members(self, user_ids, message=None):
+    def add_members(self, members, message=None):
         """
         Adds members to the channel
 
-        :param user_ids: user IDs to add as members
+        :param members: member objects to add
+        :param message: An optional to show
+        :return:
+        """
+        pass
+
+    @abc.abstractmethod
+    def assign_roles(self, members, message=None):
+        """
+        Assigns new roles to specified channel members
+
+        :param members: member objects with role information
         :param message: An optional to show
         :return:
         """

stream_chat/channel.py

@@ -145,16 +145,28 @@ def truncate(self):
         """
         return self.client.post(f"{self.url}/truncate")
 
-    def add_members(self, user_ids, message=None):
+    def add_members(self, members, message=None):
         """
         Adds members to the channel
 
-        :param user_ids: user IDs to add as members
+        :param members: member objects to add
         :param message: An optional to show
         :return:
         """
         return self.client.post(
-            self.url, data={"add_members": user_ids, "message": message}
+            self.url, data={"add_members": members, "message": message}
+        )
+
+    def assign_roles(self, members, message=None):
+        """
+        Assigns new roles to specified channel members
+
+        :param members: member objects with role information
+        :param message: An optional to show
+        :return:
+        """
+        return self.client.post(
+            self.url, data={"assign_roles": members, "message": message}
         )
 
     def invite_members(self, user_ids, message=None):

stream_chat/tests/async_chat/test_channel.py

@@ -121,6 +121,18 @@ async def test_add_moderators(self, event_loop, channel, random_user):
         response = await channel.demote_moderators([random_user["id"]])
         assert not response["members"][0].get("is_moderator", False)
 
+    @pytest.mark.asyncio
+    async def test_assign_roles_moderators(self, event_loop, channel, random_user):
+        member = {"user_id": random_user["id"], "channel_role": "channel_moderator"}
+        response = await channel.add_members([member])
+        assert len(response["members"]) == 1
+        assert response["members"][0]["channel_role"] == "channel_moderator"
+
+        member["channel_role"] = "channel_member"
+        response = await channel.assign_roles([member])
+        assert len(response["members"]) == 1
+        assert response["members"][0]["channel_role"] == "channel_member"
+
     @pytest.mark.asyncio
     async def test_mark_read(self, event_loop, channel, random_user):
         response = await channel.mark_read(random_user["id"])

stream_chat/tests/test_channel.py

@@ -105,6 +105,17 @@ def test_add_moderators(self, channel, random_user):
         response = channel.demote_moderators([random_user["id"]])
         assert not response["members"][0].get("is_moderator", False)
 
+    def test_assign_roles_moderators(self, channel, random_user):
+        member = {"user_id": random_user["id"], "channel_role": "channel_moderator"}
+        response = channel.add_members([member])
+        assert len(response["members"]) == 1
+        assert response["members"][0]["channel_role"] == "channel_moderator"
+
+        member["channel_role"] = "channel_member"
+        response = channel.assign_roles([member])
+        assert len(response["members"]) == 1
+        assert response["members"][0]["channel_role"] == "channel_member"
+
     def test_mark_read(self, channel, random_user):
         response = channel.mark_read(random_user["id"])
         assert "event" in response