mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 12:05:07 +00:00
imSp4rky
59a7c76b2f
Regex-based UrlPattern substitution (handles {Bitrate}/{start_time} spellings), AudioTag/FourCC codec fallback, Width/Height fallbacks, integer fragment times, live-manifest and missing-language guards. Extract fps from the HEVC SPS VUI (HEVC-only; AVC timing untrusted).
44 lines
1.9 KiB
Python
44 lines
1.9 KiB
Python
import pytest
|
|
|
|
from unshackle.core.manifests.ism import ISM
|
|
from unshackle.core.tracks import Audio
|
|
|
|
MANIFEST = """<?xml version="1.0"?>
|
|
<SmoothStreamingMedia MajorVersion="2" MinorVersion="0" Duration="60000000" TimeScale="10000000">
|
|
<StreamIndex Type="audio" Name="audio" Language="en"
|
|
Url="QualityLevels({Bitrate})/Fragments(audio={start_time})">
|
|
<QualityLevel Index="0" Bitrate="128000" AudioTag="255" SamplingRate="48000"
|
|
Channels="2" BitsPerSample="16" CodecPrivateData="1190"/>
|
|
<c t="0" d="20000000" r="2"/>
|
|
<c/>
|
|
</StreamIndex>
|
|
</SmoothStreamingMedia>
|
|
"""
|
|
|
|
|
|
def test_placeholder_spellings_repeat_and_audiotag_fallback() -> None:
|
|
tracks = ISM.from_text(MANIFEST, "https://cdn.example/x.ism/manifest").to_tracks()
|
|
(audio,) = tracks.audio
|
|
segments = audio.data["ism"]["segments"]
|
|
assert segments == [
|
|
"https://cdn.example/x.ism/QualityLevels(128000)/Fragments(audio=0)",
|
|
"https://cdn.example/x.ism/QualityLevels(128000)/Fragments(audio=20000000)",
|
|
"https://cdn.example/x.ism/QualityLevels(128000)/Fragments(audio=40000000)",
|
|
]
|
|
assert audio.codec == Audio.Codec.AAC # AudioTag 255 means AACL
|
|
|
|
|
|
def test_missing_language_without_fallback_raises() -> None:
|
|
# MS-SSTR Language is optional; video StreamIndexes commonly omit it.
|
|
no_lang = MANIFEST.replace(' Language="en"', "")
|
|
with pytest.raises(ValueError, match="fallback language"):
|
|
ISM.from_text(no_lang, "https://cdn.example/x.ism/manifest").to_tracks()
|
|
tracks = ISM.from_text(no_lang, "https://cdn.example/x.ism/manifest").to_tracks(language="en")
|
|
assert str(tracks.audio[0].language) == "en"
|
|
|
|
|
|
def test_live_manifest_rejected() -> None:
|
|
live = MANIFEST.replace('Duration="60000000"', 'Duration="60000000" IsLive="TRUE"', 1)
|
|
with pytest.raises(ValueError, match="Live"):
|
|
ISM.from_text(live, "https://cdn.example/x.ism/manifest").to_tracks()
|