fix: catch SMTPException when sending secret edit link
This commit is contained in:
parent
76ed2200cf
commit
9cad054746
2 changed files with 45 additions and 11 deletions
|
@ -1,10 +1,12 @@
|
||||||
import re
|
import re
|
||||||
|
from smtplib import SMTPException
|
||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from django.core.signing import get_cookie_signer
|
from django.core.signing import get_cookie_signer
|
||||||
from playwright.sync_api import expect
|
from playwright.sync_api import expect
|
||||||
|
|
||||||
from umap.models import DataLayer
|
from umap.models import DataLayer, Map
|
||||||
|
|
||||||
from ..base import DataLayerFactory
|
from ..base import DataLayerFactory
|
||||||
|
|
||||||
|
@ -152,7 +154,9 @@ def test_can_change_perms_after_create(tilelayer, live_server, page):
|
||||||
expect(option).to_have_text("Inherit")
|
expect(option).to_have_text("Inherit")
|
||||||
|
|
||||||
|
|
||||||
def test_alert_message_after_create(tilelayer, live_server, page):
|
def test_alert_message_after_create(
|
||||||
|
tilelayer, live_server, page, monkeypatch, settings
|
||||||
|
):
|
||||||
page.goto(f"{live_server.url}/en/map/new")
|
page.goto(f"{live_server.url}/en/map/new")
|
||||||
save = page.get_by_role("button", name="Save")
|
save = page.get_by_role("button", name="Save")
|
||||||
expect(save).to_be_visible()
|
expect(save).to_be_visible()
|
||||||
|
@ -160,6 +164,7 @@ def test_alert_message_after_create(tilelayer, live_server, page):
|
||||||
expect(alert).to_be_hidden()
|
expect(alert).to_be_hidden()
|
||||||
with page.expect_response(re.compile(r".*/map/create/")):
|
with page.expect_response(re.compile(r".*/map/create/")):
|
||||||
save.click()
|
save.click()
|
||||||
|
new_map = Map.objects.last()
|
||||||
expect(alert).to_be_visible()
|
expect(alert).to_be_visible()
|
||||||
expect(
|
expect(
|
||||||
alert.get_by_text(
|
alert.get_by_text(
|
||||||
|
@ -169,3 +174,28 @@ def test_alert_message_after_create(tilelayer, live_server, page):
|
||||||
).to_be_visible()
|
).to_be_visible()
|
||||||
expect(alert.get_by_role("button", name="Copy")).to_be_visible()
|
expect(alert.get_by_role("button", name="Copy")).to_be_visible()
|
||||||
expect(alert.get_by_role("button", name="Send me the link")).to_be_visible()
|
expect(alert.get_by_role("button", name="Send me the link")).to_be_visible()
|
||||||
|
alert.get_by_placeholder("Email").fill("foo@bar.com")
|
||||||
|
with patch("umap.views.send_mail") as patched:
|
||||||
|
with page.expect_response(re.compile("/en/map/.*/send-edit-link/")):
|
||||||
|
alert.get_by_role("button", name="Send me the link").click()
|
||||||
|
assert patched.called
|
||||||
|
patched.assert_called_with(
|
||||||
|
"The uMap edit link for your map: Untitled map",
|
||||||
|
f"Here is your secret edit link: {new_map.get_anonymous_edit_url()}",
|
||||||
|
"test@test.org",
|
||||||
|
["foo@bar.com"],
|
||||||
|
fail_silently=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_email_sending_error_are_catched(tilelayer, page, live_server):
|
||||||
|
page.goto(f"{live_server.url}/en/map/new")
|
||||||
|
alert = page.locator(".umap-alert")
|
||||||
|
with page.expect_response(re.compile(r".*/map/create/")):
|
||||||
|
page.get_by_role("button", name="Save").click()
|
||||||
|
alert.get_by_placeholder("Email").fill("foo@bar.com")
|
||||||
|
with patch("umap.views.send_mail", side_effect=SMTPException) as patched:
|
||||||
|
with page.expect_response(re.compile("/en/map/.*/send-edit-link/")):
|
||||||
|
alert.get_by_role("button", name="Send me the link").click()
|
||||||
|
assert patched.called
|
||||||
|
expect(alert.get_by_text("Can't send email to foo@bar.com")).to_be_visible()
|
||||||
|
|
|
@ -9,6 +9,7 @@ from datetime import datetime, timedelta
|
||||||
from http.client import InvalidURL
|
from http.client import InvalidURL
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
from smtplib import SMTPException
|
||||||
from urllib.error import HTTPError, URLError
|
from urllib.error import HTTPError, URLError
|
||||||
from urllib.parse import quote, quote_plus, urlparse
|
from urllib.parse import quote, quote_plus, urlparse
|
||||||
from urllib.request import Request, build_opener
|
from urllib.request import Request, build_opener
|
||||||
|
@ -836,16 +837,19 @@ class SendEditLink(FormLessEditMixin, FormView):
|
||||||
return HttpResponseBadRequest("Invalid")
|
return HttpResponseBadRequest("Invalid")
|
||||||
link = self.object.get_anonymous_edit_url()
|
link = self.object.get_anonymous_edit_url()
|
||||||
|
|
||||||
send_mail(
|
subject = _(
|
||||||
_(
|
"The uMap edit link for your map: %(map_name)s"
|
||||||
"The uMap edit link for your map: %(map_name)s"
|
% {"map_name": self.object.name}
|
||||||
% {"map_name": self.object.name}
|
|
||||||
),
|
|
||||||
_("Here is your secret edit link: %(link)s" % {"link": link}),
|
|
||||||
settings.DEFAULT_FROM_EMAIL,
|
|
||||||
[email],
|
|
||||||
fail_silently=False,
|
|
||||||
)
|
)
|
||||||
|
body = _("Here is your secret edit link: %(link)s" % {"link": link})
|
||||||
|
try:
|
||||||
|
send_mail(
|
||||||
|
subject, body, settings.DEFAULT_FROM_EMAIL, [email], fail_silently=False
|
||||||
|
)
|
||||||
|
except SMTPException:
|
||||||
|
return simple_json_response(
|
||||||
|
error=_("Can't send email to %(email)s" % {"email": email})
|
||||||
|
)
|
||||||
return simple_json_response(
|
return simple_json_response(
|
||||||
info=_("Email sent to %(email)s" % {"email": email})
|
info=_("Email sent to %(email)s" % {"email": email})
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in a new issue