# Copyright 2008-2015 Nokia Networks
# Copyright 2016- Robot Framework Foundation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from datetime import datetime
from robot.utils import NullMarkupWriter, XmlWriter
from robot.version import get_full_version
from robot.result import Keyword, TestCase, TestSuite, ResultVisitor
from .loggerapi import LoggerApi
from .loggerhelper import IsLogged
[docs]
class XmlLoggerAdapter(LoggerApi):
def __init__(self, path, log_level='TRACE', rpa=False, generator='Robot',
legacy_output=False):
logger = XmlLogger if not legacy_output else LegacyXmlLogger
self.logger = logger(path, log_level, rpa, generator)
@property
def flatten_level(self):
return self.logger.flatten_level
[docs]
def close(self):
self.logger.close()
[docs]
def set_log_level(self, level):
return self.logger.set_log_level(level)
[docs]
def start_suite(self, data, result):
self.logger.start_suite(result)
[docs]
def end_suite(self, data, result):
self.logger.end_suite(result)
[docs]
def start_test(self, data, result):
self.logger.start_test(result)
[docs]
def end_test(self, data, result):
self.logger.end_test(result)
[docs]
def start_keyword(self, data, result):
self.logger.start_keyword(result)
[docs]
def end_keyword(self, data, result):
self.logger.end_keyword(result)
[docs]
def start_for(self, data, result):
self.logger.start_for(result)
[docs]
def end_for(self, data, result):
self.logger.end_for(result)
[docs]
def start_for_iteration(self, data, result):
self.logger.start_for_iteration(result)
[docs]
def end_for_iteration(self, data, result):
self.logger.end_for_iteration(result)
[docs]
def start_while(self, data, result):
self.logger.start_while(result)
[docs]
def end_while(self, data, result):
self.logger.end_while(result)
[docs]
def start_while_iteration(self, data, result):
self.logger.start_while_iteration(result)
[docs]
def end_while_iteration(self, data, result):
self.logger.end_while_iteration(result)
[docs]
def start_if(self, data, result):
self.logger.start_if(result)
[docs]
def end_if(self, data, result):
self.logger.end_if(result)
[docs]
def start_if_branch(self, data, result):
self.logger.start_if_branch(result)
[docs]
def end_if_branch(self, data, result):
self.logger.end_if_branch(result)
[docs]
def start_try(self, data, result):
self.logger.start_try(result)
[docs]
def end_try(self, data, result):
self.logger.end_try(result)
[docs]
def start_try_branch(self, data, result):
self.logger.start_try_branch(result)
[docs]
def end_try_branch(self, data, result):
self.logger.end_try_branch(result)
[docs]
def start_var(self, data, result):
self.logger.start_var(result)
[docs]
def end_var(self, data, result):
self.logger.end_var(result)
[docs]
def start_break(self, data, result):
self.logger.start_break(result)
[docs]
def end_break(self, data, result):
self.logger.end_break(result)
[docs]
def start_continue(self, data, result):
self.logger.start_continue(result)
[docs]
def end_continue(self, data, result):
self.logger.end_continue(result)
[docs]
def start_return(self, data, result):
self.logger.start_return(result)
[docs]
def end_return(self, data, result):
self.logger.end_return(result)
[docs]
def start_error(self, data, result):
self.logger.start_error(result)
[docs]
def end_error(self, data, result):
self.logger.end_error(result)
[docs]
def log_message(self, message):
self.logger.log_message(message)
[docs]
def message(self, message):
self.logger.message(message)
[docs]
class XmlLogger(ResultVisitor):
def __init__(self, output, log_level='TRACE', rpa=False, generator='Robot',
suite_only=False):
self._log_message_is_logged = IsLogged(log_level)
self._error_message_is_logged = IsLogged('WARN')
# `_writer` is set to NullMarkupWriter when flattening, `_xml_writer` is not.
self._writer = self._xml_writer = self._get_writer(output, rpa, generator,
suite_only)
self.flatten_level = 0
self._errors = []
def _get_writer(self, output, rpa, generator, suite_only):
if not output:
return NullMarkupWriter()
writer = XmlWriter(output, write_empty=False, usage='output',
preamble=not suite_only)
if not suite_only:
writer.start('robot', self._get_start_attrs(rpa, generator))
return writer
def _get_start_attrs(self, rpa, generator):
return {'generator': get_full_version(generator),
'generated': datetime.now().isoformat(),
'rpa': 'true' if rpa else 'false',
'schemaversion': '5'}
[docs]
def close(self):
self.start_errors()
for msg in self._errors:
self._write_message(msg)
self.end_errors()
self._writer.end('robot')
self._writer.close()
[docs]
def set_log_level(self, level):
return self._log_message_is_logged.set_level(level)
[docs]
def message(self, msg):
if self._error_message_is_logged(msg.level):
self._errors.append(msg)
[docs]
def log_message(self, msg):
if self._log_message_is_logged(msg.level):
self._write_message(msg)
def _write_message(self, msg):
attrs = {'time': msg.timestamp.isoformat() if msg.timestamp else None,
'level': msg.level}
if msg.html:
attrs['html'] = 'true'
# Use `_xml_writer`, not `_writer` to write messages also when flattening.
self._xml_writer.element('msg', msg.message, attrs)
[docs]
def start_keyword(self, kw):
self._writer.start('kw', self._get_start_keyword_attrs(kw))
if kw.tags.robot('flatten'):
self.flatten_level += 1
self._writer = NullMarkupWriter()
def _get_start_keyword_attrs(self, kw):
attrs = {'name': kw.name, 'owner': kw.owner}
if kw.type != 'KEYWORD':
attrs['type'] = kw.type
if kw.source_name:
attrs['source_name'] = kw.source_name
return attrs
[docs]
def end_keyword(self, kw):
if kw.tags.robot('flatten'):
self.flatten_level -= 1
if self.flatten_level == 0:
self._writer = self._xml_writer
self._write_list('var', kw.assign)
self._write_list('arg', [str(a) for a in kw.args])
self._write_list('tag', kw.tags)
self._writer.element('doc', kw.doc)
if kw.timeout:
self._writer.element('timeout', attrs={'value': str(kw.timeout)})
self._write_status(kw)
self._writer.end('kw')
[docs]
def start_if(self, if_):
self._writer.start('if')
[docs]
def end_if(self, if_):
self._write_status(if_)
self._writer.end('if')
[docs]
def start_if_branch(self, branch):
self._writer.start('branch', {'type': branch.type,
'condition': branch.condition})
[docs]
def end_if_branch(self, branch):
self._write_status(branch)
self._writer.end('branch')
[docs]
def start_for(self, for_):
self._writer.start('for', {'flavor': for_.flavor,
'start': for_.start,
'mode': for_.mode,
'fill': for_.fill})
[docs]
def end_for(self, for_):
for name in for_.assign:
self._writer.element('var', name)
for value in for_.values:
self._writer.element('value', value)
self._write_status(for_)
self._writer.end('for')
[docs]
def start_for_iteration(self, iteration):
self._writer.start('iter')
[docs]
def end_for_iteration(self, iteration):
for name, value in iteration.assign.items():
self._writer.element('var', value, {'name': name})
self._write_status(iteration)
self._writer.end('iter')
[docs]
def start_try(self, root):
self._writer.start('try')
[docs]
def end_try(self, root):
self._write_status(root)
self._writer.end('try')
[docs]
def start_try_branch(self, branch):
if branch.type == branch.EXCEPT:
self._writer.start('branch', attrs={
'type': 'EXCEPT',
'pattern_type': branch.pattern_type,
'assign': branch.assign
})
self._write_list('pattern', branch.patterns)
else:
self._writer.start('branch', attrs={'type': branch.type})
[docs]
def end_try_branch(self, branch):
self._write_status(branch)
self._writer.end('branch')
[docs]
def start_while(self, while_):
self._writer.start('while', attrs={
'condition': while_.condition,
'limit': while_.limit,
'on_limit': while_.on_limit,
'on_limit_message': while_.on_limit_message
})
[docs]
def end_while(self, while_):
self._write_status(while_)
self._writer.end('while')
[docs]
def start_while_iteration(self, iteration):
self._writer.start('iter')
[docs]
def end_while_iteration(self, iteration):
self._write_status(iteration)
self._writer.end('iter')
[docs]
def start_var(self, var):
attr = {'name': var.name}
if var.scope is not None:
attr['scope'] = var.scope
if var.separator is not None:
attr['separator'] = var.separator
self._writer.start('variable', attr, write_empty=True)
[docs]
def end_var(self, var):
for val in var.value:
self._writer.element('var', val)
self._write_status(var)
self._writer.end('variable')
[docs]
def start_return(self, return_):
self._writer.start('return')
[docs]
def end_return(self, return_):
for value in return_.values:
self._writer.element('value', value)
self._write_status(return_)
self._writer.end('return')
[docs]
def start_continue(self, continue_):
self._writer.start('continue')
[docs]
def end_continue(self, continue_):
self._write_status(continue_)
self._writer.end('continue')
[docs]
def start_break(self, break_):
self._writer.start('break')
[docs]
def end_break(self, break_):
self._write_status(break_)
self._writer.end('break')
[docs]
def start_error(self, error):
self._writer.start('error')
[docs]
def end_error(self, error):
for value in error.values:
self._writer.element('value', value)
self._write_status(error)
self._writer.end('error')
[docs]
def start_test(self, test):
self._writer.start('test', {'id': test.id, 'name': test.name,
'line': str(test.lineno or '')})
[docs]
def end_test(self, test):
self._writer.element('doc', test.doc)
self._write_list('tag', test.tags)
if test.timeout:
self._writer.element('timeout', attrs={'value': str(test.timeout)})
self._write_status(test)
self._writer.end('test')
[docs]
def start_suite(self, suite):
attrs = {'id': suite.id, 'name': suite.name}
if suite.source:
attrs['source'] = str(suite.source)
self._writer.start('suite', attrs)
[docs]
def end_suite(self, suite):
self._writer.element('doc', suite.doc)
for name, value in suite.metadata.items():
self._writer.element('meta', value, {'name': name})
self._write_status(suite)
self._writer.end('suite')
[docs]
def start_statistics(self, stats):
self._writer.start('statistics')
[docs]
def end_statistics(self, stats):
self._writer.end('statistics')
[docs]
def start_total_statistics(self, total_stats):
self._writer.start('total')
[docs]
def end_total_statistics(self, total_stats):
self._writer.end('total')
[docs]
def start_tag_statistics(self, tag_stats):
self._writer.start('tag')
[docs]
def end_tag_statistics(self, tag_stats):
self._writer.end('tag')
[docs]
def start_suite_statistics(self, tag_stats):
self._writer.start('suite')
[docs]
def end_suite_statistics(self, tag_stats):
self._writer.end('suite')
[docs]
def visit_stat(self, stat):
self._writer.element('stat', stat.name,
stat.get_attributes(values_as_strings=True))
[docs]
def start_errors(self, errors=None):
self._writer.start('errors')
[docs]
def end_errors(self, errors=None):
self._writer.end('errors')
def _write_list(self, tag, items):
for item in items:
self._writer.element(tag, item)
def _write_status(self, item):
attrs = {'status': item.status,
'start': item.start_time.isoformat() if item.start_time else None,
'elapsed': format(item.elapsed_time.total_seconds(), 'f')}
self._writer.element('status', item.message, attrs)
[docs]
class LegacyXmlLogger(XmlLogger):
def _get_start_attrs(self, rpa, generator):
return {'generator': get_full_version(generator),
'generated': self._datetime_to_timestamp(datetime.now()),
'rpa': 'true' if rpa else 'false',
'schemaversion': '4'}
def _datetime_to_timestamp(self, dt):
if dt is None:
return None
return dt.isoformat(' ', timespec='milliseconds').replace('-', '')
def _get_start_keyword_attrs(self, kw):
attrs = {'name': kw.kwname, 'library': kw.libname}
if kw.type != 'KEYWORD':
attrs['type'] = kw.type
if kw.source_name:
attrs['sourcename'] = kw.source_name
return attrs
def _write_status(self, item):
attrs = {'status': item.status,
'starttime': self._datetime_to_timestamp(item.start_time),
'endtime': self._datetime_to_timestamp(item.end_time)}
if (isinstance(item, (TestSuite, TestCase))
or isinstance(item, Keyword) and item.type == 'TEARDOWN'):
message = item.message
else:
message = ''
self._writer.element('status', message, attrs)
def _write_message(self, msg):
ts = self._datetime_to_timestamp(msg.timestamp) if msg.timestamp else None
attrs = {'timestamp': ts, 'level': msg.level}
if msg.html:
attrs['html'] = 'true'
# Use `_xml_writer`, not `_writer` to write messages also when flattening.
self._xml_writer.element('msg', msg.message, attrs)