def setUp(self): if ('udp_bridge.py' not in tcollector.COLLECTORS): return self.saved_exit = sys.exit self.saved_stderr = sys.stderr self.saved_stdout = sys.stdout self.udp_bridge = tcollector.COLLECTORS['udp_bridge.py'] self.udp_globals = {} sys.exit = lambda x: None try: execfile(self.udp_bridge.filename, self.udp_globals) finally: sys.exit = self.saved_exit self.udp_globals['socket'] = mocks.Socket() self.udp_globals['sys'] = mocks.Sys() self.udp_globals['udp_bridge_conf'].enabled = lambda: True self.udp_globals['utils'] = mocks.Utils()
def setUp(self): if ('udp_bridge.py' not in tcollector.COLLECTORS): # pylint: disable=maybe-no-member raise unittest.SkipTest("udp_bridge unavailable") self.saved_exit = sys.exit self.saved_stderr = sys.stderr self.saved_stdout = sys.stdout self.udp_bridge = tcollector.COLLECTORS['udp_bridge.py'] # pylint: disable=maybe-no-member self.udp_globals = {} sys.exit = return_none bridge_file = open(self.udp_bridge.filename) try: exec(compile(bridge_file.read(), self.udp_bridge.filename, 'exec'), self.udp_globals) finally: bridge_file.close() sys.exit = self.saved_exit self.udp_globals['socket'] = mocks.Socket() self.udp_globals['sys'] = mocks.Sys() self.udp_globals['udp_bridge_conf'].enabled = always_true self.udp_globals['utils'] = mocks.Utils()