45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from restreamer.ffmpeg_runner import build_ffmpeg_args
|
|
from restreamer.resolver import ResolvedStream
|
|
|
|
|
|
def test_build_ffmpeg_args_single_input_without_headers() -> None:
|
|
stream = ResolvedStream(video_url="https://example.com/stream.m3u8")
|
|
|
|
args = build_ffmpeg_args(stream)
|
|
|
|
assert args[0] == "ffmpeg"
|
|
assert "-headers" not in args
|
|
assert args.count("-i") == 1
|
|
assert stream.video_url in args
|
|
assert args[-1] == "pipe:1"
|
|
|
|
|
|
def test_build_ffmpeg_args_includes_headers_when_present() -> None:
|
|
stream = ResolvedStream(
|
|
video_url="https://example.com/stream.m3u8",
|
|
http_headers={"User-Agent": "test-agent", "Cookie": "a=b"},
|
|
)
|
|
|
|
args = build_ffmpeg_args(stream)
|
|
|
|
assert "-headers" in args
|
|
header_value = args[args.index("-headers") + 1]
|
|
assert "User-Agent: test-agent\r\n" in header_value
|
|
assert "Cookie: a=b\r\n" in header_value
|
|
|
|
|
|
def test_build_ffmpeg_args_maps_separate_video_and_audio_inputs() -> None:
|
|
stream = ResolvedStream(
|
|
video_url="https://example.com/video.m3u8",
|
|
audio_url="https://example.com/audio.m3u8",
|
|
)
|
|
|
|
args = build_ffmpeg_args(stream)
|
|
|
|
assert args.count("-i") == 2
|
|
assert stream.video_url in args
|
|
assert stream.audio_url in args
|
|
assert "-map" in args
|
|
map_values = [args[i + 1] for i, a in enumerate(args) if a == "-map"]
|
|
assert map_values == ["0:v:0", "1:a:0"]
|