def add(self, df):
key = df.attr(DiscoveredFileAttributeEnum.DIRECTORY_ABSOLUTE_PATH) + "/" + df.attr(
DiscoveredFileAttributeEnum.FULL_NAME).upper()
if key in self.found_class_names or key in self.temp_map:
fa_msg1.format(DiscoveredFileAttributeEnum.FULL_NAME)
self.console.display_error(fa_msg1)
sys_utils.fexit()
self.temp_map[key] = df
def convert_to_dict(self, args):
try:
args = self.parser.parse_args(args[1:])
return vars(args)
except Exception as e:
print("!!!Fatal Error!!!")
print(e)
import traceback
traceback.print_exc()
sys_utils.fexit()
def convert_to_dict(self, args):
def format_value(v):
if type(v) is list:
return ",".join([str(i) for i in v])
else:
return v
try:
args = self.parser.parse_args(args[1:])
return {k:format_value(v) for k,v in vars(args).items()}
except Exception as e:
print("!!!Fatal Error!!!")
print(e)
import traceback
traceback.print_exc()
sys_utils.fexit()
def load_session(self, session_name):
from arjuna.tpi import Arjuna
from arjuna.unitee.test.defs.session import UserDefinedSessionDef
sdir = Arjuna.get_central_config().get_arjuna_option_value(
ArjunaOption.UNITEE_PROJECT_SESSIONS_DIR).as_str()
session_file_path = os.path.join(sdir, session_name + ".xml")
if not file_utils.is_file(session_file_path):
Arjuna.get_console().display_error(
"Not able to find session file {}.xml at {}".format(
session_name, sdir))
sys_utils.fexit()
sdef = UserDefinedSessionDef(session_name, session_file_path)
sdef.process()
self.__session = sdef.pick()
self.__session.context = self.__create_test_context()
self.__session.load()
def convert_to_dict(self, args):
def format_value(k, v):
# if k in {"ao", "uo", "irules", "erules", 'ip', 'ep', }:
# return v
if k == "report.formats" and v is None:
v = ['XML', 'HTML']
# if type(v) is list:
# return ",".join([str(i) for i in v])
# else:
return v
try:
args = self.parser.parse_args(args[1:])
return {k:format_value(k,v) for k,v in vars(args).items()}
except Exception as e:
print("!!!Fatal Error!!!")
print(e)
import traceback
traceback.print_exc()
sys_utils.fexit()
import sys
sys.exit(1)
def __init__(self, sdef, id, stage_xml):
super().__init__()
self.gdefs = []
self.config = ConfigContainer()
self.tmcount = 0
self.threads = 1
self.sdef = sdef
self.__iter = None
self.__fixtures = FixturesDef()
self.root = stage_xml
self.unitee = Arjuna.get_unitee_instance()
self.id = id
self.name = "stage{:d}".format(id)
if not isinstance(stage_xml, Element):
self.console.display_error(
"Fatal: [Arjuna Error] Unsuppored input argument supplied for stage creation: {}"
.format(stage_xml))
sys_utils.fexit()
else:
self.__process(stage_xml)
def display_err_and_exit(msg):
self.console.display_error(
(msg + " Fix groups template file: {}").format(self.fpath))
sys_utils.fexit()
def display_err_and_exit(conf_type, conf_path, msg):
from arjuna import ArjunaCore
console = ArjunaCore.console
console.display_error(msg + " Fix {} template file: {}".format(conf_type, conf_path))
sys_utils.fexit()
