|
| 1 | +from django.urls import reverse |
| 2 | +from django.contrib.admin.sites import AdminSite |
| 3 | +from visa.models import InvitationLetterOrganizerConfig, InvitationLetterRequest |
| 4 | +from visa.admin import InvitationLetterDocumentInline, InvitationLetterRequestAdmin |
| 5 | +import pytest |
| 6 | + |
| 7 | +from visa.tests.factories import ( |
| 8 | + InvitationLetterDocumentFactory, |
| 9 | + InvitationLetterOrganizerConfigFactory, |
| 10 | + InvitationLetterRequestFactory, |
| 11 | +) |
| 12 | + |
| 13 | +pytestmark = pytest.mark.django_db |
| 14 | + |
| 15 | + |
| 16 | +def test_edit_dynamic_document_view(rf, admin_user): |
| 17 | + admin = InvitationLetterDocumentInline( |
| 18 | + parent_model=InvitationLetterOrganizerConfig, admin_site=AdminSite() |
| 19 | + ) |
| 20 | + |
| 21 | + config = InvitationLetterOrganizerConfigFactory() |
| 22 | + document = InvitationLetterDocumentFactory( |
| 23 | + invitation_letter_organizer_config=config, |
| 24 | + document=None, |
| 25 | + ) |
| 26 | + |
| 27 | + request = rf.get("/") |
| 28 | + request.user = admin_user |
| 29 | + response = admin.edit_dynamic_document_view(request, config.id, document.id) |
| 30 | + |
| 31 | + assert response.status_code == 200 |
| 32 | + assert response.template_name == "astro/invitation-letter-document-builder.html" |
| 33 | + assert response.context_data["arguments"]["document_id"] == document.id |
| 34 | + assert response.context_data["arguments"]["breadcrumbs"] |
| 35 | + |
| 36 | + |
| 37 | +def test_edit_dynamic_document_button(): |
| 38 | + admin = InvitationLetterDocumentInline( |
| 39 | + parent_model=InvitationLetterOrganizerConfig, admin_site=AdminSite() |
| 40 | + ) |
| 41 | + |
| 42 | + config = InvitationLetterOrganizerConfigFactory() |
| 43 | + document = InvitationLetterDocumentFactory( |
| 44 | + invitation_letter_organizer_config=config, |
| 45 | + document=None, |
| 46 | + ) |
| 47 | + |
| 48 | + html = admin.edit_dynamic_document(document) |
| 49 | + |
| 50 | + url = reverse( |
| 51 | + "admin:edit_dynamic_document", |
| 52 | + kwargs={ |
| 53 | + "config_id": config.id, |
| 54 | + "document_id": document.id, |
| 55 | + }, |
| 56 | + ) |
| 57 | + assert html == f'<a href="{url}">Edit</a>' |
| 58 | + |
| 59 | + |
| 60 | +def test_edit_dynamic_document_button_is_empty_for_static_docs(): |
| 61 | + admin = InvitationLetterDocumentInline( |
| 62 | + parent_model=InvitationLetterOrganizerConfig, admin_site=AdminSite() |
| 63 | + ) |
| 64 | + |
| 65 | + config = InvitationLetterOrganizerConfigFactory() |
| 66 | + document = InvitationLetterDocumentFactory( |
| 67 | + invitation_letter_organizer_config=config |
| 68 | + ) |
| 69 | + |
| 70 | + html = admin.edit_dynamic_document(document) |
| 71 | + |
| 72 | + assert html == "" |
| 73 | + |
| 74 | + |
| 75 | +def test_invitation_letter_request_admin(): |
| 76 | + admin = InvitationLetterRequestAdmin( |
| 77 | + model=InvitationLetterRequest, admin_site=AdminSite() |
| 78 | + ) |
| 79 | + |
| 80 | + invitation_letter_request = InvitationLetterRequestFactory() |
| 81 | + |
| 82 | + assert 'name="_process_now"' in admin.process_now(invitation_letter_request) |
| 83 | + assert "Generate the invitation letter first" == admin.send_via_email( |
| 84 | + invitation_letter_request |
| 85 | + ) |
| 86 | + |
| 87 | + |
| 88 | +def test_invitation_letter_request_admin_post_processing_redirects_to_changelist( |
| 89 | + rf, admin_user |
| 90 | +): |
| 91 | + admin = InvitationLetterRequestAdmin( |
| 92 | + model=InvitationLetterRequest, admin_site=AdminSite() |
| 93 | + ) |
| 94 | + |
| 95 | + invitation_letter_request = InvitationLetterRequestFactory() |
| 96 | + request = rf.post("/") |
| 97 | + request.user = admin_user |
| 98 | + request.POST = {"_process_now": "1"} |
| 99 | + response = admin.response_post_save_change(request, invitation_letter_request) |
| 100 | + |
| 101 | + assert response.status_code == 302 |
| 102 | + assert response.url == reverse( |
| 103 | + "admin:visa_invitationletterrequest_change", args=[invitation_letter_request.id] |
| 104 | + ) |
0 commit comments