Merge pull request #9489 from edx/bderusha/teams-mock-callback

Improve teams test run times by skipping signals
This commit is contained in:
Bill DeRusha
2015-08-28 10:09:17 -04:00
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)