[CHA-662] - Support for team based roles (#194)

* added tests for team based roles

* Update stream_chat/tests/test_client.py

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

* Update stream_chat/tests/test_client.py

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

---------

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

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,19 @@ 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