e2e test livekit-rtc (#541)

Author: davidzhao

.github/workflows/tests.yml

@@ -14,15 +14,22 @@ jobs:
     name: Run tests
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v6
         with:
           submodules: true
           lfs: true
 
-      - uses: actions/setup-python@v4
+      - uses: actions/setup-python@v6
+        with:
+          python-version: '3.13' 
       - name: Run tests
+        env:
+          LIVEKIT_URL: ${{ secrets.LIVEKIT_URL }}
+          LIVEKIT_API_KEY: ${{ secrets.LIVEKIT_API_KEY }}
+          LIVEKIT_API_SECRET: ${{ secrets.LIVEKIT_API_SECRET }}
         run: |
+          pip3 install -r dev-requirements.txt
           python3 ./livekit-rtc/rust-sdks/download_ffi.py --output livekit-rtc/livekit/rtc/resources
           pip3 install ./livekit-protocol ./livekit-api ./livekit-rtc
-          pip3 install -r dev-requirements.txt
           pytest . --ignore=livekit-rtc/rust-sdks
+        

livekit-rtc/livekit/rtc/room.py

@@ -27,6 +27,7 @@
 from ._proto import participant_pb2 as proto_participant
 from ._proto import room_pb2 as proto_room
 from ._proto import stats_pb2 as proto_stats
+from ._proto.participant_pb2 import DisconnectReason
 from ._proto.room_pb2 import ConnectionState
 from ._proto.track_pb2 import TrackKind
 from ._proto.rpc_pb2 import RpcMethodInvocationEvent
@@ -581,6 +582,14 @@ async def disconnect(self) -> None:
         await self._task
         FfiClient.instance.queue.unsubscribe(self._ffi_queue)
 
+        # we should manually flip the state, since the connection could have been torn down before
+        # the callbacks were processed
+        if self._connection_state != ConnectionState.CONN_DISCONNECTED:
+            self.local_participant._info.disconnect_reason = DisconnectReason.CLIENT_INITIATED
+            self._connection_state = ConnectionState.CONN_DISCONNECTED
+            self.emit("connection_state_changed", self._connection_state)
+            self.emit("disconnected", DisconnectReason.CLIENT_INITIATED)
+
     async def _listen_task(self) -> None:
         # listen to incoming room events
         while True:
@@ -853,8 +862,6 @@ def _on_room_event(self, event: proto_room.RoomEvent):
             connection_state = event.connection_state_changed.state
             self._connection_state = connection_state
             self.emit("connection_state_changed", connection_state)
-        elif which == "connected":
-            self.emit("connected")
         elif which == "disconnected":
             self.emit("disconnected", event.disconnected.reason)
         elif which == "reconnecting":

livekit-rtc/livekit/rtc/utils.py

@@ -1,9 +1,11 @@
 from __future__ import annotations
 
+from typing import AsyncIterator
+
 from .audio_frame import AudioFrame
 
 
-__all__ = ["combine_audio_frames"]
+__all__ = ["combine_audio_frames", "sine_wave_generator"]
 
 
 def combine_audio_frames(buffer: AudioFrame | list[AudioFrame]) -> AudioFrame:
@@ -83,3 +85,54 @@ def combine_audio_frames(buffer: AudioFrame | list[AudioFrame]) -> AudioFrame:
         num_channels=num_channels,
         samples_per_channel=total_samples_per_channel,
     )
+
+
+async def sine_wave_generator(
+    freq: float,
+    duration: float,
+    sample_rate: int = 48000,
+    amplitude: float = 0.3,
+) -> AsyncIterator[AudioFrame]:
+    """
+    Generate sine wave audio frames.
+
+    Useful for testing audio pipelines and generating test signals.
+
+    Args:
+        freq: Frequency of the sine wave in Hz.
+        duration: Duration of the audio in seconds.
+        sample_rate: Sample rate in Hz (default: 48000).
+        amplitude: Amplitude of the sine wave, range [0.0, 1.0] (default: 0.3).
+
+    Yields:
+        AudioFrame: Audio frames containing sine wave data.
+
+    Example:
+        >>> import asyncio
+        >>> async def generate_audio():
+        ...     async for frame in sine_wave_generator(440, 1.0):
+        ...         print(f"Generated frame with {frame.samples_per_channel} samples")
+        >>> asyncio.run(generate_audio())
+    """
+    try:
+        import numpy as np
+    except ImportError:
+        raise ImportError(
+            "numpy is required for sine_wave_generator. Install it with: pip install numpy"
+        )
+
+    blocksize = sample_rate // 10
+    total_frames = int((duration * sample_rate) // blocksize)
+    t_frame = np.arange(blocksize) / sample_rate
+
+    for i in range(total_frames):
+        t = t_frame + i * blocksize / sample_rate
+        signal = amplitude * np.sin(2 * np.pi * freq * t)
+        signal_int16 = np.int16(signal * 32767)
+        frame = AudioFrame(
+            signal_int16.tobytes(),
+            sample_rate,
+            1,
+            blocksize,
+        )
+        yield frame

livekit-rtc/pyproject.toml

@@ -22,4 +22,4 @@ manylinux-pypy_i686-image = "manylinux_2_28"
 manylinux-pypy_aarch64-image = "manylinux_2_28"
 
 [tool.cibuildwheel.linux]
-before-build = "pip install requests && python rust-sdks/download_ffi.py --output livekit/rtc/resources && yum install -y libX11-devel libXrandr-devel libXext-devel libXfixes-devel libXdamage-devel libXcomposite-devel libXi-devel mesa-libGL-devel || apt install -y xorg-dev"
+before-build = "pip install requests && python rust-sdks/download_ffi.py --output livekit/rtc/resources && yum install -y libX11-devel libXrandr-devel libXext-devel libXfixes-devel libXdamage-devel libXcomposite-devel libXi-devel mesa-libGL-devel || apt install -y xorg-dev"