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)
