36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
from app.api.v1.tv import _rewrite_hls_uri_attributes, _should_strip_hls_metadata_line
|
|
|
|
|
|
def test_rewrite_hls_uri_attributes_rewrites_subtitle_manifest_url():
|
|
line = '#EXT-X-MEDIA:TYPE=SUBTITLES,GROUP-ID="subs",NAME="English",URI="index_3_0.m3u8"'
|
|
|
|
rewritten = _rewrite_hls_uri_attributes(
|
|
line,
|
|
base_url="https://example.com/live/master.m3u8",
|
|
)
|
|
|
|
assert 'URI="/api/v1/tv/proxy?url=https%3A%2F%2Fexample.com%2Flive%2Findex_3_0.m3u8"' in rewritten
|
|
|
|
|
|
def test_rewrite_hls_uri_attributes_rewrites_absolute_uri():
|
|
line = '#EXT-X-I-FRAME-STREAM-INF:BANDWIDTH=1234,URI="https://cdn.example.com/live/iframe.m3u8"'
|
|
|
|
rewritten = _rewrite_hls_uri_attributes(
|
|
line,
|
|
base_url="https://example.com/live/master.m3u8",
|
|
)
|
|
|
|
assert 'URI="/api/v1/tv/proxy?url=https%3A%2F%2Fcdn.example.com%2Flive%2Fiframe.m3u8"' in rewritten
|
|
|
|
|
|
def test_strip_hls_subtitle_media_metadata():
|
|
line = '#EXT-X-MEDIA:TYPE=SUBTITLES,GROUP-ID="subs",NAME="English",URI="index_3_0.m3u8"'
|
|
|
|
assert _should_strip_hls_metadata_line(line) is True
|
|
|
|
|
|
def test_keep_hls_audio_media_metadata():
|
|
line = '#EXT-X-MEDIA:TYPE=AUDIO,GROUP-ID="audio",NAME="English",URI="audio.m3u8"'
|
|
|
|
assert _should_strip_hls_metadata_line(line) is False
|