dnscms: add some basic tests

dnscms/tests/conftest.py

@@ -0,0 +1,78 @@
+import json
+
+import factory
+import pytest
+import wagtail_factories
+from wagtail.models import Page
+
+from events.models import EventIndex, EventPage
+from venues.models import VenueIndex, VenuePage
+
+
+class EventIndexFactory(wagtail_factories.PageFactory):
+    title = factory.Sequence(lambda n: f"Events {n}")
+
+    class Meta:
+        model = EventIndex
+
+
+class EventPageFactory(wagtail_factories.PageFactory):
+    title = factory.Sequence(lambda n: f"Event {n}")
+
+    class Meta:
+        model = EventPage
+
+
+class VenueIndexFactory(wagtail_factories.PageFactory):
+    title = factory.Sequence(lambda n: f"Venues {n}")
+
+    class Meta:
+        model = VenueIndex
+
+
+class VenuePageFactory(wagtail_factories.PageFactory):
+    title = factory.Sequence(lambda n: f"Venue {n}")
+
+    class Meta:
+        model = VenuePage
+
+
+@pytest.fixture
+def root_page(db):
+    return Page.objects.get(depth=1)
+
+
+@pytest.fixture
+def home_page(root_page):
+    # Wagtail's initial migration creates a default "Welcome" page at depth=2.
+    # Reuse it so we don't fight slug collisions across tests.
+    return root_page.get_children().first() or root_page.add_child(
+        instance=Page(title="Home", slug="home")
+    )
+
+
+@pytest.fixture
+def event_index(home_page):
+    return EventIndexFactory(parent=home_page)
+
+
+@pytest.fixture
+def venue(home_page):
+    venue_index = VenueIndexFactory(parent=home_page)
+    return VenuePageFactory(parent=venue_index)
+
+
+@pytest.fixture
+def graphql_post(client):
+    def _post(query, variables=None):
+        payload = {"query": query}
+        if variables is not None:
+            payload["variables"] = variables
+        response = client.post(
+            "/api/graphql/",
+            data=json.dumps(payload),
+            content_type="application/json",
+        )
+        return response, response.json()
+
+    return _post

dnscms/tests/test_events.py

@@ -0,0 +1,88 @@
+from datetime import timedelta
+
+import pytest
+from django.core.exceptions import ValidationError
+from django.utils import timezone
+
+from events.models import (
+    EventOccurrence,
+    EventOrganizer,
+    EventOrganizerLink,
+    EventPage,
+)
+from tests.conftest import EventPageFactory
+
+
+def test_eventpage_clean_unsets_specific_pricing_when_free():
+    page = EventPage(
+        title="Free event",
+        slug="free-event",
+        free=True,
+        price_regular="100",
+        price_student="50",
+        price_member="25",
+    )
+
+    page.clean()
+
+    assert page.price_regular == ""
+    assert page.price_student == ""
+    assert page.price_member == ""
+
+
+def test_eventoccurrence_clean_rejects_both_venue_and_venue_custom(event_index, venue):
+    event = EventPageFactory(parent=event_index)
+    occurrence = EventOccurrence(
+        event=event,
+        start=timezone.now(),
+        venue=venue,
+        venue_custom="Frederikkeplassen",
+    )
+
+    with pytest.raises(ValidationError) as exc:
+        occurrence.clean()
+    assert "venue_custom" in exc.value.message_dict
+
+
+def test_eventoccurrence_clean_requires_venue_or_venue_custom(event_index):
+    event = EventPageFactory(parent=event_index)
+    occurrence = EventOccurrence(event=event, start=timezone.now())
+
+    with pytest.raises(ValidationError) as exc:
+        occurrence.clean()
+    assert "venue" in exc.value.message_dict
+
+
+def test_eventpage_manager_future_filters_past_and_annotates(event_index):
+    now = timezone.now()
+
+    past = EventPageFactory(parent=event_index, title="Past")
+    EventOccurrence.objects.create(
+        event=past, start=now - timedelta(days=7), venue_custom="Old"
+    )
+
+    future = EventPageFactory(parent=event_index, title="Future")
+    EventOccurrence.objects.create(
+        event=future, start=now + timedelta(days=7), venue_custom="New"
+    )
+
+    results = list(EventPage.objects.live().future().order_by("next_occurrence"))
+
+    assert [p.pk for p in results] == [future.pk]
+    assert results[0].next_occurrence is not None
+
+
+def test_eventpage_clean_dedupes_organizers_by_name(event_index):
+    org_a = EventOrganizer.objects.create(name="DNS", slug="dns-a")
+    org_b = EventOrganizer.objects.create(name="DNS", slug="dns-b")
+
+    event = EventPageFactory(parent=event_index)
+    EventOrganizerLink.objects.create(event=event, organizer=org_a)
+    EventOrganizerLink.objects.create(event=event, organizer=org_b)
+
+    event = EventPage.objects.get(pk=event.pk)
+    assert event.organizer_links.count() == 2
+
+    event.clean()
+
+    assert event.organizer_links.count() == 1

dnscms/tests/test_graphql.py

@@ -0,0 +1,38 @@
+from datetime import timedelta
+
+from django.utils import timezone
+
+from events.models import EventOccurrence
+from tests.conftest import EventPageFactory
+
+
+def test_graphql_endpoint_responds(db, graphql_post):
+    response, body = graphql_post("{ __schema { queryType { name } } }")
+
+    assert response.status_code == 200
+    assert "errors" not in body
+    assert body["data"]["__schema"]["queryType"]["name"] == "Query"
+
+
+def test_event_index_future_events_query(event_index, graphql_post):
+    upcoming = EventPageFactory(parent=event_index, title="Upcoming gig")
+    EventOccurrence.objects.create(
+        event=upcoming,
+        start=timezone.now() + timedelta(days=3),
+        venue_custom="Storsalen",
+    )
+
+    response, body = graphql_post(
+        """
+        query {
+          eventIndex {
+            futureEvents { title }
+          }
+        }
+        """
+    )
+
+    assert response.status_code == 200
+    assert "errors" not in body, body
+    titles = [e["title"] for e in body["data"]["eventIndex"]["futureEvents"]]
+    assert "Upcoming gig" in titles