ttvdrops/chzzk/tests/test_views.py
Joakim Hellsén c9522f9d3d
All checks were successful
Deploy to Server / deploy (push) Successful in 21s
Fix typo in dashboard_view to exclude campaigns with state "TESTING"
2026-04-01 14:20:24 +02:00

62 lines
2 KiB
Python

from datetime import timedelta
from typing import TYPE_CHECKING
from django.test import TestCase
from django.urls import reverse
from django.utils import timezone
from chzzk.models import ChzzkCampaign
if TYPE_CHECKING:
from datetime import datetime
from django.test.client import _MonkeyPatchedWSGIResponse
class ChzzkDashboardViewTests(TestCase):
"""Test cases for the dashboard view of the chzzk app."""
def test_dashboard_view_excludes_testing_state_campaigns(self) -> None:
"""Test that the dashboard view excludes campaigns in the TESTING state."""
now: datetime = timezone.now()
ChzzkCampaign.objects.create(
campaign_no=1001,
title="Testing campaign",
description="Should be excluded",
category_type="game",
category_id="1",
category_value="TestGame",
service_id="chzzk",
state="TESTING",
start_date=now - timedelta(days=1),
end_date=now + timedelta(days=1),
has_ios_based_reward=False,
drops_campaign_not_started=False,
source_api="unit-test",
)
included: ChzzkCampaign = ChzzkCampaign.objects.create(
campaign_no=1002,
title="Active campaign",
description="Should be included",
category_type="game",
category_id="1",
category_value="TestGame",
service_id="chzzk",
state="ACTIVE",
start_date=now - timedelta(days=1),
end_date=now + timedelta(days=1),
has_ios_based_reward=False,
drops_campaign_not_started=False,
source_api="unit-test",
)
response: _MonkeyPatchedWSGIResponse = self.client.get(
reverse("chzzk:dashboard"),
)
assert response.status_code == 200
campaigns: list[ChzzkCampaign] = list(response.context["active_campaigns"])
assert included in campaigns
assert all(c.state != "TESTING" for c in campaigns)