fix tests and add vouchers and products endpoints

Author: artcz

intbot/core/integrations/pretix.py

@@ -29,7 +29,7 @@
 def get_event_url(event):
     assert event in PRETALX_EVENTS
 
-    pretix_url = f"https://tickets.europython.eu"
+    pretix_url = "https://tickets.europython.eu"
     url = f"{pretix_url}/api/v1/organizers/europython/events/{event}/"
     return url
 
@@ -73,9 +73,29 @@ def fetch_pretix_data(
 def download_latest_orders(event: str) -> PretixData:
     data = fetch_pretix_data(event, PretixData.PretixResources.orders)
 
-    pretalx_data = PretixData.objects.create(
+    pretix_data = PretixData.objects.create(
         resource=PretixData.PretixResources.orders,
         content=data,
     )
 
-    return pretalx_data
+    return pretix_data
+
+def download_latest_products(event: str) -> PretixData:
+    data = fetch_pretix_data(event, PretixData.PretixResources.products)
+
+    pretix_data = PretixData.objects.create(
+        resource=PretixData.PretixResources.products,
+        content=data,
+    )
+
+    return pretix_data
+
+def download_latest_vouchers(event: str) -> PretixData:
+    data = fetch_pretix_data(event, PretixData.PretixResources.vouchers)
+
+    pretix_data = PretixData.objects.create(
+        resource=PretixData.PretixResources.vouchers,
+        content=data,
+    )
+
+    return pretix_data

intbot/core/management/commands/download_pretix_data.py

@@ -0,0 +1,32 @@
+from core.integrations.pretalx import (
+    PRETIX_EVENTS,
+    download_latest_orders,
+    download_latest_products,
+    download_latest_vouchers,
+)
+from django.core.management.base import BaseCommand
+
+
+class Command(BaseCommand):
+    help = "Downloads latest pretix data"
+
+    def add_arguments(self, parser):
+        # Add keyword argument event
+        parser.add_argument(
+            "--event",
+            choices=PRETIX_EVENTS,
+            help="slug of the event (for example `ep2025`)",
+            required=True,
+        )
+
+    def handle(self, **kwargs):
+        event = kwargs["event"]
+
+        self.stdout.write(f"Downloading latest products from pretix... {event}")
+        download_latest_products(event)
+
+        self.stdout.write(f"Downloading latest vouchers from pretix... {event}")
+        download_latest_vouchers(event)
+
+        self.stdout.write(f"Downloading latest orders from pretix... {event}")
+        download_latest_orders(event)

intbot/tests/test_integrations/test_pretix.py

@@ -34,7 +34,35 @@ def orders_pages_generator(url):
     )
 
 
-def speaker_pages_generator(url):
+def vouchers_pages_generator(url):
+    """
+    Generator to simulate pagination.
+
+    Extracted to a generator because we use it in multiple places
+    """
+    yield Response(
+        200,
+        json={
+            "results": [
+                {"hello": "world"},
+            ],
+            "next": f"{url}&page=2",
+        },
+    )
+
+    yield Response(
+        200,
+        json={
+            "results": [
+                {"foo": "bar"},
+            ],
+            # It's important to make it last page in tests.
+            # Otherwise it will be infinite loop :)
+            "next": None,
+        },
+    )
+
+def products_pages_generator(url):
     """
     Generator to simulate pagination.
 
@@ -65,7 +93,7 @@ def speaker_pages_generator(url):
 
 @respx.mock
 def test_fetch_orders_from_pretix():
-    url = "https://tickets.europython.eu/api/v1/organizers/europython/ep2025/orders/"
+    url = "https://tickets.europython.eu/api/v1/organizers/europython/events/ep2025/orders/"
     data = orders_pages_generator(url)
     respx.get(url).mock(return_value=next(data))
     respx.get(url + "&page=2").mock(return_value=next(data))
@@ -80,11 +108,45 @@ def test_fetch_orders_from_pretix():
         {"foo": "bar"},
     ]
 
+@respx.mock
+def test_fetch_vouchers_from_pretix():
+    url = "https://tickets.europython.eu/api/v1/organizers/europython/events/ep2025/vouchers/"
+    data = vouchers_pages_generator(url)
+    respx.get(url).mock(return_value=next(data))
+    respx.get(url + "&page=2").mock(return_value=next(data))
+
+    orders = pretix.fetch_pretix_data(
+        "ep2025",
+        PretixData.PretixResources.vouchers,
+    )
+
+    assert orders == [
+        {"hello": "world"},
+        {"foo": "bar"},
+    ]
+
+@respx.mock
+def test_fetch_products_from_pretix():
+    url = "https://tickets.europython.eu/api/v1/organizers/europython/events/ep2025/items/"
+    data = products_pages_generator(url)
+    respx.get(url).mock(return_value=next(data))
+    respx.get(url + "&page=2").mock(return_value=next(data))
+
+    orders = pretix.fetch_pretix_data(
+        "ep2025",
+        PretixData.PretixResources.products,
+    )
+
+    assert orders == [
+        {"hello": "world"},
+        {"foo": "bar"},
+    ]
+
 
 @respx.mock
 @pytest.mark.django_db
 def test_download_latest_orders():
-    url = "https://tickets.europython.eu/api/v1/organizers/europython/ep2025/orders/"
+    url = "https://tickets.europython.eu/api/v1/organizers/europython/events/ep2025/orders/"
     data = orders_pages_generator(url)
     respx.get(url).mock(return_value=next(data))
     respx.get(url + "&page=2").mock(return_value=next(data))
@@ -97,3 +159,37 @@ def test_download_latest_orders():
         {"hello": "world"},
         {"foo": "bar"},
     ]
+
+@respx.mock
+@pytest.mark.django_db
+def test_download_latest_products():
+    url = "https://tickets.europython.eu/api/v1/organizers/europython/events/ep2025/items/"
+    data = orders_pages_generator(url)
+    respx.get(url).mock(return_value=next(data))
+    respx.get(url + "&page=2").mock(return_value=next(data))
+
+    pretix.download_latest_products("ep2025")
+
+    pd = PretixData.objects.get(resource=PretixData.PretixResources.products)
+    assert pd.resource == "products"
+    assert pd.content == [
+        {"hello": "world"},
+        {"foo": "bar"},
+    ]
+
+@respx.mock
+@pytest.mark.django_db
+def test_download_latest_vouchers():
+    url = "https://tickets.europython.eu/api/v1/organizers/europython/events/ep2025/vouchers/"
+    data = orders_pages_generator(url)
+    respx.get(url).mock(return_value=next(data))
+    respx.get(url + "&page=2").mock(return_value=next(data))
+
+    pretix.download_latest_vouchers("ep2025")
+
+    pd = PretixData.objects.get(resource=PretixData.PretixResources.vouchers)
+    assert pd.resource == "vouchers"
+    assert pd.content == [
+        {"hello": "world"},
+        {"foo": "bar"},
+    ]