diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_packetizer.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tests/test_packetizer.py b/tests/test_packetizer.py index 15a7c93f..a7cac6de 100644 --- a/tests/test_packetizer.py +++ b/tests/test_packetizer.py @@ -101,9 +101,13 @@ class PacketizerTest (unittest.TestCase): import signal class TimeoutError(Exception): - pass + def __init__(self, error_message): + if hasattr(errno, 'ETIME'): + self.message = os.sterror(errno.ETIME) + else: + self.messaage = error_message - def timeout(seconds=1, error_message=os.strerror(errno.ETIME)): + def timeout(seconds=1, error_message='Timer expired'): def decorator(func): def _handle_timeout(signum, frame): raise TimeoutError(error_message) |