refactor: add type parameter to end_output_group

Author: d-biehl

robotcode/debugger/debugger.py

@@ -646,7 +646,7 @@ def start_output_group(self, name: str, attributes: Dict[str, Any], type: Option
                 ),
             )
 
-    def end_output_group(self, name: str, attributes: Dict[str, Any]) -> None:
+    def end_output_group(self, name: str, attributes: Dict[str, Any], type: Optional[str] = None) -> None:
         if self.group_output:
             source = attributes.get("source", None)
             line_no = attributes.get("lineno", None)

robotcode/debugger/listeners.py

@@ -44,7 +44,7 @@ def end_suite(self, name: str, attributes: Dict[str, Any]) -> None:
 
         Debugger.instance().end_suite(name, attributes)
 
-        Debugger.instance().end_output_group(name, attributes)
+        Debugger.instance().end_output_group(name, attributes, "SUITE")
 
         Debugger.instance().send_event(
             self,
@@ -82,7 +82,7 @@ def start_test(self, name: str, attributes: Dict[str, Any]) -> None:
     def end_test(self, name: str, attributes: Dict[str, Any]) -> None:
         Debugger.instance().end_test(name, attributes)
 
-        Debugger.instance().end_output_group(name, attributes)
+        Debugger.instance().end_output_group(name, attributes, "TEST")
 
         Debugger.instance().send_event(
             self,
@@ -114,7 +114,7 @@ def end_keyword(self, name: str, attributes: Dict[str, Any]) -> None:
         Debugger.instance().end_keyword(name, attributes)
 
         if attributes["type"] in ["KEYWORD", "SETUP", "TEARDOWN"]:
-            Debugger.instance().end_output_group(name, attributes)
+            Debugger.instance().end_output_group(name, attributes, attributes.get("type", None))
 
             if attributes["status"] == "FAIL" and attributes.get("source", None):
                 if self.failed_keywords is None: