25 lines
568 B
Python
25 lines
568 B
Python
|
import unittest
|
||
|
|
|
||
|
|
from huey.utils import wrap_exception
|
||
|
|
|
||
|
|
|
||
|
|
class MyException(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class TestWrapException(unittest.TestCase):
|
||
|
|
def test_wrap_exception(self):
|
||
|
|
def raise_keyerror():
|
||
|
|
try:
|
||
|
|
{}['huey']
|
||
|
|
except KeyError as exc:
|
||
|
|
raise wrap_exception(MyException)
|
||
|
|
|
||
|
|
self.assertRaises(MyException, raise_keyerror)
|
||
|
|
try:
|
||
|
|
raise_keyerror()
|
||
|
|
except MyException as exc:
|
||
|
|
self.assertEqual(str(exc), "KeyError: 'huey'")
|
||
|
|
else:
|
||
|
|
assert False
|