|
5 | 5 |
|
6 | 6 | from conferences.tests.factories import ConferenceFactory, DeadlineFactory |
7 | 7 | from grants.tasks import ( |
| 8 | + create_and_send_grant_voucher, |
8 | 9 | send_grant_reply_approved_email, |
9 | 10 | send_grant_reply_rejected_email, |
10 | 11 | send_grant_reply_waiting_list_email, |
@@ -466,3 +467,109 @@ def test_send_grant_waiting_list_email_missing_deadline(): |
466 | 467 |
|
467 | 468 | with pytest.raises(ValueError, match="missing grants_waiting_list_update deadline"): |
468 | 469 | send_grant_reply_waiting_list_email(grant_id=grant.id) |
| 470 | + |
| 471 | + |
| 472 | +def test_create_and_send_grant_voucher(mocker, sent_emails): |
| 473 | + from conferences.models.conference_voucher import ConferenceVoucher |
| 474 | + from notifications.models import EmailTemplateIdentifier |
| 475 | + from notifications.tests.factories import EmailTemplateFactory |
| 476 | + |
| 477 | + mock_create_voucher = mocker.patch( |
| 478 | + "grants.tasks.create_conference_voucher", |
| 479 | + ) |
| 480 | + mock_conference_voucher = mocker.MagicMock() |
| 481 | + mock_conference_voucher.id = 123 |
| 482 | + mock_create_voucher.return_value = mock_conference_voucher |
| 483 | + |
| 484 | + mock_send_email = mocker.patch( |
| 485 | + "grants.tasks.send_conference_voucher_email", |
| 486 | + ) |
| 487 | + |
| 488 | + user = UserFactory( |
| 489 | + full_name="Marco Acierno", |
| 490 | + email="marco@placeholder.it", |
| 491 | + ) |
| 492 | + grant = GrantFactory(user=user) |
| 493 | + |
| 494 | + EmailTemplateFactory( |
| 495 | + conference=grant.conference, |
| 496 | + identifier=EmailTemplateIdentifier.voucher_code, |
| 497 | + ) |
| 498 | + |
| 499 | + create_and_send_grant_voucher(grant_id=grant.id) |
| 500 | + |
| 501 | + mock_create_voucher.assert_called_once_with( |
| 502 | + conference=grant.conference, |
| 503 | + user=user, |
| 504 | + voucher_type=ConferenceVoucher.VoucherType.GRANT, |
| 505 | + ) |
| 506 | + mock_send_email.delay.assert_called_once_with(conference_voucher_id=123) |
| 507 | + |
| 508 | + |
| 509 | +def test_create_and_send_grant_voucher_user_already_has_voucher(mocker): |
| 510 | + from conferences.models.conference_voucher import ConferenceVoucher |
| 511 | + from conferences.tests.factories import ConferenceVoucherFactory |
| 512 | + |
| 513 | + mock_create_voucher = mocker.patch( |
| 514 | + "grants.tasks.create_conference_voucher", |
| 515 | + ) |
| 516 | + mock_send_email = mocker.patch( |
| 517 | + "grants.tasks.send_conference_voucher_email", |
| 518 | + ) |
| 519 | + |
| 520 | + user = UserFactory( |
| 521 | + full_name="Marco Acierno", |
| 522 | + email="marco@placeholder.it", |
| 523 | + ) |
| 524 | + grant = GrantFactory(user=user) |
| 525 | + |
| 526 | + # Create an existing voucher for this user and conference |
| 527 | + ConferenceVoucherFactory( |
| 528 | + conference=grant.conference, |
| 529 | + user=user, |
| 530 | + voucher_type=ConferenceVoucher.VoucherType.SPEAKER, |
| 531 | + ) |
| 532 | + |
| 533 | + create_and_send_grant_voucher(grant_id=grant.id) |
| 534 | + |
| 535 | + # Should not create a new voucher |
| 536 | + mock_create_voucher.assert_not_called() |
| 537 | + mock_send_email.delay.assert_not_called() |
| 538 | + |
| 539 | + |
| 540 | +def test_create_and_send_grant_voucher_upgrades_co_speaker_voucher(mocker): |
| 541 | + from conferences.models.conference_voucher import ConferenceVoucher |
| 542 | + from conferences.tests.factories import ConferenceVoucherFactory |
| 543 | + |
| 544 | + mock_create_voucher = mocker.patch( |
| 545 | + "grants.tasks.create_conference_voucher", |
| 546 | + ) |
| 547 | + mock_send_email = mocker.patch( |
| 548 | + "grants.tasks.send_conference_voucher_email", |
| 549 | + ) |
| 550 | + |
| 551 | + user = UserFactory( |
| 552 | + full_name="Marco Acierno", |
| 553 | + email="marco@placeholder.it", |
| 554 | + ) |
| 555 | + grant = GrantFactory(user=user) |
| 556 | + |
| 557 | + # Create an existing co-speaker voucher for this user and conference |
| 558 | + existing_voucher = ConferenceVoucherFactory( |
| 559 | + conference=grant.conference, |
| 560 | + user=user, |
| 561 | + voucher_type=ConferenceVoucher.VoucherType.CO_SPEAKER, |
| 562 | + ) |
| 563 | + |
| 564 | + create_and_send_grant_voucher(grant_id=grant.id) |
| 565 | + |
| 566 | + # Should not create a new voucher but should upgrade the existing one |
| 567 | + mock_create_voucher.assert_not_called() |
| 568 | + |
| 569 | + existing_voucher.refresh_from_db() |
| 570 | + assert existing_voucher.voucher_type == ConferenceVoucher.VoucherType.GRANT |
| 571 | + |
| 572 | + # Should send email with the upgraded voucher |
| 573 | + mock_send_email.delay.assert_called_once_with( |
| 574 | + conference_voucher_id=existing_voucher.id |
| 575 | + ) |
0 commit comments