tests

Author: marcoacierno

backend/association_membership/handlers/pretix/api.py

@@ -32,13 +32,13 @@ def run_request(
         if qs:
             url = f"{url}?" + "&".join([f"{key}={value}" for key, value in qs.items()])
 
-        response = getattr(requests, method)(url, headers=headers)
-        response.raise_for_status()
-        return response
+        return getattr(requests, method)(url, headers=headers)
 
     def _request(self, endpoint: str, **kwargs):
         url = f"{self.base_url}/{endpoint}/"
-        return self.run_request(url, **kwargs)
+        response = self.run_request(url, **kwargs)
+        response.raise_for_status()
+        return response
 
     def get_order_data(self, order_code: str) -> dict:
         response = self._request(f"orders/{order_code}")

backend/organizers/tests/factories.py

@@ -9,4 +9,4 @@ class Meta:
         model = Organizer
 
     name = factory.Faker("word")
-    slug = factory.Faker("text")
+    slug = factory.Faker("word")

backend/visa/admin.py

@@ -110,7 +110,7 @@ def response_post_save_change(self, request, obj):
 
 class InvitationLetterAssetInline(admin.TabularInline):
     fields = (
-        "handle",
+        "identifier",
         "image",
     )
     model = InvitationLetterAsset

backend/visa/migrations/0007_rename_field_to_identifier.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.4 on 2024-12-20 23:40
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('visa', '0006_invitationletterdocument_name_and_more'),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name='invitationletterasset',
+            old_name='handle',
+            new_name='identifier',
+        ),
+    ]

backend/visa/models.py

@@ -208,7 +208,7 @@ class InvitationLetterAsset(TimeStampedModel):
         related_name="assets",
         verbose_name=_("invitation letter organizer config"),
     )
-    handle = models.CharField(_("handle"), max_length=100)
+    identifier = models.CharField(_("identifier"), max_length=100)
     image = models.ImageField(
         _("image"),
         upload_to="invitation_letters_assets/",
@@ -218,4 +218,4 @@ class InvitationLetterAsset(TimeStampedModel):
     )
 
     def __str__(self):
-        return f"Asset: {self.handle}"
+        return f"Asset: {self.identifier}"

backend/visa/tasks.py

@@ -83,6 +83,7 @@ def process_invitation_letter_request(*, invitation_letter_request_id: int):
 
     pretix_ticket = download_pretix_ticket(invitation_letter_request)
     merger.append(pretix_ticket)
+    merger.compress_identical_objects(remove_identicals=True, remove_orphans=True)
 
     with tempfile.NamedTemporaryFile() as invitation_letter_file:
         merger.write(invitation_letter_file)
@@ -163,13 +164,16 @@ def download_pretix_ticket(invitation_letter_request):
     ticket_url = attendee_ticket["downloads"][0]["url"]
 
     attempts = 0
-    while attempts < 3:
+
+    while True:
         attempts += 1
 
         response = pretix_api.run_request(ticket_url)
-        if response.status_code == 409:
+        if response.status_code == 409 and attempts <= 3:
             time.sleep(2 * attempts)
-        else:
-            response.raise_for_status()
+            continue
+
+        response.raise_for_status()
+        break
 
     return io.BytesIO(response.content)

backend/visa/templatetags/invitation_letter_asset.py

@@ -5,11 +5,11 @@
 
 
 @register.simple_tag(takes_context=True)
-def invitation_letter_asset(context, asset_handle: str, **kwargs):
+def invitation_letter_asset(context, identifier: str, **kwargs):
     config = context["config"]
     assets = config.assets.all()
     matching_asset = next(
-        (asset for asset in assets if asset.handle == asset_handle), None
+        (asset for asset in assets if asset.identifier == identifier), None
     )
 
     width = kwargs.get("width", None)

backend/visa/tests/__init__.py

@@ -0,0 +1,57 @@
+from pathlib import Path
+from users.tests.factories import UserFactory
+from conferences.tests.factories import ConferenceFactory
+from organizers.tests.factories import OrganizerFactory
+from visa.models import (
+    InvitationLetterAsset,
+    InvitationLetterDocument,
+    InvitationLetterRequest,
+    InvitationLetterOrganizerConfig,
+)
+import factory
+import factory.fuzzy
+from factory.django import DjangoModelFactory
+
+
+class InvitationLetterRequestFactory(DjangoModelFactory):
+    conference = factory.SubFactory(ConferenceFactory)
+    requester = factory.SubFactory(UserFactory)
+    date_of_birth = factory.Faker("date")
+    full_name = factory.Faker("name")
+    nationality = factory.Faker("country")
+    address = factory.Faker("address")
+    passport_number = factory.Faker("ssn")
+    embassy_name = factory.Faker("company")
+
+    class Meta:
+        model = InvitationLetterRequest
+
+
+class InvitationLetterOrganizerConfigFactory(DjangoModelFactory):
+    organizer = factory.SubFactory(OrganizerFactory)
+
+    class Meta:
+        model = InvitationLetterOrganizerConfig
+
+
+class InvitationLetterDocumentFactory(DjangoModelFactory):
+    invitation_letter_organizer_config = factory.SubFactory(
+        InvitationLetterOrganizerConfigFactory
+    )
+    document = factory.django.FileField(
+        from_path=Path(__file__).parent / "fixtures" / "sample-pdf.pdf"
+    )
+
+    class Meta:
+        model = InvitationLetterDocument
+
+
+class InvitationLetterAssetFactory(DjangoModelFactory):
+    invitation_letter_organizer_config = factory.SubFactory(
+        InvitationLetterOrganizerConfigFactory
+    )
+    identifier = factory.Faker("md5")
+    image = factory.django.ImageField()
+
+    class Meta:
+        model = InvitationLetterAsset