Joakim Hellsén
415dd12fd9
All checks were successful
Deploy to Server / deploy (push) Successful in 9s
113 lines
4.6 KiB
Python
113 lines
4.6 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from datetime import UTC
|
|
from datetime import datetime
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
from django.conf import settings
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
|
|
from twitch import sitemaps
|
|
from twitch.models import Channel
|
|
from twitch.models import ChatBadgeSet
|
|
from twitch.models import DropCampaign
|
|
from twitch.models import Game
|
|
from twitch.models import Organization
|
|
from twitch.models import RewardCampaign
|
|
|
|
if TYPE_CHECKING:
|
|
from pathlib import Path
|
|
|
|
|
|
class SitemapTests(TestCase):
|
|
"""Tests for Twitch sitemaps."""
|
|
|
|
def test_static_view_sitemap_items_and_location(self) -> None:
|
|
"""Test that StaticViewSitemap returns expected items and correct locations."""
|
|
sitemap = sitemaps.TwitchSitemapGenerator()
|
|
items: list[str] = sitemap.items()
|
|
expected: list[str] = [
|
|
"twitch:dashboard",
|
|
"twitch:campaign_list",
|
|
"twitch:reward_campaign_list",
|
|
"twitch:games_grid",
|
|
"twitch:games_list",
|
|
"twitch:org_list",
|
|
"twitch:channel_list",
|
|
"twitch:badge_list",
|
|
"twitch:emote_gallery",
|
|
"twitch:search",
|
|
# the two items below were added later and need coverage
|
|
"twitch:dataset_backups",
|
|
"twitch:docs_rss",
|
|
]
|
|
assert set(items) == set(expected)
|
|
for name in items:
|
|
assert sitemap.location(name) == reverse(name)
|
|
|
|
def test_game_sitemap_items_and_location(self) -> None:
|
|
"""Test that GameSitemap returns expected items and correct locations."""
|
|
game: Game = Game.objects.create(twitch_id="g-1", display_name="Test Game")
|
|
sitemap = sitemaps.GameSitemap()
|
|
items: list[Game] = list(sitemap.items())
|
|
assert game in items
|
|
assert sitemap.location(game) == reverse("twitch:game_detail", args=[game.twitch_id])
|
|
|
|
org: Organization = Organization.objects.create(twitch_id="o-1", name="Org One")
|
|
channel: Channel = Channel.objects.create(twitch_id="c-1", name="chan", display_name="Chan One")
|
|
badge_set: ChatBadgeSet = ChatBadgeSet.objects.create(set_id="b-1")
|
|
game: Game = Game.objects.create(twitch_id="g-2", display_name="Game Two")
|
|
campaign: DropCampaign = DropCampaign.objects.create(twitch_id="dc-1", name="Campaign One", game=game)
|
|
reward: RewardCampaign = RewardCampaign.objects.create(twitch_id="rc-1", name="Reward One")
|
|
|
|
campaign_sitemap = sitemaps.CampaignSitemap()
|
|
assert campaign in list(campaign_sitemap.items())
|
|
assert campaign_sitemap.location(campaign) == reverse("twitch:campaign_detail", args=[campaign.twitch_id])
|
|
|
|
org_sitemap = sitemaps.OrganizationSitemap()
|
|
assert org in list(org_sitemap.items())
|
|
assert org_sitemap.location(org) == reverse("twitch:organization_detail", args=[org.twitch_id])
|
|
|
|
channel_sitemap = sitemaps.ChannelSitemap()
|
|
assert channel in list(channel_sitemap.items())
|
|
assert channel_sitemap.location(channel) == reverse("twitch:channel_detail", args=[channel.twitch_id])
|
|
|
|
badge_sitemap = sitemaps.BadgeSitemap()
|
|
assert badge_set in list(badge_sitemap.items())
|
|
assert badge_sitemap.location(badge_set) == reverse("twitch:badge_set_detail", args=[badge_set.set_id])
|
|
|
|
reward_sitemap = sitemaps.RewardCampaignSitemap()
|
|
assert reward in list(reward_sitemap.items())
|
|
assert reward_sitemap.location(reward) == reverse("twitch:reward_campaign_detail", args=[reward.twitch_id])
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_static_view_lastmod_behavior(tmp_path: sitemaps.Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""Standalone pytest test for :meth:`TwitchSitemapGenerator.lastmod`.
|
|
|
|
We exercise both the docs RSS branch (which should always return ``None``)
|
|
and the dataset backups branch, including scenarios with and without
|
|
backup files present.
|
|
"""
|
|
sitemap = sitemaps.TwitchSitemapGenerator()
|
|
|
|
assert sitemap.lastmod("twitch:docs_rss") is None
|
|
|
|
monkeypatch.setattr(settings, "DATA_DIR", tmp_path)
|
|
assert sitemap.lastmod("twitch:dataset_backups") is None
|
|
|
|
datasets: Path = tmp_path / "datasets"
|
|
datasets.mkdir()
|
|
older: Path = datasets / "dataset_backup_old.zip"
|
|
newer: Path = datasets / "dataset_backup_new.zip"
|
|
older.write_text("old", encoding="utf-8")
|
|
newer.write_text("new", encoding="utf-8")
|
|
|
|
os.utime(older, (1_000, 1_000))
|
|
os.utime(newer, (2_000, 2_000))
|
|
|
|
expected: datetime = datetime.fromtimestamp(2_000, tz=UTC)
|
|
assert sitemap.lastmod("twitch:dataset_backups") == expected
|