142 lines
5.2 KiB
Python
142 lines
5.2 KiB
Python
"""Test RSS feeds."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import timedelta
|
|
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
from django.utils import timezone
|
|
|
|
from twitch.models import DropCampaign
|
|
from twitch.models import Game
|
|
from twitch.models import Organization
|
|
|
|
|
|
class RSSFeedTestCase(TestCase):
|
|
"""Test RSS feeds."""
|
|
|
|
def setUp(self) -> None:
|
|
"""Set up test fixtures."""
|
|
self.org = Organization.objects.create(
|
|
twitch_id="test-org-123",
|
|
name="Test Organization",
|
|
)
|
|
self.game = Game.objects.create(
|
|
twitch_id="test-game-123",
|
|
slug="test-game",
|
|
name="Test Game",
|
|
display_name="Test Game",
|
|
owner=self.org,
|
|
)
|
|
self.campaign = DropCampaign.objects.create(
|
|
twitch_id="test-campaign-123",
|
|
name="Test Campaign",
|
|
game=self.game,
|
|
start_at=timezone.now(),
|
|
end_at=timezone.now() + timedelta(days=7),
|
|
operation_name="DropCampaignDetails",
|
|
)
|
|
|
|
def test_organization_feed(self) -> None:
|
|
"""Test organization feed returns 200."""
|
|
url = reverse("twitch:organization_feed")
|
|
response = self.client.get(url)
|
|
assert response.status_code == 200
|
|
assert response["Content-Type"] == "application/rss+xml; charset=utf-8"
|
|
|
|
def test_game_feed(self) -> None:
|
|
"""Test game feed returns 200."""
|
|
url = reverse("twitch:game_feed")
|
|
response = self.client.get(url)
|
|
assert response.status_code == 200
|
|
assert response["Content-Type"] == "application/rss+xml; charset=utf-8"
|
|
|
|
def test_campaign_feed(self) -> None:
|
|
"""Test campaign feed returns 200."""
|
|
url = reverse("twitch:campaign_feed")
|
|
response = self.client.get(url)
|
|
assert response.status_code == 200
|
|
assert response["Content-Type"] == "application/rss+xml; charset=utf-8"
|
|
|
|
def test_game_campaign_feed(self) -> None:
|
|
"""Test game-specific campaign feed returns 200."""
|
|
url = reverse("twitch:game_campaign_feed", args=[self.game.twitch_id])
|
|
response = self.client.get(url)
|
|
assert response.status_code == 200
|
|
assert response["Content-Type"] == "application/rss+xml; charset=utf-8"
|
|
# Verify the game name is in the feed
|
|
content = response.content.decode("utf-8")
|
|
assert "Test Game" in content
|
|
|
|
def test_organization_campaign_feed(self) -> None:
|
|
"""Test organization-specific campaign feed returns 200."""
|
|
url = reverse("twitch:organization_campaign_feed", args=[self.org.twitch_id])
|
|
response = self.client.get(url)
|
|
assert response.status_code == 200
|
|
assert response["Content-Type"] == "application/rss+xml; charset=utf-8"
|
|
# Verify the organization name is in the feed
|
|
content = response.content.decode("utf-8")
|
|
assert "Test Organization" in content
|
|
|
|
def test_game_campaign_feed_filters_correctly(self) -> None:
|
|
"""Test game campaign feed only shows campaigns for that game."""
|
|
# Create another game with a campaign
|
|
other_game = Game.objects.create(
|
|
twitch_id="other-game-123",
|
|
slug="other-game",
|
|
name="Other Game",
|
|
display_name="Other Game",
|
|
owner=self.org,
|
|
)
|
|
DropCampaign.objects.create(
|
|
twitch_id="other-campaign-123",
|
|
name="Other Campaign",
|
|
game=other_game,
|
|
start_at=timezone.now(),
|
|
end_at=timezone.now() + timedelta(days=7),
|
|
operation_name="DropCampaignDetails",
|
|
)
|
|
|
|
# Get feed for first game
|
|
url = reverse("twitch:game_campaign_feed", args=[self.game.twitch_id])
|
|
response = self.client.get(url)
|
|
content = response.content.decode("utf-8")
|
|
|
|
# Should contain first campaign
|
|
assert "Test Campaign" in content
|
|
# Should NOT contain other campaign
|
|
assert "Other Campaign" not in content
|
|
|
|
def test_organization_campaign_feed_filters_correctly(self) -> None:
|
|
"""Test organization campaign feed only shows campaigns for that organization."""
|
|
# Create another organization with a game and campaign
|
|
other_org = Organization.objects.create(
|
|
twitch_id="other-org-123",
|
|
name="Other Organization",
|
|
)
|
|
other_game = Game.objects.create(
|
|
twitch_id="other-game-456",
|
|
slug="other-game-2",
|
|
name="Other Game 2",
|
|
display_name="Other Game 2",
|
|
owner=other_org,
|
|
)
|
|
DropCampaign.objects.create(
|
|
twitch_id="other-campaign-456",
|
|
name="Other Campaign 2",
|
|
game=other_game,
|
|
start_at=timezone.now(),
|
|
end_at=timezone.now() + timedelta(days=7),
|
|
operation_name="DropCampaignDetails",
|
|
)
|
|
|
|
# Get feed for first organization
|
|
url = reverse("twitch:organization_campaign_feed", args=[self.org.twitch_id])
|
|
response = self.client.get(url)
|
|
content = response.content.decode("utf-8")
|
|
|
|
# Should contain first campaign
|
|
assert "Test Campaign" in content
|
|
# Should NOT contain other campaign
|
|
assert "Other Campaign 2" not in content
|