diff --git a/mock/tests/testasync.py b/mock/tests/testasync.py index 7fd680f4..f21b9fa8 100644 --- a/mock/tests/testasync.py +++ b/mock/tests/testasync.py @@ -469,9 +469,10 @@ async def addition(self, var): pass self.assertEqual(output, 10) async def test_add_side_effect_exception(self): + class CustomError(Exception): pass async def addition(var): pass - mock = AsyncMock(addition, side_effect=Exception('err')) - with self.assertRaises(Exception): + mock = AsyncMock(addition, side_effect=CustomError('side-effect')) + with self.assertRaisesRegex(CustomError, 'side-effect'): await mock(5) async def test_add_side_effect_coroutine(self):