def save_report(self, fname):
"""
:type fname: str
"""
try:
if os.path.exists(fname):
self.log.warning(
"File %s already exists, it will be overwritten", fname)
else:
dirname = os.path.dirname(fname)
if dirname and not os.path.exists(dirname):
os.makedirs(dirname)
testsuites = etree.Element("testsuites")
for _, suite in iteritems(self.test_suites):
testsuites.append(suite)
etree_obj = etree.ElementTree(testsuites)
self.log.info("Writing JUnit XML report into: %s", fname)
with open(get_full_path(fname), 'wb') as _fds:
etree_obj.write(_fds,
xml_declaration=True,
encoding="UTF-8",
pretty_print=True)
except BaseException:
raise TaurusInternalException("Cannot create file %s" % fname)
def __dump_xml(self, filename):
self.log.info("Dumping final status as XML: %s", filename)
root = etree.Element("FinalStatus")
if self.first_ts < float("inf") and self.last_ts > 0:
duration_elem = etree.Element("TestDuration")
duration_elem.text = str(
round(float(self.last_ts - self.first_ts), 3))
root.append(duration_elem)
report_info = get_bza_report_info(self.engine, self.log)
if report_info:
link, _ = report_info[0]
report_element = etree.Element("ReportURL")
report_element.text = link
root.append(report_element)
if self.last_sec:
for label, kpiset in iteritems(
self.last_sec[DataPoint.CUMULATIVE]):
root.append(self.__get_xml_summary(label, kpiset))
with open(get_full_path(filename), 'wb') as fhd:
tree = etree.ElementTree(root)
tree.write(fhd,
pretty_print=True,
encoding="UTF-8",
xml_declaration=True)
def load(self, path):
try:
self.tree = etree.ElementTree()
self.tree.parse(path)
except BaseException as exc:
msg = "XML parsing failed for file %s: %s"
raise TaurusInternalException(msg % (path, exc))
def save_report(self, root_node):
"""
:param root_node:
:return:
"""
try:
if os.path.exists(self.report_file_path):
self.log.warning("File %s already exists, will be overwritten",
self.report_file_path)
else:
dirname = os.path.dirname(self.report_file_path)
if dirname and not os.path.exists(dirname):
os.makedirs(dirname)
etree_obj = etree.ElementTree(root_node)
self.log.info("Writing JUnit XML report into: %s",
self.report_file_path)
with open(self.report_file_path, 'wb') as _fds:
etree_obj.write(_fds,
xml_declaration=True,
encoding="UTF-8",
pretty_print=True)
except BaseException:
self.log.error("Cannot create file %s", self.report_file_path)
raise
def load(self, filename):
try:
self.tree = etree.ElementTree()
self.tree.parse(filename)
self.root = self.tree.getroot()
except BaseException as exc:
self.log.debug("Tsung: XML parsing error: %s", traceback.format_exc())
raise TaurusInternalException("Tsung: XML parsing failed for file %s: %s" % (filename, exc))
def __init__(self, tsung_tool):
self.log = logging.getLogger(self.__class__.__name__)
self.root = etree.Element("tsung",
loglevel="notice",
version="1.0",
dumptraffic="protocol",
backend="text")
self.tree = etree.ElementTree(self.root)
self.tsung_tool = tsung_tool
def __dump_xml(self, filename):
self.log.info("Dumping final status as XML: %s", filename)
root = etree.Element("FinalStatus")
if self.last_sec:
for label, kpiset in iteritems(self.last_sec[DataPoint.CUMULATIVE]):
root.append(self.__get_xml_summary(label, kpiset))
with open(get_full_path(filename), 'wb') as fhd:
tree = etree.ElementTree(root)
tree.write(fhd, pretty_print=True, encoding="UTF-8", xml_declaration=True)
def load(self, original):
"""
Load existing JMX file
:param original: JMX file path
:raise TaurusInternalException: in case of XML parsing error
"""
try:
self.tree = etree.ElementTree()
self.tree.parse(original)
except BaseException as exc:
msg = "XML parsing failed for file %s: %s"
raise TaurusInternalException(msg % (original, exc))
def load(self, original):
"""
Load existing JMX file
:param original: JMX file path
:raise RuntimeError: in case of XML parsing error
"""
try:
self.tree = etree.ElementTree()
self.tree.parse(original)
except BaseException as exc:
self.log.debug("XML parsing error: %s", traceback.format_exc())
data = (original, exc)
raise RuntimeError("XML parsing failed for file %s: %s" % data)
def __init__(self, original=None, test_plan_name="BZT Generated Test Plan"):
self.log = logging.getLogger(self.__class__.__name__)
if original:
self.load(original)
else:
root = etree.Element("jmeterTestPlan")
self.tree = etree.ElementTree(root)
test_plan = etree.Element("TestPlan", guiclass="TestPlanGui",
testname=test_plan_name,
testclass="TestPlan")
htree = etree.Element("hashTree")
htree.append(test_plan)
htree.append(etree.Element("hashTree"))
self.append("jmeterTestPlan", htree)
element_prop = self._get_arguments_panel("TestPlan.user_defined_variables")
self.append("jmeterTestPlan>hashTree>TestPlan", element_prop)
def __init__(self):
self.root = etree.Element("NoseTest")
self.tree = etree.ElementTree(self.root)
def __init__(self, scenario, parent_logger):
self.root = etree.Element("PythonCode")
self.tree = etree.ElementTree(self.root)
self.log = parent_logger.getChild(self.__class__.__name__)
self.scenario = scenario
