def do_start_and_return (self, line): """Start T-Rex run and once in 'Running' mode, return to cmd prompt""" print termstyle.green("*** Starting T-Rex run, wait until in 'Running' state ***") try: ret = self.trex.start_trex(**self.run_params) print termstyle.green("*** End of scenario (T-Rex is probably still running!) ***") except TRexException as inst: print termstyle.red(inst)
def do_stop_trex (self, line): """Try to stop T-Rex run (if T-Rex is currently running)""" print termstyle.green("*** Starting T-Rex termination ***") try: ret = self.trex.stop_trex() print termstyle.green("*** End of scenario (T-Rex is not running now) ***") except TRexException as inst: print termstyle.red(inst)
def do_stop_trex(self, line): """Try to stop T-Rex run (if T-Rex is currently running)""" print termstyle.green("*** Starting T-Rex termination ***") try: ret = self.trex.stop_trex() print termstyle.green( "*** End of scenario (T-Rex is not running now) ***") except TRexException as inst: print termstyle.red(inst)
def do_kill_indiscriminately (self, line): """Force killing of running T-Rex process (if exists) on the server.""" print termstyle.green("*** Starting T-Rex termination ***") ret = self.trex.force_kill() if ret: print termstyle.green("*** End of scenario (T-Rex is not running now) ***") elif ret is None: print termstyle.magenta("*** End of scenario (T-Rex termination aborted) ***") else: print termstyle.red("*** End of scenario (T-Rex termination failed) ***")
def do_start_and_return(self, line): """Start T-Rex run and once in 'Running' mode, return to cmd prompt""" print termstyle.green( "*** Starting T-Rex run, wait until in 'Running' state ***") try: ret = self.trex.start_trex(**self.run_params) print termstyle.green( "*** End of scenario (T-Rex is probably still running!) ***") except TRexException as inst: print termstyle.red(inst)
def do_cancel_reservation (self, user): """Cancels a current reservation of T-Rex to a certain user""" try: if not user: ret = self.trex.cancel_reservation() else: ret = self.trex.cancel_reservation(user.split(' ')[0]) print termstyle.green("*** T-Rex reservation canceled successfully ***") except TRexException as inst: print termstyle.red(inst)
def do_reserve_trex(self, user): """Reserves the usage of T-Rex to a certain user""" try: if not user: ret = self.trex.reserve_trex() else: ret = self.trex.reserve_trex(user.split(' ')[0]) print termstyle.green("*** T-Rex reserved successfully ***") except TRexException as inst: print termstyle.red(inst)
def do_reserve_trex (self, user): """Reserves the usage of T-Rex to a certain user""" try: if not user: ret = self.trex.reserve_trex() else: ret = self.trex.reserve_trex(user.split(' ')[0]) print termstyle.green("*** T-Rex reserved successfully ***") except TRexException as inst: print termstyle.red(inst)
def do_cancel_reservation(self, user): """Cancels a current reservation of T-Rex to a certain user""" try: if not user: ret = self.trex.cancel_reservation() else: ret = self.trex.cancel_reservation(user.split(' ')[0]) print termstyle.green( "*** T-Rex reservation canceled successfully ***") except TRexException as inst: print termstyle.red(inst)
def do_poll_once (self, line): """Performs a single poll of T-Rex current data dump (if T-Rex is running) and prompts and short version of latest result_obj""" print termstyle.green("*** Trying T-Rex single poll ***") try: last_res = dict() if self.trex.is_running(dump_out = last_res): obj = self.trex.get_result_obj() print obj else: print termstyle.magenta("T-Rex isn't currently running.") print termstyle.green("*** End of scenario (T-Rex is posssibly still running!) ***") except TRexException as inst: print termstyle.red(inst)
def do_kill_indiscriminately(self, line): """Force killing of running T-Rex process (if exists) on the server.""" print termstyle.green("*** Starting T-Rex termination ***") ret = self.trex.force_kill() if ret: print termstyle.green( "*** End of scenario (T-Rex is not running now) ***") elif ret is None: print termstyle.magenta( "*** End of scenario (T-Rex termination aborted) ***") else: print termstyle.red( "*** End of scenario (T-Rex termination failed) ***")
def do_poll_once(self, line): """Performs a single poll of T-Rex current data dump (if T-Rex is running) and prompts and short version of latest result_obj""" print termstyle.green("*** Trying T-Rex single poll ***") try: last_res = dict() if self.trex.is_running(dump_out=last_res): obj = self.trex.get_result_obj() print obj else: print termstyle.magenta("T-Rex isn't currently running.") print termstyle.green( "*** End of scenario (T-Rex is posssibly still running!) ***") except TRexException as inst: print termstyle.red(inst)
def do_run_until_finish (self, sample_rate): """Starts T-Rex and sample server until run is done.""" print termstyle.green("*** Starting T-Rex run_until_finish scenario ***") if not sample_rate: # use default sample rate if not passed sample_rate = 5 try: sample_rate = int(sample_rate) ret = self.trex.start_trex(**self.run_params) self.trex.sample_to_run_finish(sample_rate) print termstyle.green("*** End of T-Rex run ***") except ValueError as inst: print termstyle.magenta("Provided illegal sample rate value. Please try again.\n[", inst,"]") except TRexException as inst: print termstyle.red(inst)
def do_run_until_finish(self, sample_rate): """Starts T-Rex and sample server until run is done.""" print termstyle.green( "*** Starting T-Rex run_until_finish scenario ***") if not sample_rate: # use default sample rate if not passed sample_rate = 5 try: sample_rate = int(sample_rate) ret = self.trex.start_trex(**self.run_params) self.trex.sample_to_run_finish(sample_rate) print termstyle.green("*** End of T-Rex run ***") except ValueError as inst: print termstyle.magenta( "Provided illegal sample rate value. Please try again.\n[", inst, "]") except TRexException as inst: print termstyle.red(inst)
def do_run_until_condition (self, sample_rate): """Starts T-Rex and sample server until condition is satisfied.""" print termstyle.green("*** Starting T-Rex run until condition is satisfied scenario ***") def condition (result_obj): return result_obj.get_current_tx_rate()['m_tx_pps'] > 200000 if not sample_rate: # use default sample rate if not passed sample_rate = 5 try: sample_rate = int(sample_rate) ret = self.trex.start_trex(**self.run_params) ret_val = self.trex.sample_until_condition(condition, sample_rate) print ret_val print termstyle.green("*** End of T-Rex run ***") except ValueError as inst: print termstyle.magenta("Provided illegal sample rate value. Please try again.\n[", inst,"]") except TRexException as inst: print termstyle.red(inst)
def do_run_and_poll (self, sample_rate): """Starts T-Rex and sample server manually until run is done.""" print termstyle.green("*** Starting T-Rex run and manually poll scenario ***") if not sample_rate: # use default sample rate if not passed sample_rate = 5 try: sample_rate = int(sample_rate) ret = self.trex.start_trex(**self.run_params) last_res = dict() while self.trex.is_running(dump_out = last_res): obj = self.trex.get_result_obj() if (self.verbose): print obj # do WHATEVER here time.sleep(sample_rate) print termstyle.green("*** End of T-Rex run ***") except ValueError as inst: print termstyle.magenta("Provided illegal sample rate value. Please try again.\n[", inst,"]") except TRexException as inst: print termstyle.red(inst)
def do_run_until_condition(self, sample_rate): """Starts T-Rex and sample server until condition is satisfied.""" print termstyle.green( "*** Starting T-Rex run until condition is satisfied scenario ***") def condition(result_obj): return result_obj.get_current_tx_rate()['m_tx_pps'] > 200000 if not sample_rate: # use default sample rate if not passed sample_rate = 5 try: sample_rate = int(sample_rate) ret = self.trex.start_trex(**self.run_params) ret_val = self.trex.sample_until_condition(condition, sample_rate) print ret_val print termstyle.green("*** End of T-Rex run ***") except ValueError as inst: print termstyle.magenta( "Provided illegal sample rate value. Please try again.\n[", inst, "]") except TRexException as inst: print termstyle.red(inst)
def do_run_and_poll(self, sample_rate): """Starts T-Rex and sample server manually until run is done.""" print termstyle.green( "*** Starting T-Rex run and manually poll scenario ***") if not sample_rate: # use default sample rate if not passed sample_rate = 5 try: sample_rate = int(sample_rate) ret = self.trex.start_trex(**self.run_params) last_res = dict() while self.trex.is_running(dump_out=last_res): obj = self.trex.get_result_obj() if (self.verbose): print obj # do WHATEVER here time.sleep(sample_rate) print termstyle.green("*** End of T-Rex run ***") except ValueError as inst: print termstyle.magenta( "Provided illegal sample rate value. Please try again.\n[", inst, "]") except TRexException as inst: print termstyle.red(inst)
def _show(self): if self.pidfile.is_locked(): print termstyle.red("T-Rex server daemon is running") else: print termstyle.red("T-Rex server daemon is NOT running")
def _show (self): if self.pidfile.is_locked(): print termstyle.red("T-Rex server daemon is running") else: print termstyle.red("T-Rex server daemon is NOT running")