150 lines
5.0 KiB
Python
150 lines
5.0 KiB
Python
"""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)
|