def spy_on(*args, **kwargs): """Spies on a function. By default, the spy will allow the call to go through to the original function. This can be disabled by passing call_original=False when initiating the spy. If disabled, the original function will never be called. This can also be passed a call_fake parameter pointing to another function to call instead of the original. If passed, this will take precedence over call_original. The spy will only remain throughout the duration of the context. """ agency = SpyAgency() spy = agency.spy_on(*args, **kwargs) yield spy spy.unspy()
def spy_on(*args, **kwargs): """Spy on a function. By default, the spy will allow the call to go through to the original function. This can be disabled by passing ``call_original=False`` when initiating the spy. If disabled, the original function will never be called. This can also be passed a ``call_fake`` parameter pointing to another function to call instead of the original. If passed, this will take precedence over ``call_original``. The spy will only remain throughout the duration of the context. See :py:class:`~kgb.spies.FunctionSpy` for more details on arguments. Args: *args (tuple): Positional arguments to pass to :py:class:`~kgb.spies.FunctionSpy`. **kwargs (dict): Keyword arguments to pass to :py:class:`~kgb.spies.FunctionSpy`. Context: kgb.spies.FunctionSpy: The newly-created spy. """ agency = SpyAgency() spy = agency.spy_on(*args, **kwargs) try: yield spy finally: spy.unspy()
def setUp(self): self.agency = SpyAgency() self.orig_class_do_math = MathClass.class_do_math
"""Independent assertion functions. These assertion functions can be used in pytest or in other places where :py:class:`kgb.SpyAgency` can't be mixed. Version Added: 7.0 """ from __future__ import unicode_literals from kgb.agency import SpyAgency _agency = SpyAgency() __all__ = [] for name in vars(SpyAgency): if name.startswith('assert_'): globals()[name] = getattr(_agency, name) __all__.append(name)