"""FC-3d: scheduler_service unit tests. Covers compute_effective_interval (override/artist/default precedence, backoff exponent at failure counts 0/1/3/6/10, floor and ceiling clamp) and select_due_sources (auto_check off, never-checked, enabled=False, due/not-due boundary). """ from datetime import UTC, datetime, timedelta import pytest from backend.app.models import Artist, ImportSettings, Source from backend.app.services.scheduler_service import ( compute_effective_interval, select_due_sources, ) pytestmark = pytest.mark.integration # --- compute_effective_interval (pure, no DB) ----------------------------- def _src(override=None, failures=0): return Source( artist_id=1, platform="patreon", url="https://x", enabled=True, check_interval_override=override, consecutive_failures=failures, ) def _art(interval=None, auto=True): return Artist(name="A", slug="a", auto_check=auto, check_interval_seconds=interval) def _settings(default=28800): return ImportSettings(id=1, download_schedule_default_seconds=default) def test_interval_uses_source_override_when_set(): assert compute_effective_interval(_src(override=600), _art(interval=3600), _settings()) == 600 def test_interval_falls_back_to_artist_interval(): assert compute_effective_interval(_src(), _art(interval=3600), _settings()) == 3600 def test_interval_falls_back_to_global_default(): assert compute_effective_interval(_src(), _art(), _settings(default=900)) == 900 def test_interval_backoff_zero_failures_is_identity(): assert compute_effective_interval(_src(override=600), _art(), _settings()) == 600 def test_interval_backoff_one_failure_doubles(): assert compute_effective_interval(_src(override=600, failures=1), _art(), _settings()) == 1200 def test_interval_backoff_three_failures_x8(): assert compute_effective_interval(_src(override=600, failures=3), _art(), _settings()) == 4800 def test_interval_backoff_caps_at_exponent_six(): # 600 * 2^6 = 38400; 2^7 would be 76800 but exponent caps at 6. assert compute_effective_interval(_src(override=600, failures=6), _art(), _settings()) == 38400 assert compute_effective_interval(_src(override=600, failures=10), _art(), _settings()) == 38400 def test_interval_floor_at_60_seconds(): # base 30 with 0 failures would be 30s; floor lifts to 60. assert compute_effective_interval(_src(override=30), _art(), _settings()) == 60 def test_interval_ceiling_at_86400_seconds(): # base 28800 * 2^6 = 1843200; ceiling caps at 86400. assert compute_effective_interval(_src(failures=6), _art(), _settings(default=28800)) == 86400 # --- select_due_sources (real session) ------------------------------------ async def _seed_artist(db, *, auto=True, interval=None, name="A"): a = Artist(name=name, slug=name.lower(), auto_check=auto, check_interval_seconds=interval) db.add(a) await db.flush() return a @pytest.mark.asyncio async def test_select_skips_disabled_sources(db): artist = await _seed_artist(db, name="sel-disabled") db.add(Source( artist_id=artist.id, platform="patreon", url="https://sel-disabled", enabled=False, consecutive_failures=0, )) await db.commit() due = await select_due_sources(db) assert all(s.url != "https://sel-disabled" for s in due) @pytest.mark.asyncio async def test_select_skips_artist_auto_check_false(db): artist = await _seed_artist(db, auto=False, name="sel-noauto") db.add(Source( artist_id=artist.id, platform="patreon", url="https://sel-noauto", enabled=True, consecutive_failures=0, )) await db.commit() due = await select_due_sources(db) assert all(s.url != "https://sel-noauto" for s in due) @pytest.mark.asyncio async def test_select_includes_never_checked(db): artist = await _seed_artist(db, name="sel-never") src = Source( artist_id=artist.id, platform="patreon", url="https://sel-never", enabled=True, consecutive_failures=0, ) db.add(src) await db.commit() due = await select_due_sources(db) assert any(s.url == "https://sel-never" for s in due) @pytest.mark.asyncio async def test_select_excludes_not_yet_due(db): artist = await _seed_artist(db, interval=3600, name="sel-recent") db.add(Source( artist_id=artist.id, platform="patreon", url="https://sel-recent", enabled=True, consecutive_failures=0, last_checked_at=datetime.now(UTC) - timedelta(seconds=60), )) await db.commit() due = await select_due_sources(db) assert all(s.url != "https://sel-recent" for s in due) @pytest.mark.asyncio async def test_select_includes_past_due(db): artist = await _seed_artist(db, interval=60, name="sel-past") db.add(Source( artist_id=artist.id, platform="patreon", url="https://sel-past", enabled=True, consecutive_failures=0, last_checked_at=datetime.now(UTC) - timedelta(seconds=3600), )) await db.commit() due = await select_due_sources(db) assert any(s.url == "https://sel-past" for s in due)