Improve existing tests to check for alert notifications

This commit is contained in:
Baptiste Jonglez 2020-05-18 13:27:13 +02:00
parent 9fa00cb201
commit 3d7e6b212e

View file

@ -142,10 +142,15 @@ class BudgetTestCase(IhatemoneyTestCase):
self.login("raclette")
self.post_project("raclette")
self.client.post(
"/raclette/invite", data={"emails": "zorglub@notmyidea.org"}
resp = self.client.post(
"/raclette/invite",
data={"emails": "zorglub@notmyidea.org"},
follow_redirects=True,
)
# success notification
self.assertIn("Your invitations have been sent", resp.data.decode("utf-8"))
self.assertEqual(len(outbox), 2)
self.assertEqual(outbox[0].recipients, ["raclette@notmyidea.org"])
self.assertEqual(outbox[1].recipients, ["zorglub@notmyidea.org"])
@ -226,7 +231,15 @@ class BudgetTestCase(IhatemoneyTestCase):
self.create_project("raclette")
# Get password resetting link from mail
with self.app.mail.record_messages() as outbox:
self.client.post("/password-reminder", data={"id": "raclette"})
resp = self.client.post(
"/password-reminder", data={"id": "raclette"}, follow_redirects=True
)
# Check that we are redirected to the right page
self.assertIn(
"A link to reset your password has been sent to you",
resp.data.decode("utf-8"),
)
# Check that an email was sent
self.assertEqual(len(outbox), 1)
url_start = outbox[0].body.find("You can reset it here: ") + 23
url_end = outbox[0].body.find(".\n", url_start)
@ -251,8 +264,9 @@ class BudgetTestCase(IhatemoneyTestCase):
def test_project_creation(self):
with self.app.test_client() as c:
with self.app.mail.record_messages() as outbox:
# add a valid project
c.post(
resp = c.post(
"/create",
data={
"name": "The fabulous raclette party",
@ -261,6 +275,14 @@ class BudgetTestCase(IhatemoneyTestCase):
"contact_email": "raclette@notmyidea.org",
"default_currency": "USD",
},
follow_redirects=True,
)
# an email is sent to the owner with a reminder of the password
self.assertEqual(len(outbox), 1)
self.assertEqual(outbox[0].recipients, ["raclette@notmyidea.org"])
self.assertIn(
"A reminder email has just been sent to you",
resp.data.decode("utf-8"),
)
# session is updated