gh-106300: Improve assertRaises(Exception) usages in tests (GH-106302)

This commit is contained in:
Nikita Sobolev 2023-07-07 23:42:40 +03:00 committed by GitHub
parent 80b9b3a517
commit 6e6a4cd523
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 20 additions and 12 deletions

View file

@ -116,10 +116,13 @@ class TestMailbox(TestBase):
self.assertMailboxEmpty()
def test_add_that_raises_leaves_mailbox_empty(self):
class CustomError(Exception): ...
exc_msg = "a fake error"
def raiser(*args, **kw):
raise Exception("a fake error")
raise CustomError(exc_msg)
support.patch(self, email.generator.BytesGenerator, 'flatten', raiser)
with self.assertRaises(Exception):
with self.assertRaisesRegex(CustomError, exc_msg):
self._box.add(email.message_from_string("From: Alphöso"))
self.assertEqual(len(self._box), 0)
self._box.close()