def testStart(self): self.task = utils.Runner('echo', arguments = 'Test text', start_shell=True) self.task.start() while self.task.isRunning(): pass self.task.stop() self.assertIn('Test text', self.task.getOutput())
def __init__(self): self.logger = logging.getLogger(HSS_Processor.__name__) self._log_dir = tempfile.TemporaryDirectory(prefix='hss.processor') self._log_dir_name = self._log_dir.name self._runner = utils.Runner(self.HSS_EXECUTABLE, log_dir=self._log_dir_name) self._hss_config = HSS_Config()
def __init__(self): self.logger = logging.getLogger(PGW_Processor.__name__) self._log_dir = tempfile.TemporaryDirectory(prefix='pgw.processor') self._log_dir_name = self._log_dir.name self._runner = utils.Runner(self.PGW_EXECUTABLE, log_dir=self._log_dir_name) self._pgw_config = PGW_Config()
def __init__(self, spgw_config_path=SPGW_CONFIG_PATH): self.logger = logging.getLogger(SPGW_Processor.__name__) self._configurator = SPGW_Configurator(config_path=spgw_config_path) self._log_dir = tempfile.TemporaryDirectory(prefix='spgw.processor') self._log_dir_name = self._log_dir.name self._runner = utils.Runner(self.SPGW_EXECUTABLE, log_dir=self._log_dir_name)
def _addService(self, ip, port): if ip is not None and port is not None: r = utils.Runner(self.CMD_IPVSADM, log_dir=self._log_dir_name, append_log = True) r.setArguments(self.ADD_SERVICE % (ip, port)) r.start() r.join()
def _add_iptables(self, ip): if ip is None: return r = utils.Runner(self.CMD_IPTABLES, arguments = self.ADD_IPTABLES % ip, log_dir=self._iptables_log_dir_name, append_log = True) r.start() r.join()
def __init__(self): self.logger = logging.getLogger(SGW_Processor.__name__) self._log_dir = tempfile.TemporaryDirectory(prefix='sgw.processor') self._log_dir_name = self._log_dir.name self._iptables_log_dir_name = os.path.join(self._log_dir_name, 'iptables.out') os.mkdir(self._iptables_log_dir_name) self._runner = utils.Runner(self.SGW_EXECUTABLE, log_dir=self._log_dir_name) self._sgw_config = SGW_Config()
def _addReplica(self, lb_ip, lb_port, server_ip, server_port): r = utils.Runner(self.CMD_IPVSADM, log_dir=self._log_dir_name, append_log = True) if server_port is None: r.setArguments(self.ADD_REPLICA % (lb_ip, lb_port, server_ip, lb_port)) else: r.setArguments(self.ADD_REPLICA % (lb_ip, lb_port, server_ip, server_port)) r.start() r.join()
def __init__(self, mme_config_path=MME_CONFIG_PATH, mme_freediameter_config_path=MME_FREEDIAMETER_CONFIG_PATH, host_file_path=HOST_FILE_PATH, cert_exe=MME_CERTIFICATE_CREATOR, cert_path=MME_CERTIFICATE_PATH): self.logger = logging.getLogger(MME_Processor.__name__) self._configurator = MME_Configurator(mme_config_path, mme_freediameter_config_path, host_file_path, cert_exe, cert_path) self._log_dir = tempfile.TemporaryDirectory(prefix='mme.processor') self._log_dir_name = self._log_dir.name self._runner = utils.Runner(self.MME_EXECUTABLE, log_dir=self._log_dir_name)
def __init__(self, hss_config_path=HSS_CONFIG_PATH, hss_freediameter_config_path=HSS_FREEDIAMETER_CONFIG_PATH, hss_certificate_exe=HSS_CERTIFICATE_EXECUTABLE, hss_certificate_path=HSS_CERTIFICATE_PATH, host_file_path=HOST_FILE_PATH): self.logger = logging.getLogger(HSS_Processor.__name__) self._configurator = HSS_Configurator(hss_config_path, hss_freediameter_config_path, host_file_path, hss_certificate_exe, hss_certificate_path) self._log_dir = tempfile.TemporaryDirectory(prefix='hss.processor') self._log_dir_name = self._log_dir.name self._runner = utils.Runner(self.HSS_EXECUTABLE, log_dir=self._log_dir_name)
def testLogFile(self): with tempfile.TemporaryDirectory() as log_dir: self.task = utils.Runner('echo', arguments = 'Test text', log_dir = log_dir, start_shell=True) self.task.start() while self.task.isRunning(): pass self.task.stop() self.assertIn('Test text', self.task.getOutput()) self.assertEqual('', self.task.getOutput(stderr = True)) stdout_content = "" with open(os.path.join(log_dir, "stdout")) as log_file: for line in log_file: stdout_content += line stderr_content = "" with open(os.path.join(log_dir, "stderr")) as log_file: for line in log_file: stderr_content += line self.assertIn('Test text', stdout_content) self.assertEqual('', stderr_content)
def _clear(self): r = utils.Runner(self.CMD_IPVSADM, log_dir=self._log_dir_name, append_log = True) r.setArguments(self.CLEAR) r.start()