mirror of
https://github.com/zulip/zulip.git
synced 2025-11-03 21:43:21 +00:00
ad2ca0e668
Previously if `test_forward_address_details` failed, the file created when setting the `forward_address` may not have been removed, which would then cause an `EmailNotDeliveredException` to be raised when then creating a new user in the dev environment. Wraps the test in a try block, with a finally block for the call to remove the file.
51 lines
2.3 KiB
Python
51 lines
2.3 KiB
Python
import os
|
|
from unittest import mock
|
|
|
|
from django.conf import settings
|
|
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zproject.email_backends import get_forward_address
|
|
|
|
|
|
class EmailLogTest(ZulipTestCase):
|
|
def test_generate_and_clear_email_log(self) -> None:
|
|
with self.settings(EMAIL_BACKEND="zproject.email_backends.EmailLogBackEnd"), mock.patch(
|
|
"zproject.email_backends.EmailLogBackEnd._do_send_messages", lambda *args: 1
|
|
), self.assertLogs(level="INFO") as m, self.settings(DEVELOPMENT_LOG_EMAILS=True):
|
|
result = self.client_get("/emails/generate/")
|
|
self.assertEqual(result.status_code, 302)
|
|
self.assertIn("emails", result["Location"])
|
|
|
|
result = self.client_get("/emails/")
|
|
self.assert_in_success_response(["All the emails sent in the Zulip"], result)
|
|
|
|
result = self.client_get("/emails/clear/")
|
|
self.assertEqual(result.status_code, 302)
|
|
result = self.client_get(result["Location"])
|
|
self.assertIn("manually generate most of the emails by clicking", str(result.content))
|
|
output_log = (
|
|
"INFO:root:Emails sent in development are available at http://testserver/emails"
|
|
)
|
|
self.assertEqual(m.output, [output_log for i in range(15)])
|
|
|
|
def test_forward_address_details(self) -> None:
|
|
try:
|
|
forward_address = "forward-to@example.com"
|
|
result = self.client_post("/emails/", {"forward_address": forward_address})
|
|
self.assert_json_success(result)
|
|
|
|
self.assertEqual(get_forward_address(), forward_address)
|
|
|
|
with self.settings(EMAIL_BACKEND="zproject.email_backends.EmailLogBackEnd"), mock.patch(
|
|
"zproject.email_backends.EmailLogBackEnd._do_send_messages", lambda *args: 1
|
|
):
|
|
result = self.client_get("/emails/generate/")
|
|
self.assertEqual(result.status_code, 302)
|
|
self.assertIn("emails", result["Location"])
|
|
result = self.client_get(result["Location"])
|
|
self.assert_in_success_response([forward_address], result)
|
|
# Remove this file, even if the test fails, so that it does
|
|
# not impact the state of the development environment.
|
|
finally:
|
|
os.remove(settings.FORWARD_ADDRESS_CONFIG_FILE)
|