diff --git a/packages/roboview/roboview/models/robot_parsing/called_keyword_parsing.py b/packages/roboview/roboview/models/robot_parsing/called_keyword_parsing.py index 7a51a4e..5d04c25 100644 --- a/packages/roboview/roboview/models/robot_parsing/called_keyword_parsing.py +++ b/packages/roboview/roboview/models/robot_parsing/called_keyword_parsing.py @@ -8,6 +8,8 @@ from robot.parsing.model.statements import ( KeywordCall, Setup, + SuiteSetup, + SuiteTeardown, Teardown, TestSetup, TestTeardown, @@ -51,6 +53,24 @@ def visit_KeywordCall(self, node: KeywordCall) -> None: # noqa: N802 """ self._add_keyword(node.keyword) + def visit_SuiteSetup(self, node: SuiteSetup) -> None: # noqa: N802 + """Visit a SuiteSetup node and extract the keyword. + + Arguments: + node (SuiteSetup): SuiteSetup node in the AST. + + """ + self._add_keyword(node.name) + + def visit_SuiteTeardown(self, node: SuiteTeardown) -> None: # noqa: N802 + """Visit a SuiteTeardown node and extract the keyword. + + Arguments: + node (SuiteTeardown): SuiteTeardown node in the AST. + + """ + self._add_keyword(node.name) + def visit_Setup(self, node: Setup) -> None: # noqa: N802 """Visit a Setup node and extract the keyword.