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()