49222dd0b8
The calendar pulled every airing show in the library regardless of whether you follow it. Now it's scoped to the EFFECTIVE watchlist by default — explicit show follows ∪ airing library shows (not muted), same logic as the Shows watchlist tab — so it tracks what you actually care about, and you can mute a show off it. A 'Watchlist / All library' toggle on the calendar lets you flip to everything you own (remembered in localStorage). - calendar_upcoming(watchlist_only=) adds the watchlist filter; /calendar takes ?scope=watchlist|all (default watchlist). - Calendar page gets a scope toggle (defaults watchlist, persists, refetches on change). Tests: DB scope (followed-only / airing-default / mute drops out / all-library sees all) + frontend wiring. node --check clean.
85 lines
3.4 KiB
Python
85 lines
3.4 KiB
Python
"""Calendar source scope: watchlist (default) vs all-library.
|
||
|
||
watchlist_only restricts the feed to the EFFECTIVE watchlist — explicit show
|
||
follows ∪ airing library shows (not muted) — so the calendar tracks what you
|
||
follow, with an 'All library' escape hatch.
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import pytest
|
||
|
||
from database.video_database import VideoDatabase
|
||
|
||
|
||
@pytest.fixture()
|
||
def db(tmp_path):
|
||
return VideoDatabase(database_path=str(tmp_path / "video_library.db"))
|
||
|
||
|
||
def _seed_show(db, tmdb_id, title, server_id, air_date="2026-06-20"):
|
||
return db.upsert_show_tree("plex", {
|
||
"server_id": server_id, "title": title, "tmdb_id": tmdb_id,
|
||
"seasons": [{"season_number": 1, "episodes": [
|
||
{"episode_number": 1, "title": "E1", "air_date": air_date}]}]})
|
||
|
||
|
||
def _set_status(db, show_id, status):
|
||
conn = db._get_connection()
|
||
conn.execute("UPDATE shows SET status=? WHERE id=?", (status, show_id))
|
||
conn.commit()
|
||
conn.close()
|
||
|
||
|
||
def _tmdbs(rows):
|
||
return {r["show_tmdb_id"] for r in rows}
|
||
|
||
|
||
def test_watchlist_scope_only_returns_followed_shows(db):
|
||
_seed_show(db, 11, "A", "sA")
|
||
_seed_show(db, 22, "B", "sB")
|
||
db.add_to_watchlist("show", 11, "A") # follow A only (B is not airing → not auto-included)
|
||
|
||
allp = db.calendar_upcoming("2026-06-01", "2026-06-30", server_source="plex", watchlist_only=False)
|
||
wl = db.calendar_upcoming("2026-06-01", "2026-06-30", server_source="plex", watchlist_only=True)
|
||
assert _tmdbs(allp) == {11, 22} # all-library sees both
|
||
assert _tmdbs(wl) == {11} # watchlist sees only the followed one
|
||
|
||
|
||
def test_watchlist_scope_includes_airing_default_and_respects_mute(db):
|
||
a = _seed_show(db, 33, "Airing", "sA")
|
||
_set_status(db, a, "Returning Series")
|
||
|
||
# airing show, not explicitly followed → still in the watchlist scope by default
|
||
wl = db.calendar_upcoming("2026-06-01", "2026-06-30", server_source="plex", watchlist_only=True)
|
||
assert _tmdbs(wl) == {33}
|
||
|
||
# mute it → drops out of the watchlist scope (but still in all-library)
|
||
db.remove_from_watchlist("show", 33) # stores a 'mute' tombstone
|
||
wl2 = db.calendar_upcoming("2026-06-01", "2026-06-30", server_source="plex", watchlist_only=True)
|
||
assert wl2 == []
|
||
allp = db.calendar_upcoming("2026-06-01", "2026-06-30", server_source="plex", watchlist_only=False)
|
||
assert _tmdbs(allp) == {33}
|
||
|
||
|
||
# ── frontend wiring (toggle defaults to watchlist, persists, refetches) ─────
|
||
from pathlib import Path
|
||
|
||
_ROOT = Path(__file__).resolve().parent.parent
|
||
_JS = (_ROOT / "webui" / "static" / "video" / "video-calendar.js").read_text(encoding="utf-8")
|
||
_INDEX = (_ROOT / "webui" / "index.html").read_text(encoding="utf-8")
|
||
|
||
|
||
def test_calendar_has_scope_toggle_defaulting_to_watchlist():
|
||
assert 'data-video-cal-scope="watchlist"' in _INDEX
|
||
assert 'data-video-cal-scope="all"' in _INDEX
|
||
# the watchlist button is the one pre-selected (--on)
|
||
i = _INDEX.index('data-video-cal-scope="watchlist"')
|
||
assert 'vcal-filter-btn--on' in _INDEX[i - 80:i]
|
||
|
||
|
||
def test_calendar_js_defaults_watchlist_and_sends_scope():
|
||
assert "scope: 'watchlist'" in _JS # default state
|
||
assert "'&scope=' + (state.scope" in _JS # sent to the API
|
||
assert "function setScope(" in _JS # toggle refetches
|
||
assert "localStorage.setItem('vcalScope'" in _JS # remembers the choice
|