Fixed tests filter by ID and selectors (via #546)

Author: Stilver

allure-robotframework/src/listener/allure_testplan.py

@@ -8,22 +8,27 @@
 class allure_testplan(SuiteVisitor):
     def __init__(self):
         self.testplan = get_testplan()
+        self.allure_ids = [test["id"] for test in self.testplan] if self.testplan else []
+        self.selectors = [test["selector"] for test in self.testplan] if self.testplan else []
 
     def start_suite(self, suite):
         if self.testplan:
-            # included_tests = [test["selector"] for test in self.testplan]
-            included_tests = self.included_tests(suite)
-            if included_tests:
-                suite.filter(included_tests=self.included_tests(suite))
+            suite.tests = self.included_tests(suite)
 
     def included_tests(self, suite):
         included_tests = []
+
         for test in suite.tests:
             allure_id = None
             for label in allure_labels(test.tags):
                 if label.name == LabelType.ID:
                     allure_id = str(label.value)
-            if allure_id and any([allure_id == item.get("id", None) for item in self.testplan]):
-                included_tests.append(test.name)
+            if allure_id and allure_id in self.allure_ids:
+                included_tests.append(test)
+            elif test.longname in self.selectors:
+                included_tests.append(test)
+
+        return included_tests
 
-        return included_tests or [test["selector"] for test in self.testplan]
+    def end_suite(self, suite):
+        suite.suites = [s for s in suite.suites if s.test_count > 0]