import pytest from sqlalchemy import select from backend.app.models import Artist from backend.app.services.source_service import ( KNOWN_PLATFORMS, ArtistNotFoundError, DuplicateSourceError, EmptyUrlError, InvalidConfigError, SourceService, UnknownPlatformError, ) pytestmark = pytest.mark.integration async def _artist(db, name="Alice"): a = Artist(name=name, slug=name.lower()) db.add(a) await db.flush() return a @pytest.mark.asyncio async def test_known_platforms_contains_starter_set(db): assert {"patreon", "fanbox", "subscribestar", "pixiv", "deviantart"} <= KNOWN_PLATFORMS @pytest.mark.asyncio async def test_create_flips_is_subscription_on_first_source(db): artist = await _artist(db) svc = SourceService(db) rec = await svc.create( artist_id=artist.id, platform="patreon", url="https://patreon.com/alice", ) assert rec.id is not None is_sub = (await db.execute( select(Artist.is_subscription).where(Artist.id == artist.id) )).scalar_one() assert is_sub is True @pytest.mark.asyncio async def test_delete_last_source_flips_is_subscription_off(db): artist = await _artist(db) svc = SourceService(db) rec = await svc.create( artist_id=artist.id, platform="patreon", url="https://patreon.com/alice", ) await svc.delete(rec.id) is_sub = (await db.execute( select(Artist.is_subscription).where(Artist.id == artist.id) )).scalar_one() assert is_sub is False @pytest.mark.asyncio async def test_create_rejects_unknown_platform(db): artist = await _artist(db) svc = SourceService(db) with pytest.raises(UnknownPlatformError): await svc.create( artist_id=artist.id, platform="myspace", url="https://m/x", ) @pytest.mark.asyncio async def test_create_rejects_non_dict_config(db): artist = await _artist(db) svc = SourceService(db) with pytest.raises(InvalidConfigError): await svc.create( artist_id=artist.id, platform="patreon", url="https://patreon.com/alice", config_overrides=[1, 2, 3], ) @pytest.mark.asyncio async def test_create_rejects_empty_url(db): artist = await _artist(db) svc = SourceService(db) with pytest.raises(EmptyUrlError): await svc.create(artist_id=artist.id, platform="patreon", url=" ") @pytest.mark.asyncio async def test_create_rejects_unknown_artist(db): svc = SourceService(db) with pytest.raises(ArtistNotFoundError): await svc.create(artist_id=99999, platform="patreon", url="https://x/y") @pytest.mark.asyncio async def test_create_duplicate_raises_with_existing_id(db): artist = await _artist(db) svc = SourceService(db) first = await svc.create( artist_id=artist.id, platform="patreon", url="https://patreon.com/alice", ) with pytest.raises(DuplicateSourceError) as exc: await svc.create( artist_id=artist.id, platform="patreon", url="https://patreon.com/alice", ) assert exc.value.existing_id == first.id @pytest.mark.asyncio async def test_list_filters_by_artist(db): a = await _artist(db, "Alice") b = await _artist(db, "Bob") svc = SourceService(db) await svc.create(artist_id=a.id, platform="patreon", url="https://patreon.com/a") await svc.create(artist_id=b.id, platform="patreon", url="https://patreon.com/b") only_a = await svc.list(artist_id=a.id) assert [s.artist_id for s in only_a] == [a.id] all_rows = await svc.list() assert len(all_rows) == 2 @pytest.mark.asyncio async def test_update_changes_fields(db): artist = await _artist(db) svc = SourceService(db) rec = await svc.create( artist_id=artist.id, platform="patreon", url="https://patreon.com/a", ) updated = await svc.update(rec.id, enabled=False, config_overrides={"videos": False}) assert updated.enabled is False assert updated.config_overrides == {"videos": False}