IQ.Pilot Release Commit @ bec7652

artifacts/package_sources/teleoprtc/tests/test_track.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+
+import pytest
+
+import aiortc
+
+from teleoprtc.tracks import video_track_id, parse_video_track_id, TiciVideoStreamTrack, TiciTrackWrapper
+
+
+class TestTracks:
+  def test_track_id(self):
+    expected_camera_type, expected_track_id = "driver", "test"
+    track_id = video_track_id(expected_camera_type, expected_track_id)
+    camera_type, track_id = parse_video_track_id(track_id)
+    assert expected_camera_type == camera_type
+    assert expected_track_id == track_id
+
+  def test_track_id_invalid(self):
+    with pytest.raises(ValueError):
+      parse_video_track_id("test")
+
+  def test_tici_track_id(self):
+    class VideoStream(TiciVideoStreamTrack):
+      async def recv(self):
+        raise NotImplementedError()
+
+    track = VideoStream("driver", 0.1)
+    camera_type, _ = parse_video_track_id(track.id)
+    assert "driver" == camera_type
+
+  def test_tici_wrapper_id(self):
+    track = TiciTrackWrapper("driver", aiortc.mediastreams.VideoStreamTrack())
+    camera_type, _ = parse_video_track_id(track.id)
+    assert "driver" == camera_type