Merge branch 'srt'

Author: tbarbugli

getstream/video/__init__.py

@@ -1,8 +1,14 @@
+from dataclasses import dataclass
 from typing import Optional, Dict
 
 from getstream.models import CallResponse
 
 
+@dataclass
+class SRTCredentials:
+    address: str
+
+
 class BaseCall:
     def __init__(
         self, client, call_type: str, call_id: str = None, custom_data: Dict = None
@@ -42,7 +48,7 @@ def connect_openai(
         # Wrap the connection manager to check for errors in the first message
         return ConnectionManagerWrapper(connection_manager, self.call_type, self.id)
 
-    def create_srt_token(self, user_id: str) -> str:
+    def create_srt_credentials(self, user_id: str) -> SRTCredentials:
         if self._data is None:
             raise TypeError(
                 "call object is not initialized, make sure to call .get or .get_or_create first"
@@ -51,6 +57,8 @@ def create_srt_token(self, user_id: str) -> str:
         token = self.client.stream.create_token(user_id)
         passphrase = token.split(".")[2]
 
-        return self._data.ingress.srt.address.replace(
-            "{passphrase}", passphrase
-        ).replace("{token}", token)
+        return SRTCredentials(
+            address=self._data.ingress.srt.address.replace(
+                "{passphrase}", passphrase
+            ).replace("{token}", token)
+        )

tests/test_video_examples.py

@@ -13,6 +13,7 @@
     BackstageSettingsRequest,
     SessionSettingsRequest,
     FrameRecordingSettingsRequest,
+    MemberRequest,
 )
 from getstream.video.call import Call
 from datetime import datetime, timezone, timedelta
@@ -507,7 +508,7 @@ async def test_async_create_user(async_client: AsyncStream):
 async def test_srt(async_client: AsyncStream):
     user_id = str(uuid.uuid4())
 
-    call = async_client.video.call("default", str(uuid.uuid4()))
+    call = async_client.video.call("livestream", "eRfEfBI5UQxYRyIhR8BAa")
     response = await call.get_or_create(
         data=CallRequest(
             created_by_id=user_id,
@@ -519,5 +520,7 @@ async def test_srt(async_client: AsyncStream):
         )
     )
 
+    await call.update_call_members(update_members=[MemberRequest(user_id=user_id)])
+
     assert response.data.call.ingress.srt != ""
-    assert call.create_srt_token("tommaso-id") != ""
+    assert call.create_srt_credentials(user_id).address != ""