diff options
| -rw-r--r-- | tests/mail/tests.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/tests/mail/tests.py b/tests/mail/tests.py index 1f8c6b12bf..72c22ad341 100644 --- a/tests/mail/tests.py +++ b/tests/mail/tests.py @@ -188,14 +188,22 @@ class MailTests(HeadersCheckMixin, SimpleTestCase): EmailMessage(reply_to='reply_to@example.com') def test_header_injection(self): + msg = "Header values can't contain newlines " email = EmailMessage('Subject\nInjection Test', 'Content', 'from@example.com', ['to@example.com']) - with self.assertRaises(BadHeaderError): + with self.assertRaisesMessage(BadHeaderError, msg): email.message() email = EmailMessage( gettext_lazy('Subject\nInjection Test'), 'Content', 'from@example.com', ['to@example.com'] ) - with self.assertRaises(BadHeaderError): + with self.assertRaisesMessage(BadHeaderError, msg): email.message() + with self.assertRaisesMessage(BadHeaderError, msg): + EmailMessage( + 'Subject', + 'Content', + 'from@example.com', + ['Name\nInjection test <to@example.com>'], + ).message() def test_space_continuation(self): """ |