diff --git a/src/functional_tests/test_login.py b/src/functional_tests/test_login.py index 047fa68..ca52ec3 100644 --- a/src/functional_tests/test_login.py +++ b/src/functional_tests/test_login.py @@ -1,9 +1,11 @@ import re + from unittest.mock import patch from selenium.webdriver.common.by import By from selenium.webdriver.common.keys import Keys from .base import FunctionalTest +from apps.lyric.tasks import send_login_email_task TEST_EMAIL = "discoman@example.com" @@ -11,8 +13,10 @@ SUBJECT = "A magic login link to your Dashboard" class LoginTest(FunctionalTest): - @patch('apps.lyric.views.requests.post') - def test_login_using_magic_link(self, mock_post): + @patch('apps.lyric.tasks.requests.post') + @patch('apps.lyric.views.send_login_email_task.delay', + side_effect=send_login_email_task) + def test_login_using_magic_link(self, mock_delay, mock_post): # Mock successful Mailgun API response mock_post.return_value.status_code = 200