added tests for team based roles

Author: totalimmersion

stream_chat/tests/test_client.py

@@ -143,6 +143,18 @@ def test_update_user(self, client: StreamChat):
         assert "users" in response
         assert user["id"] in response["users"]
 
+    def test_update_user_with_team(self, client: StreamChat):
+        user = {
+            "id": str(uuid.uuid4()),
+            "team": "blue",
+            "teams_role": {"blue": "admin"}
+        }
+        response = client.upsert_user(user)
+        assert "users" in response
+        assert user["id"] in response["users"]
+        assert response["users"][user["id"]]["team"] == "blue"
+        assert response["users"][user["id"]]["teams_role"]["blue"] == "admin"
+
     def test_update_users(self, client: StreamChat):
         user = {"id": str(uuid.uuid4())}
         response = client.upsert_users([user])
@@ -161,6 +173,23 @@ def test_update_user_partial(self, client: StreamChat):
         assert user_id in response["users"]
         assert response["users"][user_id]["field"] == "updated"
 
+    def test_update_user_partial_with_team(self, client: StreamChat):
+        user_id = str(uuid.uuid4())
+        client.upsert_user({"id": user_id, "name": "Test User"})
+
+        response = client.update_user_partial({
+            "id": user_id,
+            "set": {
+                "team": "blue",
+                "teams_role": {"blue": "admin"}
+            }
+        })
+
+        assert "users" in response
+        assert user_id in response["users"]
+        assert response["users"][user_id]["team"] == "blue"
+        assert response["users"][user_id]["teams_role"]["blue"] == "admin"
+
     def test_delete_user(self, client: StreamChat, random_user: Dict):
         response = client.delete_user(random_user["id"])
         assert "user" in response