test(grants): cover voucher task and grant voucher email

Add tests for send_grant_voucher_email and create_and_send_voucher_to_grantee
(not confirmed, no user, existing speaker/grant voucher, co-speaker upgrade,
voucher on other conference), aligned with schedule voucher task tests.

Author: estyxx

backend/grants/tests/test_tasks.py

@@ -1,14 +1,22 @@
 from datetime import datetime, timezone
 from decimal import Decimal
-
 import pytest
+import time_machine
 
-from conferences.tests.factories import ConferenceFactory, DeadlineFactory
+from conferences.models.conference_voucher import ConferenceVoucher
+from conferences.tests.factories import (
+    ConferenceFactory,
+    ConferenceVoucherFactory,
+    DeadlineFactory,
+)
+from grants.models import Grant
 from grants.tasks import (
+    create_and_send_voucher_to_grantee,
     send_grant_reply_approved_email,
     send_grant_reply_rejected_email,
     send_grant_reply_waiting_list_email,
     send_grant_reply_waiting_list_update_email,
+    send_grant_voucher_email,
 )
 from grants.tests.factories import GrantFactory, GrantReimbursementFactory
 from users.tests.factories import UserFactory
@@ -380,7 +388,6 @@ def test_handle_grant_approved_travel_reply_sent(settings, sent_emails):
 
 
 def test_send_grant_approved_email_raises_for_no_reimbursements(settings) -> None:
-    """Verify error is raised when grant has no valid reimbursements."""
     from notifications.models import EmailTemplateIdentifier
     from notifications.tests.factories import EmailTemplateFactory
 
@@ -462,7 +469,189 @@ def test_send_grant_reply_waiting_list_update_email(settings, sent_emails):
 
 def test_send_grant_waiting_list_email_missing_deadline():
     grant = GrantFactory()
-    # No deadline created
 
     with pytest.raises(ValueError, match="missing grants_waiting_list_update deadline"):
         send_grant_reply_waiting_list_email(grant_id=grant.id)
