tests

Author: marcoacierno

backend/visa/tests/test_tasks.py

@@ -1,6 +1,11 @@
 from uuid import uuid4
 import requests
-from visa.models import InvitationLetterRequestStatus
+from grants.tests.factories import GrantFactory
+from grants.models import Grant
+from visa.models import (
+    InvitationLetterDocumentInclusionPolicy,
+    InvitationLetterRequestStatus,
+)
 import pytest
 from django.test import override_settings
 from visa.tasks import (
@@ -18,8 +23,35 @@
 pytestmark = pytest.mark.django_db
 
 
+@pytest.fixture()
+def mock_ticket_present(requests_mock):
+    def _mock_ticket_present(request):
+        requests_mock.get(
+            "https://example.com/ticket.pdf",
+            content=open("visa/tests/fixtures/sample-ticket.pdf", "rb").read(),
+        )
+        requests_mock.get(
+            f"https://pretix/api/organizers/base-pretix-organizer-id/events/base-pretix-event-id/tickets/attendee-tickets/?attendee_email={request.requester.email}",
+            json=[
+                {
+                    "item": {
+                        "admission": False,
+                    }
+                },
+                {
+                    "item": {
+                        "admission": True,
+                    },
+                    "downloads": [{"url": "https://example.com/ticket.pdf"}],
+                },
+            ],
+        )
+
+    return _mock_ticket_present
+
+
 @override_settings(PRETIX_API="https://pretix/api/")
-def test_process_invitation_letter_request(requests_mock):
+def test_process_invitation_letter_request(requests_mock, mock_ticket_present):
     config = InvitationLetterConferenceConfigFactory()
     InvitationLetterAssetFactory(
         invitation_letter_conference_config=config, identifier="test"
@@ -48,6 +80,7 @@ def test_process_invitation_letter_request(requests_mock):
     request = InvitationLetterRequestFactory(
         conference=config.conference, nationality="Italian"
     )
+    mock_ticket_present(request)
 
     requests_mock.get(
         "https://example.com/ticket.pdf",
@@ -85,6 +118,94 @@ def test_process_invitation_letter_request(requests_mock):
     assert output.pages[3].extract_text() == "Thisisasampleticket pdf"
 
 
+@pytest.mark.parametrize(
+    "grant_approved_type",
+    [None, Grant.ApprovedType.ticket_only, Grant.ApprovedType.ticket_travel],
+)
+@override_settings(PRETIX_API="https://pretix/api/")
+def test_process_invitation_letter_request_skips_docs_with_no_accommodation(
+    mock_ticket_present, grant_approved_type
+):
+    config = InvitationLetterConferenceConfigFactory()
+    InvitationLetterDocumentFactory(
+        invitation_letter_conference_config=config,
+        document=None,
+        dynamic_document={
+            "header": {"content": "header", "margin": "0", "align": "top-left"},
+            "footer": {"content": "footer", "margin": "0", "align": "bottom-left"},
+            "page_layout": {"margin": "0"},
+            "pages": [
+                {"content": "accommodation details"},
+            ],
+        },
+        inclusion_policy=InvitationLetterDocumentInclusionPolicy.GRANT_ACCOMMODATION,
+    )
+
+    request = InvitationLetterRequestFactory(
+        conference=config.conference, nationality="Italian"
+    )
+    mock_ticket_present(request)
+
+    if grant_approved_type:
+        GrantFactory(
+            conference=config.conference,
+            user=request.requester,
+            approved_type=grant_approved_type,
+        )
+
+    process_invitation_letter_request(invitation_letter_request_id=request.id)
+
+    request.refresh_from_db()
+
+    assert request.status == InvitationLetterRequestStatus.PROCESSED
+
+    output = PdfReader(request.invitation_letter.open())
+    assert output.get_num_pages() == 1
+    assert output.pages[0].extract_text() == "Thisisasampleticket pdf"
+
+
+@override_settings(PRETIX_API="https://pretix/api/")
+def test_process_invitation_letter_request_with_doc_only_for_accommodation(
+    mock_ticket_present,
+):
+    config = InvitationLetterConferenceConfigFactory()
+    InvitationLetterDocumentFactory(
+        invitation_letter_conference_config=config,
+        document=None,
+        dynamic_document={
+            "header": {"content": "header", "margin": "0", "align": "top-left"},
+            "footer": {"content": "footer", "margin": "0", "align": "bottom-left"},
+            "page_layout": {"margin": "0"},
+            "pages": [
+                {"content": "accommodation details"},
+            ],
+        },
+        inclusion_policy=InvitationLetterDocumentInclusionPolicy.GRANT_ACCOMMODATION,
+    )
+
+    request = InvitationLetterRequestFactory(
+        conference=config.conference, nationality="Italian"
+    )
+    mock_ticket_present(request)
+
+    GrantFactory(
+        conference=config.conference,
+        user=request.requester,
+        approved_type=Grant.ApprovedType.ticket_travel_accommodation,
+    )
+
+    process_invitation_letter_request(invitation_letter_request_id=request.id)
+
+    request.refresh_from_db()
+
+    assert request.status == InvitationLetterRequestStatus.PROCESSED
+
+    output = PdfReader(request.invitation_letter.open())
+    assert output.get_num_pages() == 2
+    assert output.pages[0].extract_text() == "accommodation details \nheader\nfooter"
+    assert output.pages[1].extract_text() == "Thisisasampleticket pdf"
+
+
 @override_settings(PRETIX_API="https://pretix/api/")
 def test_process_invitation_letter_request_handles_generating_ticket_pdfs(
     requests_mock, mocker