Refactor HTML

twitch/tests/test_exports.py

@@ -0,0 +1,125 @@
+from __future__ import annotations
+
+import json
+from datetime import timedelta
+
+from django.test import Client
+from django.test import TestCase
+from django.utils import timezone
+
+from twitch.models import DropCampaign
+from twitch.models import Game
+from twitch.models import Organization
+
+
+class ExportViewsTestCase(TestCase):
+    """Test export views for CSV and JSON formats."""
+
+    def setUp(self) -> None:
+        """Set up test data."""
+        self.client = Client()
+
+        # Create test organization
+        self.org = Organization.objects.create(
+            twitch_id="org123",
+            name="Test Organization",
+        )
+
+        # Create test game
+        self.game = Game.objects.create(
+            twitch_id="game123",
+            name="Test Game",
+            display_name="Test Game Display",
+        )
+        self.game.owners.add(self.org)
+
+        # Create test campaign
+        now = timezone.now()
+        self.campaign = DropCampaign.objects.create(
+            twitch_id="campaign123",
+            name="Test Campaign",
+            description="A test campaign description",
+            game=self.game,
+            start_at=now - timedelta(days=1),
+            end_at=now + timedelta(days=1),
+        )
+
+    def test_export_campaigns_csv(self) -> None:
+        """Test CSV export of campaigns."""
+        response = self.client.get("/export/campaigns/csv/")
+        assert response.status_code == 200
+        assert response["Content-Type"] == "text/csv"
+        assert b"Twitch ID" in response.content
+        assert b"campaign123" in response.content
+        assert b"Test Campaign" in response.content
+
+    def test_export_campaigns_json(self) -> None:
+        """Test JSON export of campaigns."""
+        response = self.client.get("/export/campaigns/json/")
+        assert response.status_code == 200
+        assert response["Content-Type"] == "application/json"
+
+        data = json.loads(response.content)
+        assert isinstance(data, list)
+        assert len(data) == 1
+        assert data[0]["twitch_id"] == "campaign123"
+        assert data[0]["name"] == "Test Campaign"
+        assert data[0]["status"] == "Active"
+
+    def test_export_games_csv(self) -> None:
+        """Test CSV export of games."""
+        response = self.client.get("/export/games/csv/")
+        assert response.status_code == 200
+        assert response["Content-Type"] == "text/csv"
+        assert b"Twitch ID" in response.content
+        assert b"game123" in response.content
+        assert b"Test Game Display" in response.content
+
+    def test_export_games_json(self) -> None:
+        """Test JSON export of games."""
+        response = self.client.get("/export/games/json/")
+        assert response.status_code == 200
+        assert response["Content-Type"] == "application/json"
+
+        data = json.loads(response.content)
+        assert isinstance(data, list)
+        assert len(data) == 1
+        assert data[0]["twitch_id"] == "game123"
+        assert data[0]["display_name"] == "Test Game Display"
+
+    def test_export_organizations_csv(self) -> None:
+        """Test CSV export of organizations."""
+        response = self.client.get("/export/organizations/csv/")
+        assert response.status_code == 200
+        assert response["Content-Type"] == "text/csv"
+        assert b"Twitch ID" in response.content
+        assert b"org123" in response.content
+        assert b"Test Organization" in response.content
+
+    def test_export_organizations_json(self) -> None:
+        """Test JSON export of organizations."""
+        response = self.client.get("/export/organizations/json/")
+        assert response.status_code == 200
+        assert response["Content-Type"] == "application/json"
+
+        data = json.loads(response.content)
+        assert isinstance(data, list)
+        assert len(data) == 1
+        assert data[0]["twitch_id"] == "org123"
+        assert data[0]["name"] == "Test Organization"
+
+    def test_export_campaigns_csv_with_filters(self) -> None:
+        """Test CSV export of campaigns with status filter."""
+        response = self.client.get("/export/campaigns/csv/?status=active")
+        assert response.status_code == 200
+        assert b"campaign123" in response.content
+
+    def test_export_campaigns_json_with_filters(self) -> None:
+        """Test JSON export of campaigns with status filter."""
+        response = self.client.get("/export/campaigns/json/?status=active")
+        assert response.status_code == 200
+
+        data = json.loads(response.content)
+        assert isinstance(data, list)
+        assert len(data) == 1
+        assert data[0]["status"] == "Active"