mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
4a21dbebb5
--daily flag plus Service.DAILY/Title.daily, -w selection by ISO air date or colon date range, and a pre-1970 air-date guard.
167 lines
6.2 KiB
Python
167 lines
6.2 KiB
Python
"""The --daily air-date fill: dl.fill_air_dates and the daily_hint gate."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import date, timedelta
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from unshackle.core import providers
|
|
from unshackle.core.titles.episode import Episode, Series
|
|
|
|
|
|
class DummyService:
|
|
pass
|
|
|
|
|
|
def make_episode(**overrides) -> Episode:
|
|
kwargs = dict(
|
|
id_="ep-0001",
|
|
service=DummyService,
|
|
title="The Show",
|
|
season=1,
|
|
number=1,
|
|
)
|
|
kwargs.update(overrides)
|
|
return Episode(**kwargs)
|
|
|
|
|
|
class _FakeProvider:
|
|
def __init__(self, episodes: list, source_order: str = "official") -> None:
|
|
self.episodes = episodes
|
|
self.source_order = source_order
|
|
|
|
def detect_order(self, _id, _keys) -> str:
|
|
return self.source_order
|
|
|
|
def get_episodes(self, _id, _order) -> list:
|
|
return self.episodes
|
|
|
|
|
|
def _fill(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
titles: list,
|
|
episodes: list,
|
|
tvdb_id=73871,
|
|
provider=True,
|
|
daily=True,
|
|
service_daily=False,
|
|
):
|
|
from unshackle.commands.dl import dl
|
|
|
|
fake = _FakeProvider(episodes) if provider else None
|
|
monkeypatch.setattr(providers, "get_provider", lambda _name: fake)
|
|
logged: list = []
|
|
logger = SimpleNamespace(
|
|
debug=lambda *a, **k: logged.append(("debug", a)),
|
|
info=lambda *a, **k: logged.append(("info", a)),
|
|
warning=lambda *a, **k: logged.append(("warning", a)),
|
|
)
|
|
stub = SimpleNamespace(tvdb_id=tvdb_id, log=logger, daily=daily, service_daily=service_daily)
|
|
stub.daily_hint = lambda title=None: dl.daily_hint(stub, title)
|
|
series = Series(titles)
|
|
dl.fill_air_dates(stub, series, tvdb_id)
|
|
return series, logged
|
|
|
|
|
|
def tvdb_episode(season: int, number: int, aired) -> dict:
|
|
return {"seasonNumber": season, "number": number, "aired": aired}
|
|
|
|
|
|
LISTING = [tvdb_episode(1, 1, "2026-08-10"), tvdb_episode(1, 2, "2026-08-11")]
|
|
|
|
|
|
def test_fill_populates_missing_air_dates(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
episodes = [make_episode(number=1), make_episode(number=2)]
|
|
series, logged = _fill(monkeypatch, episodes, LISTING)
|
|
assert [t.air_date for t in series] == [date(2026, 8, 10), date(2026, 8, 11)]
|
|
assert [level for level, _ in logged] == ["info"]
|
|
|
|
|
|
def test_a_service_set_air_date_wins(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
episodes = [make_episode(number=1, air_date="1999-01-02"), make_episode(number=2)]
|
|
series, _ = _fill(monkeypatch, episodes, LISTING)
|
|
assert [t.air_date for t in series] == [date(1999, 1, 2), date(2026, 8, 11)]
|
|
|
|
|
|
def test_fill_never_touches_season_or_number(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
series, _ = _fill(monkeypatch, [make_episode(number=2)], LISTING)
|
|
assert [(t.season, t.number) for t in series] == [(1, 2)]
|
|
|
|
|
|
def test_an_unmapped_episode_keeps_no_air_date(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
series, _ = _fill(monkeypatch, [make_episode(season=9, number=9)], LISTING)
|
|
assert series[0].air_date is None
|
|
|
|
|
|
@pytest.mark.parametrize("aired", [None, "", "not-a-date", "2026-13-40"])
|
|
def test_an_unusable_aired_value_is_skipped(monkeypatch: pytest.MonkeyPatch, aired) -> None:
|
|
series, _ = _fill(monkeypatch, [make_episode(number=1)], [tvdb_episode(1, 1, aired)])
|
|
assert series[0].air_date is None
|
|
|
|
|
|
def test_a_pre_1970_date_is_skipped(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
series, _ = _fill(monkeypatch, [make_episode(number=1)], [tvdb_episode(1, 1, "1900-01-01")])
|
|
assert series[0].air_date is None
|
|
|
|
|
|
def test_a_future_placeholder_date_is_skipped(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
future = (date.today() + timedelta(days=30)).isoformat()
|
|
series, _ = _fill(monkeypatch, [make_episode(number=1)], [tvdb_episode(1, 1, future)])
|
|
assert series[0].air_date is None
|
|
|
|
|
|
def test_tomorrow_still_fills(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
tomorrow = date.today() + timedelta(days=1)
|
|
series, _ = _fill(monkeypatch, [make_episode(number=1)], [tvdb_episode(1, 1, tomorrow.isoformat())])
|
|
assert series[0].air_date == tomorrow
|
|
|
|
|
|
def test_a_datetime_stamped_aired_value_keeps_the_date(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
series, _ = _fill(monkeypatch, [make_episode(number=1)], [tvdb_episode(1, 1, "2026-08-10T20:00:00Z")])
|
|
assert series[0].air_date == date(2026, 8, 10)
|
|
|
|
|
|
def test_fill_skips_silently_without_an_api_key(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
series, logged = _fill(monkeypatch, [make_episode(number=1)], LISTING, provider=False)
|
|
assert series[0].air_date is None
|
|
assert [level for level, _ in logged] == ["debug"]
|
|
|
|
|
|
def test_fill_skips_silently_without_a_tvdb_id(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
series, logged = _fill(monkeypatch, [make_episode(number=1)], LISTING, tvdb_id=None)
|
|
assert series[0].air_date is None
|
|
assert [level for level, _ in logged] == ["debug"]
|
|
|
|
|
|
def test_fill_skips_silently_when_tvdb_lists_no_episodes(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
series, logged = _fill(monkeypatch, [make_episode(number=1)], [])
|
|
assert series[0].air_date is None
|
|
assert [level for level, _ in logged] == ["debug"]
|
|
|
|
|
|
def test_nothing_is_filled_when_no_title_is_daily(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
series, logged = _fill(monkeypatch, [make_episode(number=1)], LISTING, daily=False)
|
|
assert series[0].air_date is None
|
|
assert logged == []
|
|
|
|
|
|
def test_the_service_daily_flag_fills_without_the_cli_flag(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
series, _ = _fill(monkeypatch, [make_episode(number=1)], LISTING, daily=False, service_daily=True)
|
|
assert series[0].air_date == date(2026, 8, 10)
|
|
|
|
|
|
def test_a_per_title_daily_false_is_skipped_while_others_fill(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
first, second = make_episode(number=1), make_episode(number=2)
|
|
first.daily = False
|
|
series, _ = _fill(monkeypatch, [first, second], LISTING, daily=False, service_daily=True)
|
|
assert [t.air_date for t in series] == [None, date(2026, 8, 11)]
|
|
|
|
|
|
def test_the_cli_flag_overrides_a_per_title_daily_false(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
episode = make_episode(number=1)
|
|
episode.daily = False
|
|
series, _ = _fill(monkeypatch, [episode], LISTING, daily=True)
|
|
assert series[0].air_date == date(2026, 8, 10)
|