mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-07-24 03:35:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

6
Lib/test/test_email/test_message.py
View file

@ -696,14 +696,16 @@ class TestEmailMessageBase:
self.assertIsNone(part['Content-Disposition'])
class _TestSetRaisingContentManager:
class CustomError(Exception):
pass
def set_content(self, msg, content, *args, **kw):
raise Exception('test')
raise self.CustomError('test')
def test_default_content_manager_for_add_comes_from_policy(self):
cm = self._TestSetRaisingContentManager()
m = self.message(policy=self.policy.clone(content_manager=cm))
for method in ('add_related', 'add_alternative', 'add_attachment'):
with self.assertRaises(Exception) as ar:
with self.assertRaises(self._TestSetRaisingContentManager.CustomError) as ar:
getattr(m, method)('')
self.assertEqual(str(ar.exception), 'test')