55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
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
|
|
|
|
TEST_EMAIL = "discoman@example.com"
|
|
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):
|
|
# Mock successful Mailgun API response
|
|
mock_post.return_value.status_code = 200
|
|
|
|
self.browser.get(self.live_server_url)
|
|
self.browser.find_element(By.CSS_SELECTOR, "input[name=email]").send_keys(
|
|
TEST_EMAIL, Keys.ENTER
|
|
)
|
|
|
|
self.wait_for(
|
|
lambda: self.assertIn(
|
|
"Check your email",
|
|
self.browser.find_element(By.CSS_SELECTOR, "body").text,
|
|
)
|
|
)
|
|
|
|
if self.test_server:
|
|
return
|
|
|
|
# Verify Mailgun API was called
|
|
self.assertEqual(mock_post.call_count, 1)
|
|
call_kwargs = mock_post.call_args.kwargs
|
|
|
|
# Check email data
|
|
self.assertEqual(call_kwargs['data']['to'], TEST_EMAIL)
|
|
self.assertEqual(call_kwargs['data']['subject'], SUBJECT)
|
|
|
|
# Extract magic link URL from email body
|
|
email_body = call_kwargs['data']['text']
|
|
self.assertIn("Use this magic link to login to your Dashboard", email_body)
|
|
url_search = re.search(r"http://.+/.+$", email_body)
|
|
if not url_search:
|
|
self.fail(f"Could not find url in email body:\n{email_body}")
|
|
url = url_search.group(0)
|
|
self.assertIn(self.live_server_url, url)
|
|
|
|
self.browser.get(url)
|
|
|
|
self.wait_to_be_logged_in(email=TEST_EMAIL)
|
|
|
|
self.browser.find_element(By.CSS_SELECTOR, "#id_logout").click()
|
|
|
|
self.wait_to_be_logged_out(email=TEST_EMAIL)
|