Improve teams test run times by skipping signals

TNL-3126

add **kwargs to skip_signal method
This commit is contained in:
Bill DeRusha
2015-08-26 15:01:34 -04:00
parent ba1b84210d
commit 24a41dc59e
3 changed files with 92 additions and 47 deletions

View File

@@ -96,3 +96,14 @@ class MockSignalHandlerMixin(object):
del kwargs['exclude_args']
self.assertEqual(mock_args, args)
self.assertEqual(mock_kwargs, dict(kwargs, signal=mock_signal))
@contextmanager
def skip_signal(signal, **kwargs):
"""
ContextManager to skip a signal by disconnecting it, yielding,
and then reconnecting the signal.
"""
signal.disconnect(**kwargs)
yield
signal.connect(**kwargs)