def setUp(self):
logger.manager.disable = logging.CRITICAL
self.listener = JsonStreamListener()
# Patch a bunch of methods on the listener so we can check
# the handling of different message types
self.patched_methods = ['on_status', 'on_delete', 'on_scrub_geo',
'on_limit', 'on_status_withheld', 'on_user_withheld',
'on_disconnect', 'on_stall_warning',
'on_error', 'on_unknown', 'on_exception']
for patch in self.patched_methods:
replacement = mock.Mock(wraps=getattr(self.listener, patch))
mock.patch.object(self.listener, patch, replacement).start()
def setUp(self):
logger.manager.disable = logging.CRITICAL
self.listener = JsonStreamListener()
