Question

我正在尝试使用django-rest-auth对电子邮件确认视图进行测试。这是我所拥有的：

def test_verify_email(self):
    # Verify email address
    username = 'userTest'
    payload = {
        'email': 'test@example.com',
        'password1': 'TestpassUltra1',
        'password2': 'TestpassUltra1',
        'username': username,
    }
    res = self.client.post(REGISTER_USER_URL, payload)

    self.assertEqual(res.status_code, status.HTTP_201_CREATED)
    user = get_user_model().objects.get(email='test@example.com')
    # TODO retrieve the confirmation key from the user
    resp = self.client.post(VERIFY_USER_URL, {'key': ''})
    self.assertEqual(resp.status_code, status.HTTP_200_OK)

self.client.post(REGISTER_USER_URL, payload)将发送一封包含确认代码的电子邮件，虽然我知道我可以在代码中使用django.core.mail.outbox来检索该代码，但我不想这样做，因为我必须解析电子邮件内容查找确认代码（如果电子邮件发生更改，这可能会杀死我的测试）。我找不到此代码存储在数据库中的任何位置，它似乎确实只存在于所发送电子邮件的正文中。

我的问题是：是否可以在不解析电子邮件的情况下在我的测试中恢复此验证码？我只想检索它以启动我的self.client.post(VERIFY_USER_URL, {'key': ''})。

以下是电子邮件内容的示例：

Hello from example.com!

You're receiving this e-mail from NomadSpeed because user detro1 has given yours as an e-mail address to connect their account.

To confirm this is correct, go to http://127.0.0.1:8000/registration/account-confirm-email/Mg:1hmqdS:J-2jGV028nd4qZQ3lPmFgXGFhsM/

Thank you from example.com!
example.com

我需要的是Mg:1hmqdS:J-2jGV028nd4qZQ3lPmFgXGFhsM。谢谢。

Answer 1

我认为解决此问题的最佳方法是使用正则表达式来匹配预期的链接，并从中解析出所需的位。公平地担心，对电子邮件所做的更改可能会破坏测试，但是在这种情况下，您正在测试电子邮件中的 link 是否有效，并且如果此更改以某种方式出现，则可能会破坏您的测试。如果您更改了其他一些文本，例如问候语，介绍等，则不会影响链接和令牌的正则表达式。

无论如何，这就是我要如何构造该测试：

import re

def test_verify_email(self):
    # Verify email address
    username = 'userTest'
    payload = {
        'email': 'test@example.com',
        'password1': 'TestpassUltra1',
        'password2': 'TestpassUltra1',
        'username': username,
    }
    response = self.client.post(REGISTER_USER_URL, payload)
    self.assertEqual(response.status_code, status.HTTP_201_CREATED)
    user = get_user_model().objects.get(email='test@example.com')

    # Get token from email
    token_regex = r"registration\/account-confirm-email\/([A-Za-z0-9:\-]+)\/"
    email_content = django.core.mail.outbox[0].body
    match = re.search(token_regex, email_content)
    assert match.groups(), "Could not find the token in the email" # You might want to use some other way to raise an error for this
    token = match.group(1)

    # Verify 
    response = self.client.post(VERIFY_USER_URL, {'key': token})
    self.assertEqual(response.status_code, status.HTTP_200_OK)

您甚至可以断言该链接路径的正确性，因此，如果有人更改了该链接，您将无法通过测试以表明可能会损坏。

如何测试电子邮件确认？

1 个答案: