def run_bridge_test(self, udpInputLines, stdoutLines, stderrLines):
mockSocket = self.udp_globals['socket'] = mocks.Socket()
mockSocket.state['udp_in'] = list(udpInputLines)
self.udp_globals['sys'] = mocks.Sys()
self.udp_globals['sys'].stderr.lines = stderrLines
self.udp_globals['sys'].stdout.lines = stdoutLines
sys.stderr = self.udp_globals['sys'].stderr
sys.stdout = self.udp_globals['sys'].stdout
try:
self.udp_globals['main']()
except mocks.SocketDone:
pass
finally:
sys.stderr = self.saved_stderr
sys.stdout = self.saved_stdout
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()