+
+
+def test_send_grant_voucher_email(settings, sent_emails):
+    from notifications.models import EmailTemplateIdentifier
+    from notifications.tests.factories import EmailTemplateFactory
+
+    settings.FRONTEND_URL = "https://pycon.it"
+    user = UserFactory(
+        full_name="Marco Acierno",
+        email="marco@placeholder.it",
+        name="Marco",
+        username="marco",
+    )
+    grant = GrantFactory(user=user, status=Grant.Status.confirmed)
+    ConferenceVoucherFactory(
+        user=user,
+        conference=grant.conference,
+        voucher_type=ConferenceVoucher.VoucherType.GRANT,
+        voucher_code="GRANT99",
+    )
+
+    EmailTemplateFactory(
+        conference=grant.conference,
+        identifier=EmailTemplateIdentifier.grant_voucher_code,
+    )
+
+    with time_machine.travel("2020-10-10 10:00:00Z", tick=False):
+        send_grant_voucher_email(grant_id=grant.id)
+
+    emails_sent = sent_emails()
+    assert emails_sent.count() == 1
+
+    sent_email = emails_sent.first()
+    assert (
+        sent_email.email_template.identifier
+        == EmailTemplateIdentifier.grant_voucher_code
+    )
+    assert sent_email.email_template.conference == grant.conference
+    assert sent_email.recipient == user
+    assert sent_email.placeholders["user_name"] == "Marco Acierno"
+    assert sent_email.placeholders["conference_name"] == grant.conference.name.localize(
+        "en"
+    )
+    assert sent_email.placeholders["voucher_code"] == "GRANT99"
+    assert sent_email.placeholders["has_approved_accommodation"] is False
+    assert sent_email.placeholders["visa_page_link"] == "https://pycon.it/visa"
+
+    voucher = ConferenceVoucher.objects.get(
+        conference=grant.conference,
+        user=user,
+    )
+    assert voucher.voucher_email_sent_at == datetime(
+        2020, 10, 10, 10, 0, 0, tzinfo=timezone.utc
+    )
+
+
+def test_create_and_send_voucher_to_grantee(mocker):
+    mock_create = mocker.patch(
+        "conferences.vouchers.create_voucher", return_value={"id": 123}
+    )
+    mock_send_email = mocker.patch("grants.tasks.send_grant_voucher_email")
+
+    grant = GrantFactory(status=Grant.Status.confirmed)
+
+    create_and_send_voucher_to_grantee(grant_id=grant.id)
+
+    assert (
+        ConferenceVoucher.objects.for_conference(grant.conference)
+        .filter(
+            user=grant.user,
+            voucher_type=ConferenceVoucher.VoucherType.GRANT,
+        )
+        .exists()
+    )
+
+    mock_create.assert_called_once()
+    mock_send_email.delay.assert_called_once_with(grant_id=grant.id)
+
+
+def test_create_and_send_voucher_to_grantee_does_nothing_if_not_confirmed(mocker):
+    mock_create = mocker.patch("conferences.vouchers.create_voucher")
+    mock_send_email = mocker.patch("grants.tasks.send_grant_voucher_email")
+
+    grant = GrantFactory(status=Grant.Status.waiting_for_confirmation)
+
+    create_and_send_voucher_to_grantee(grant_id=grant.id)
+
+    mock_create.assert_not_called()
+    mock_send_email.delay.assert_not_called()
+
+
+def test_create_and_send_voucher_to_grantee_does_nothing_if_no_user(mocker):
+    mock_create = mocker.patch("conferences.vouchers.create_voucher")
+    mock_send_email = mocker.patch("grants.tasks.send_grant_voucher_email")
+
+    grant = GrantFactory(status=Grant.Status.confirmed)
+    Grant.objects.filter(pk=grant.pk).update(user_id=None)
+    grant.refresh_from_db()
+
+    create_and_send_voucher_to_grantee(grant_id=grant.id)
+
+    mock_create.assert_not_called()
+    mock_send_email.delay.assert_not_called()
+
+
+@pytest.mark.parametrize(
+    "voucher_type",
+    [
+        ConferenceVoucher.VoucherType.SPEAKER,
+        ConferenceVoucher.VoucherType.GRANT,
+    ],
+)
+def test_create_and_send_voucher_to_grantee_does_nothing_if_voucher_exists(
+    mocker, voucher_type
+):
+    mock_create = mocker.patch("conferences.vouchers.create_voucher")
+    mock_send_email = mocker.patch("grants.tasks.send_grant_voucher_email")
+
+    grant = GrantFactory(status=Grant.Status.confirmed)
+    ConferenceVoucherFactory(
+        conference=grant.conference,
+        user=grant.user,
+        voucher_type=voucher_type,
+    )
+
+    create_and_send_voucher_to_grantee(grant_id=grant.id)
+
+    mock_create.assert_not_called()
+    mock_send_email.delay.assert_not_called()
+
+
+def test_create_and_send_voucher_to_grantee_upgrades_co_speaker(mocker):
+    mock_create = mocker.patch("conferences.vouchers.create_voucher")
+    mock_send_email = mocker.patch("grants.tasks.send_grant_voucher_email")
+
+    grant = GrantFactory(status=Grant.Status.confirmed)
+    ConferenceVoucherFactory(
+        conference=grant.conference,
+        user=grant.user,
+        voucher_type=ConferenceVoucher.VoucherType.CO_SPEAKER,
+    )
+
+    create_and_send_voucher_to_grantee(grant_id=grant.id)
+
+    voucher = ConferenceVoucher.objects.get(
+        conference=grant.conference,
+        user=grant.user,
+    )
+    assert voucher.voucher_type == ConferenceVoucher.VoucherType.GRANT
+
+    mock_create.assert_not_called()
+    mock_send_email.delay.assert_called_once_with(grant_id=grant.id)
+
+
+def test_create_and_send_voucher_to_grantee_creates_when_voucher_on_other_conference(
+    mocker,
+):
+    mock_create = mocker.patch(
+        "conferences.vouchers.create_voucher", return_value={"id": 123}
+    )
+    mock_send_email = mocker.patch("grants.tasks.send_grant_voucher_email")
+
+    other_conference = ConferenceFactory()
+    grant = GrantFactory(status=Grant.Status.confirmed)
+    ConferenceVoucherFactory(
+        conference=other_conference,
+        user=grant.user,
+        voucher_type=ConferenceVoucher.VoucherType.SPEAKER,
+    )
+
+    create_and_send_voucher_to_grantee(grant_id=grant.id)
+
+    assert (
+        ConferenceVoucher.objects.for_conference(grant.conference)
+        .filter(
+            user=grant.user,
+            voucher_type=ConferenceVoucher.VoucherType.GRANT,
+        )
+        .exists()
+    )
+
+    mock_create.assert_called_once()
+    mock_send_email.delay.assert_called_once_with(grant_id=grant.id)