Joakim Hellsén
415dd12fd9
All checks were successful
Deploy to Server / deploy (push) Successful in 9s
130 lines
4.9 KiB
Python
130 lines
4.9 KiB
Python
from __future__ import annotations
|
|
|
|
import datetime
|
|
import json
|
|
from datetime import timedelta
|
|
from typing import TYPE_CHECKING
|
|
|
|
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
|
|
|
|
if TYPE_CHECKING:
|
|
from django.test.client import _MonkeyPatchedWSGIResponse
|
|
|
|
|
|
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 = Organization.objects.create(
|
|
twitch_id="org123",
|
|
name="Test Organization",
|
|
)
|
|
|
|
# Create test game
|
|
self.game: 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: datetime.datetime = timezone.now()
|
|
self.campaign: DropCampaign = 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: _MonkeyPatchedWSGIResponse = 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: _MonkeyPatchedWSGIResponse = 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: _MonkeyPatchedWSGIResponse = 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: _MonkeyPatchedWSGIResponse = 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: _MonkeyPatchedWSGIResponse = 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: _MonkeyPatchedWSGIResponse = 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: _MonkeyPatchedWSGIResponse = 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: _MonkeyPatchedWSGIResponse = 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"
|