69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
|
from typing import Any, Callable, Optional, Type
|
||
|
|
from unittest import TestCase
|
||
|
|
|
||
|
|
|
||
|
|
class HyperlinkTestCase(TestCase):
|
||
|
|
"""This type mostly exists to provide a backwards-compatible
|
||
|
|
assertRaises method for Python 2.6 testing.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def assertRaises( # type: ignore[override]
|
||
|
|
self,
|
||
|
|
expected_exception, # type: Type[BaseException]
|
||
|
|
callableObj=None, # type: Optional[Callable[..., Any]]
|
||
|
|
*args, # type: Any
|
||
|
|
**kwargs # type: Any
|
||
|
|
):
|
||
|
|
# type: (...) -> Any
|
||
|
|
"""Fail unless an exception of class expected_exception is raised
|
||
|
|
by callableObj when invoked with arguments args and keyword
|
||
|
|
arguments kwargs. If a different type of exception is
|
||
|
|
raised, it will not be caught, and the test case will be
|
||
|
|
deemed to have suffered an error, exactly as for an
|
||
|
|
unexpected exception.
|
||
|
|
|
||
|
|
If called with callableObj omitted or None, will return a
|
||
|
|
context object used like this::
|
||
|
|
|
||
|
|
with self.assertRaises(SomeException):
|
||
|
|
do_something()
|
||
|
|
|
||
|
|
The context manager keeps a reference to the exception as
|
||
|
|
the 'exception' attribute. This allows you to inspect the
|
||
|
|
exception after the assertion::
|
||
|
|
|
||
|
|
with self.assertRaises(SomeException) as cm:
|
||
|
|
do_something()
|
||
|
|
the_exception = cm.exception
|
||
|
|
self.assertEqual(the_exception.error_code, 3)
|
||
|
|
"""
|
||
|
|
context = _AssertRaisesContext(expected_exception, self)
|
||
|
|
if callableObj is None:
|
||
|
|
return context
|
||
|
|
with context:
|
||
|
|
callableObj(*args, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
class _AssertRaisesContext(object):
|
||
|
|
"A context manager used to implement HyperlinkTestCase.assertRaises."
|
||
|
|
|
||
|
|
def __init__(self, expected, test_case):
|
||
|
|
# type: (Type[BaseException], TestCase) -> None
|
||
|
|
self.expected = expected
|
||
|
|
self.failureException = test_case.failureException
|
||
|
|
|
||
|
|
def __enter__(self):
|
||
|
|
# type: () -> "_AssertRaisesContext"
|
||
|
|
return self
|
||
|
|
|
||
|
|
def __exit__(self, exc_type, exc_value, tb):
|
||
|
|
# type: (Optional[Type[BaseException]], Any, Any) -> bool
|
||
|
|
if exc_type is None:
|
||
|
|
exc_name = self.expected.__name__
|
||
|
|
raise self.failureException("%s not raised" % (exc_name,))
|
||
|
|
if not issubclass(exc_type, self.expected):
|
||
|
|
# let unexpected exceptions pass through
|
||
|
|
return False
|
||
|
|
self.exception = exc_value # store for later retrieval
|
||
|
|
return True
|