35 lines
1.0 KiB
Python
Executable File
35 lines
1.0 KiB
Python
Executable File
#!/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
|