disconnect のテスト追加

tests/client.py

@@ -45,6 +45,9 @@ def __init__(
         self._signaling_urls: list[str] = signaling_urls
         self._channel_id: str = channel_id
 
+        self._audio = audio
+        self._video = video
+
         self._audio_channels: int = audio_channels
         self._audio_sample_rate: int = audio_sample_rate
 
@@ -104,6 +107,14 @@ def __init__(
         self._connection.on_disconnect = self._on_disconnect
         self._connection.on_ws_close = self._on_ws_close
 
+    def __enter__(self):
+        print(f"self._audio: {self._audio}")
+        print(f"self._video: {self._video}")
+        return self.connect(fake_audio=bool(self._audio), fake_video=bool(self._video))
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.disconnect()
+
     def connect(self, fake_audio=False, fake_video=False) -> None:
         self._connection.connect()
 
@@ -120,7 +131,6 @@ def connect(self, fake_audio=False, fake_video=False) -> None:
         ), "Could not connect to Sora."
 
     def disconnect(self) -> None:
-        """Sora から切断します。"""
         self._connection.disconnect()
 
     def get_stats(self):

tests/test_disconnect.py

@@ -0,0 +1,42 @@
+import sys
+import uuid
+
+from client import Sendonly
+
+
+def test_sendonly_disconnect(setup):
+    signaling_urls = setup.get("signaling_urls")
+    channel_id_prefix = setup.get("channel_id_prefix")
+    metadata = setup.get("metadata")
+
+    channel_id = f"{channel_id_prefix}_{__name__}_{sys._getframe().f_code.co_name}_{uuid.uuid4()}"
+
+    with Sendonly(
+        signaling_urls,
+        channel_id,
+        audio=True,
+        video=True,
+        metadata=metadata,
+    ) as sendonly1:
+        with Sendonly(
+            signaling_urls,
+            channel_id,
+            audio=True,
+            video=True,
+            metadata=metadata,
+        ) as sendonly2:
+            with Sendonly(
+                signaling_urls,
+                channel_id,
+                audio=True,
+                video=True,
+                metadata=metadata,
+            ) as sendonly3:
+                with Sendonly(
+                    signaling_urls,
+                    channel_id,
+                    audio=True,
+                    video=True,
+                    metadata=metadata,
+                ) as sendonly4:
+                    pass