Current File : //lib/python3/dist-packages/hyperlink/test/common.py
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
Mini Shell