robotcode-modifiers 0.82.0__tar.gz → 0.82.2__tar.gz

{robotcode_modifiers-0.82.0 → robotcode_modifiers-0.82.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: robotcode-modifiers
-Version: 0.82.0
+Version: 0.82.2
 Summary: Some Robot Framework Modifiers for RobotCode
 Project-URL: Homepage, https://robotcode.io
 Project-URL: Donate, https://opencollective.com/robotcode

robotcode_modifiers-0.82.2/src/robotcode/modifiers/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.82.2"

robotcode_modifiers-0.82.2/src/robotcode/modifiers/longname_modifiers.py

@@ -0,0 +1,61 @@
+from typing import Optional, Union
+
+from robot.api import SuiteVisitor
+from robot.running import TestCase, TestSuite
+
+
+class _BaseSuiteVisitor(SuiteVisitor):
+    def __init__(self, *included: str, root_name: Optional[str] = None) -> None:
+        super().__init__()
+        self.included = list(included)
+
+        self.root_name = root_name
+        self.real_root_name: Optional[str] = None
+
+    def start_suite(self, suite: TestSuite) -> None:
+        if self.real_root_name is None and self.root_name is not None:
+            self.real_root_name = suite.longname
+            new_included = []
+            for i in self.included:
+                if "." in i:
+                    root_name, rest = c if len(c := i.split(".", 1)) > 1 else (c, None)
+                    if root_name == self.root_name:
+                        if rest is not None:
+                            new_included.append(f"{self.real_root_name}.{rest}")
+                        else:
+                            new_included.append(f"{self.real_root_name}'")
+                    else:
+                        new_included.append(i)
+                else:
+                    new_included.append(i)
+            self.included = new_included
+
+    def _is_included(self, test: Union[TestCase, TestSuite]) -> bool:
+        names = []
+        names.append(test.longname)
+        current = test.parent
+        while current:
+            if current.parent is None:
+                names.append(self.root_name if self.root_name is not None else current.longname)
+            else:
+                names.append(current.longname)
+            current = current.parent
+
+        return any((s in names) for s in self.included)
+
+    def end_suite(self, suite: TestSuite) -> None:
+        suite.suites = [s for s in suite.suites if s.test_count > 0]
+
+
+class ByLongName(_BaseSuiteVisitor):
+    def start_suite(self, suite: TestSuite) -> None:
+        super().start_suite(suite)
+
+        suite.tests = [t for t in suite.tests if self._is_included(t)]
+
+
+class ExcludedByLongName(_BaseSuiteVisitor):
+    def start_suite(self, suite: TestSuite) -> None:
+        super().start_suite(suite)
+
+        suite.tests = [t for t in suite.tests if not self._is_included(t)]

robotcode_modifiers-0.82.0/src/robotcode/modifiers/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.82.0"

robotcode_modifiers-0.82.0/src/robotcode/modifiers/longname_modifiers.py

@@ -1,48 +0,0 @@
-from typing import Union
-
-from robot.api import SuiteVisitor
-from robot.running import TestCase, TestSuite
-
-
-class ByLongName(SuiteVisitor):
-    def __init__(self, *included: str) -> None:
-        super().__init__()
-        self.included = included
-
-    def start_suite(self, suite: TestSuite) -> None:
-        suite.tests = [t for t in suite.tests if self._is_included(t)]
-
-    def _is_included(self, test: Union[TestCase, TestSuite]) -> bool:
-        names = []
-        names.append(test.longname)
-        current = test.parent
-        while current:
-            names.append(current.longname)
-            current = current.parent
-
-        return any((s in names) for s in self.included)
-
-    def end_suite(self, suite: TestSuite) -> None:
-        suite.suites = [s for s in suite.suites if s.test_count > 0]
-
-
-class ExcludedByLongName(SuiteVisitor):
-    def __init__(self, *included: str) -> None:
-        super().__init__()
-        self.included = included
-
-    def start_suite(self, suite: TestSuite) -> None:
-        suite.tests = [t for t in suite.tests if not self._is_included(t)]
-
-    def _is_included(self, test: Union[TestCase, TestSuite]) -> bool:
-        names = []
-        names.append(test.longname)
-        current = test.parent
-        while current:
-            names.append(current.longname)
-            current = current.parent
-
-        return any((s in names) for s in self.included)
-
-    def end_suite(self, suite: TestSuite) -> None:
-        suite.suites = [s for s in suite.suites if s.test_count > 0]