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_update_run_params(self, json_str):
"""Updates provided parameters on T-Rex running configuration. Provide using JSON string"""
if json_str:
try:
upd_params = self.decoder.decode(json_str)
self.run_params.update(upd_params)
print termstyle.green("*** End of T-Rex parameters update ***")
except ValueError as inst:
print termstyle.magenta("Provided illegal JSON string. Please try again.\n[", inst,"]")
else:
print termstyle.magenta("JSON configuration string is missing. Please try again.")
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_push_files (self, filepaths):
"""Pushes a custom file to be stored locally on T-Rex server.\nPush multiple files by spefiying their path separated by ' ' (space)."""
try:
filepaths = filepaths.split(' ')
print termstyle.green("*** Starting pushing files ({trex_files}) to T-Rex. ***".format (trex_files = ', '.join(filepaths)) )
ret_val = self.trex.push_files(filepaths)
if ret_val:
print termstyle.green("*** End of T-Rex push_files method (success) ***")
else:
print termstyle.magenta("*** End of T-Rex push_files method (failed) ***")
except IOError as inst:
print termstyle.magenta(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_update_run_params(self, json_str):
"""Updates provided parameters on T-Rex running configuration. Provide using JSON string"""
if json_str:
try:
upd_params = self.decoder.decode(json_str)
self.run_params.update(upd_params)
print termstyle.green("*** End of T-Rex parameters update ***")
except ValueError as inst:
print termstyle.magenta(
"Provided illegal JSON string. Please try again.\n[", inst,
"]")
else:
print termstyle.magenta(
"JSON configuration string is missing. Please try again.")
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_push_files(self, filepaths):
"""Pushes a custom file to be stored locally on T-Rex server.\nPush multiple files by spefiying their path separated by ' ' (space)."""
try:
filepaths = filepaths.split(' ')
print termstyle.green(
"*** Starting pushing files ({trex_files}) to T-Rex. ***".
format(trex_files=', '.join(filepaths)))
ret_val = self.trex.push_files(filepaths)
if ret_val:
print termstyle.green(
"*** End of T-Rex push_files method (success) ***")
else:
print termstyle.magenta(
"*** End of T-Rex push_files method (failed) ***")
except IOError as inst:
print termstyle.magenta(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)