feat: adding tests

Author: totalimmersion

stream_chat/tests/async_chat/test_channel.py

@@ -380,3 +380,59 @@ async def test_export_channel(
                 assert "error" not in resp
                 break
             time.sleep(0.5)
+
+    async def test_pin_channel(
+        self, client: StreamChatAsync, channel: Channel, random_users: List[Dict]
+    ):
+        user_id = random_users[0]["id"]
+        await channel.add_members([user_id])
+
+        # Pin the channel
+        response = await channel.pin(user_id)
+        assert response is not None
+
+        # Query for pinned channels
+        response = await client.query_channels(
+            {"pinned": True, "cid": channel.cid}, user_id=user_id
+        )
+        assert len(response["channels"]) == 1
+        assert response["channels"][0]["channel"]["cid"] == channel.cid
+
+        # Unpin the channel
+        response = await channel.unpin(user_id)
+        assert response is not None
+
+        # Query for pinned channels
+        response = await client.query_channels(
+            {"pinned": False, "cid": channel.cid}, user_id=user_id
+        )
+        assert len(response["channels"]) == 1
+        assert response["channels"][0]["channel"]["cid"] == channel.cid
+
+    async def test_archive_channel(
+        self, client: StreamChatAsync, channel: Channel, random_users: List[Dict]
+    ):
+        user_id = random_users[0]["id"]
+        await channel.add_members([user_id])
+
+        # Archive the channel
+        response = await channel.archive(user_id)
+        assert response is not None
+
+        # Query for archived channels
+        response = await client.query_channels(
+            {"archived": True, "cid": channel.cid}, user_id=user_id
+        )
+        assert len(response["channels"]) == 1
+        assert response["channels"][0]["channel"]["cid"] == channel.cid
+
+        # Unarchive the channel
+        response = await channel.unarchive(user_id)
+        assert response is not None
+
+        # Query for archived channels
+        response = await client.query_channels(
+            {"archived": False, "cid": channel.cid}, user_id=user_id
+        )
+        assert len(response["channels"]) == 1
+        assert response["channels"][0]["channel"]["cid"] == channel.cid

stream_chat/tests/test_channel.py

@@ -392,7 +392,6 @@ def test_pin_channel(
         response = client.query_channels(
             {"pinned": True, "cid": channel.cid}, user_id=user_id
         )
-        print(response)
         assert len(response["channels"]) == 1
         assert response["channels"][0]["channel"]["cid"] == channel.cid
 
@@ -406,3 +405,31 @@ def test_pin_channel(
         )
         assert len(response["channels"]) == 1
         assert response["channels"][0]["channel"]["cid"] == channel.cid
+
+    def test_archive_channel(
+        self, client: StreamChat, channel: Channel, random_users: List[Dict]
+    ):
+        user_id = random_users[0]["id"]
+        channel.add_members([user_id])
+
+        # Archive the channel
+        response = channel.archive(user_id)
+        assert response is not None
+
+        # Query for archived channels
+        response = client.query_channels(
+            {"archived": True, "cid": channel.cid}, user_id=user_id
+        )
+        assert len(response["channels"]) == 1
+        assert response["channels"][0]["channel"]["cid"] == channel.cid
+
+        # Unarchive the channel
+        response = channel.unarchive(user_id)
+        assert response is not None
+
+        # Query for archhived channels
+        response = client.query_channels(
+            {"archived": False, "cid": channel.cid}, user_id=user_id
+        )
+        assert len(response["channels"]) == 1
+        assert response["channels"][0]["channel"]["cid"] == channel.cid