failure exceptiontype in output.xml

mkorpela · mkorpela · commit d4886ffde769 · 2020-09-17T10:11:40.000+03:00
diff --git a/src/robot/output/loggerhelper.py b/src/robot/output/loggerhelper.py
@@ -49,30 +49,31 @@ def info(self, msg):
     def warn(self, msg):
         self.write(msg, 'WARN')
 
-    def fail(self, msg):
+    def fail(self, msg, error_type=None):
         html = False
         if msg.startswith("*HTML*"):
             html = True
             msg = msg[6:].lstrip()
-        self.write(msg, 'FAIL', html)
+        self.write(msg, 'FAIL', html, error_type)
 
     def error(self, msg):
         self.write(msg, 'ERROR')
 
-    def write(self, message, level, html=False):
-        self.message(Message(message, level, html))
+    def write(self, message, level, html=False, error_type=None):
+        self.message(Message(message, level, html, error_type=error_type))
 
     def message(self, msg):
         raise NotImplementedError(self.__class__)
 
 
 class Message(BaseMessage):
-    __slots__ = ['_message']
+    __slots__ = ['_message', 'error_type']
 
-    def __init__(self, message, level='INFO', html=False, timestamp=None):
+    def __init__(self, message, level='INFO', html=False, timestamp=None, error_type=None):
         message = self._normalize_message(message)
         level, html = self._get_level_and_html(level, html)
         timestamp = timestamp or get_timestamp()
+        self.error_type=error_type
         BaseMessage.__init__(self, message, level, html, timestamp)
 
     def _normalize_message(self, msg):
diff --git a/src/robot/output/xmllogger.py b/src/robot/output/xmllogger.py
@@ -58,6 +58,8 @@ def log_message(self, msg):
 
     def _write_message(self, msg):
         attrs = {'timestamp': msg.timestamp or 'N/A', 'level': msg.level}
+        if msg.error_type:
+            attrs['exceptiontype'] = msg.error_type.__name__
         if msg.html:
             attrs['html'] = 'yes'
         self._writer.element('msg', msg.message, attrs)
diff --git a/src/robot/running/context.py b/src/robot/running/context.py
@@ -194,5 +194,5 @@ def info(self, message):
     def warn(self, message):
         self.output.warn(message)
 
-    def fail(self, message):
-        self.output.fail(message)
+    def fail(self, message, error_type=None):
+        self.output.fail(message, error_type)
diff --git a/src/robot/running/statusreporter.py b/src/robot/running/statusreporter.py
@@ -69,7 +69,7 @@ def _get_failure(self, exc_type, exc_value, exc_tb, context):
         failure = HandlerExecutionFailed(ErrorDetails(exc_info))
         if failure.timeout:
             context.timeout_occurred = True
-        context.fail(failure.full_message)
+        context.fail(failure.full_message, exc_type)
         if failure.traceback:
             context.debug(failure.traceback)
         return failure
