diff options
author | 2015-08-26 14:27:43 +0300 | |
---|---|---|
committer | 2015-08-26 14:27:43 +0300 | |
commit | cdcc62972d42f009f55e6aeb2ca5c60c3acd75eb (patch) | |
tree | 5c3fef81ac01407a89740f2d9b8b01b0f3a47c7f /scripts/automation/trex_control_plane/examples | |
parent | 42053c95419042f36242b19d2416d112f7643e14 (diff) |
added dpkt package, initial stateless client implementation
Diffstat (limited to 'scripts/automation/trex_control_plane/examples')
-rw-r--r-- | scripts/automation/trex_control_plane/examples/interactive_stateless.py | 74 |
1 files changed, 62 insertions, 12 deletions
diff --git a/scripts/automation/trex_control_plane/examples/interactive_stateless.py b/scripts/automation/trex_control_plane/examples/interactive_stateless.py index c0fc51a7..016888d2 100644 --- a/scripts/automation/trex_control_plane/examples/interactive_stateless.py +++ b/scripts/automation/trex_control_plane/examples/interactive_stateless.py @@ -1,14 +1,17 @@ #!/router/bin/python import trex_root_path -from client.trex_client import * +from client.trex_stateless_client import * from common.trex_exceptions import * import cmd -from python_lib.termstyle import termstyle +from termstyle import termstyle +# import termstyle import os from argparse import ArgumentParser import socket import errno +import ast +import json class InteractiveStatelessTRex(cmd.Cmd): @@ -17,10 +20,12 @@ class InteractiveStatelessTRex(cmd.Cmd): \nType help to view available pre-defined scenarios\n(c) All rights reserved.\n") prompt = '> ' - def __init__(self, verbose_mode=False): + def __init__(self, trex_host, trex_port, virtual, verbose): cmd.Cmd.__init__(self) - self.verbose = verbose_mode - self.trex = None + + self.verbose = verbose + self.virtual = virtual + self.trex = CTRexStatelessClient(trex_host, trex_port, self.virtual) self.DEFAULT_RUN_PARAMS = dict(m=1.5, nc=True, p=True, @@ -29,6 +34,47 @@ class InteractiveStatelessTRex(cmd.Cmd): l=1000) self.run_params = dict(self.DEFAULT_RUN_PARAMS) + def do_transmit(self, line): + """Transmits a request over using a given link to server.\ + \nuse: transmit [method_name] [method_params]""" + if line == "": + print "\nUsage: [method name] [param dict as string]\n" + print "Example: rpc test_add {'x': 12, 'y': 17}\n" + return + + args = line.split(' ', 1) # args will have max length of 2 + method_name = args[0] + params = None + bad_parse = False + + try: + params = ast.literal_eval(args[1]) + if not isinstance(params, dict): + bad_parse = True + except ValueError as e1: + bad_parse = True + except SyntaxError as e2: + bad_parse = True + + if bad_parse: + print "\nValue should be a valid dict: '{0}'".format(args[1]) + print "\nUsage: [method name] [param dict as string]\n" + print "Example: rpc test_add {'x': 12, 'y': 17}\n" + return + + response = self.trex.transmit(method_name, params) + if not self.virtual: + # expect response + rc, msg = response + if rc: + print "\nServer Response:\n\n" + json.dumps(msg) + "\n" + else: + print "\n*** " + msg + "\n" + + + + + def do_push_files(self, filepaths): """Pushes a custom file to be stored locally on T-Rex server.\ \nPush multiple files by specifying their path separated by ' ' (space).""" @@ -52,21 +98,25 @@ if __name__ == "__main__": parser.add_argument('-v', '--version', action='version', version='%(prog)s 1.0 \t (C) Cisco Systems Inc.\n') - # parser.add_argument("-t", "--trex-host", required = True, dest="trex_host", - # action="store", help="Specify the hostname or ip to connect with T-Rex server.", - # metavar="HOST" ) - # parser.add_argument("-p", "--trex-port", type=int, default = 8090, metavar="PORT", dest="trex_port", - # help="Select port on which the T-Rex server listens. Default port is 8090.", action="store") + parser.add_argument("-t", "--trex-host", required = True, dest="trex_host", + action="store", help="Specify the hostname or ip to connect with T-Rex server.", + metavar="HOST" ) + parser.add_argument("-p", "--trex-port", type=int, default = 8090, metavar="PORT", dest="trex_port", + help="Select port on which the T-Rex server listens. Default port is 8090.", action="store") # parser.add_argument("-m", "--maxhist", type=int, default = 100, metavar="SIZE", dest="hist_size", # help="Specify maximum history size saved at client side. Default size is 100.", action="store") + parser.add_argument("--virtual", dest="virtual", + action="store_true", + help="Switch ON virtual option at TRex client. Default is: OFF.", + default=False) parser.add_argument("--verbose", dest="verbose", action="store_true", - help="Switch ON verbose option at T-Rex client. Default is: OFF.", + help="Switch ON verbose option at TRex client. Default is: OFF.", default=False) args = parser.parse_args() try: - InteractiveStatelessTRex(args.verbose).cmdloop() + InteractiveStatelessTRex(**vars(args)).cmdloop() except KeyboardInterrupt: print termstyle.cyan('Bye Bye!') |