mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
619c86cd10
Fingerprint-gated reuse of atomically-published segment files (DASH/ISM/HLS) behind a continue_downloads config option and one-off dl --continue-downloads flag.
110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
"""Lifecycle tests for the conditional wipe in Track.download itself.
|
|
|
|
The manifest-parser gates are covered by the DASH/HLS integration tests; these cover the
|
|
layer above them: with continue_downloads on, Track.download's entry sweep and its
|
|
exception handler must both preserve completed segments and the sidecar while purging
|
|
every partial, and with the flag off both must wipe the directory entirely. The manifest
|
|
parser is stubbed so only Track.download's own behaviour is under test.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from functools import partial
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from unshackle.core.config import config
|
|
from unshackle.core.constants import DOWNLOAD_CANCELLED, DOWNLOAD_LICENCE_ONLY
|
|
from unshackle.core.tracks import Video, resume
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clean_flags(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(config.directories, "temp", tmp_path)
|
|
DOWNLOAD_CANCELLED.clear()
|
|
DOWNLOAD_LICENCE_ONLY.clear()
|
|
yield
|
|
DOWNLOAD_CANCELLED.clear()
|
|
|
|
|
|
def make_track() -> Video:
|
|
return Video(
|
|
id_="t1",
|
|
url="https://example.test/manifest.mpd",
|
|
language="en",
|
|
codec=Video.Codec.AVC,
|
|
range_=Video.Range.SDR,
|
|
width=1920,
|
|
height=1080,
|
|
bitrate=5_000_000,
|
|
descriptor=Video.Descriptor.DASH,
|
|
)
|
|
|
|
|
|
def seed_prior_run(tmp_path: Path) -> Path:
|
|
save_dir = tmp_path / "Video_t1.mp4_segments"
|
|
(save_dir / "segments").mkdir(parents=True)
|
|
(save_dir / "0.mp4").write_bytes(b"complete segment")
|
|
(save_dir / "1.mp4.p.!dev").write_bytes(b"partial")
|
|
(save_dir / "segments" / "2.bin.h.!dev").write_bytes(b"hedge partial")
|
|
resume.write_sidecar(save_dir, "f" * 64)
|
|
return save_dir
|
|
|
|
|
|
def run_download(track: Video, monkeypatch, fail: bool):
|
|
import requests as requests_lib
|
|
|
|
from unshackle.core.manifests import DASH
|
|
|
|
def stub_download_track(track, ctx):
|
|
if fail:
|
|
raise RuntimeError("simulated mid-download crash")
|
|
ctx.save_path.write_bytes(b"output")
|
|
track.path = ctx.save_path
|
|
|
|
monkeypatch.setattr(DASH, "download_track", staticmethod(stub_download_track))
|
|
track.download(
|
|
session=requests_lib.Session(),
|
|
prepare_drm=partial(lambda *a, **kw: None),
|
|
progress=partial(lambda **kw: None),
|
|
)
|
|
|
|
|
|
def test_flag_on_keeps_segments_and_sidecar_sweeps_partials(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(config, "continue_downloads", True)
|
|
save_dir = seed_prior_run(tmp_path)
|
|
track = make_track()
|
|
|
|
with pytest.raises(RuntimeError):
|
|
run_download(track, monkeypatch, fail=True)
|
|
|
|
assert (save_dir / "0.mp4").read_bytes() == b"complete segment"
|
|
assert resume.sidecar_path(save_dir).exists()
|
|
assert not list(save_dir.rglob("*.!dev"))
|
|
|
|
|
|
def test_flag_off_wipes_segments_and_sidecar(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(config, "continue_downloads", False)
|
|
save_dir = seed_prior_run(tmp_path)
|
|
track = make_track()
|
|
|
|
with pytest.raises(RuntimeError):
|
|
run_download(track, monkeypatch, fail=True)
|
|
|
|
assert not save_dir.exists()
|
|
assert not resume.sidecar_path(save_dir).exists()
|
|
|
|
|
|
def test_flag_off_clears_orphaned_sidecar_without_dir(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(config, "continue_downloads", False)
|
|
save_dir = tmp_path / "Video_t1.mp4_segments"
|
|
# a kill between a parser's dir removal and its sidecar clear leaves exactly this
|
|
resume.write_sidecar(save_dir, "f" * 64)
|
|
track = make_track()
|
|
|
|
with pytest.raises(RuntimeError):
|
|
run_download(track, monkeypatch, fail=True)
|
|
|
|
assert not resume.sidecar_path(save_dir).exists()
|