fixed RF 7 rerunfailed, due to Variable moved from robot.running.model

Signed-off-by: Rene <snooz@posteo.de>
Snooz82 · Mar 29, 2024 · e6eeb75 · e6eeb75
1 parent a5df582
commit e6eeb75
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/DataDriver/DataDriver.py b/src/DataDriver/DataDriver.py
@@ -51,7 +51,7 @@
     warn,
 )
 
-__version__ = "1.11.0"
+__version__ = "1.11.1"
 
 
 class DataDriver:

diff --git a/src/DataDriver/rerunfailed.py b/src/DataDriver/rerunfailed.py
@@ -5,7 +5,10 @@
 from pathlib import Path
 
 from robot.api import ExecutionResult, ResultVisitor, SuiteVisitor  # type: ignore
-from robot.running.model import Variable  # type: ignore
+try:
+    from robot.running.model import Variable  # type: ignore
+except ImportError:
+    from robot.running.resourcemodel import Variable  # type: ignore / robotframework>=7.0
 
 
 class rerunfailed(SuiteVisitor):
@@ -23,7 +26,7 @@ def start_suite(self, suite):
             suite.tests.clear()
             return
         if self._suite_is_data_driven(suite):
-            dynamic_tests = Variable("@{DYNAMICTESTS}", self._failed_tests, suite.source)
+            dynamic_tests = Variable("${DYNAMICTESTS}", self._failed_tests, suite.source)
             suite.resource.variables.append(dynamic_tests)
         else:
             suite.tests = [