diff --git a/tests/test_api_artists_directory.py b/tests/test_api_artists_directory.py index 64b3dda..1df5863 100644 --- a/tests/test_api_artists_directory.py +++ b/tests/test_api_artists_directory.py @@ -43,7 +43,8 @@ async def test_directory_card_shape(client, seeded): body = await resp.get_json() card = next(c for c in body["cards"] if c["name"] == "alice-api") assert set(card.keys()) == { - "id", "name", "slug", "is_subscription", "image_count", "preview_thumbnails", + "id", "name", "slug", "is_subscription", "image_count", + "unseen_count", "preview_thumbnails", } assert card["is_subscription"] is True assert card["image_count"] == 1 diff --git a/tests/test_artist_visit.py b/tests/test_artist_visit.py index 7236d33..b080e33 100644 --- a/tests/test_artist_visit.py +++ b/tests/test_artist_visit.py @@ -11,7 +11,7 @@ Covers: last_viewed_at = NOW()) """ -from datetime import datetime, timedelta, timezone +from datetime import UTC, datetime, timedelta import pytest from sqlalchemy import select @@ -23,8 +23,8 @@ from backend.app.services.artist_service import ArtistService pytestmark = pytest.mark.integration -_LONG_AGO = datetime(2000, 1, 1, tzinfo=timezone.utc) -_RECENTLY = datetime(2099, 1, 1, tzinfo=timezone.utc) +_LONG_AGO = datetime(2000, 1, 1, tzinfo=UTC) +_RECENTLY = datetime(2099, 1, 1, tzinfo=UTC) async def _seed_artist(db, name: str) -> Artist: