add tags to failed init descriptors too

helmethair-co · May 11, 2022 · f4a979c · f4a979c
1 parent a1e2b7f
commit f4a979c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/main/java/co/helmethair/scalatest/descriptor/ScalatestDescriptor.java b/src/main/java/co/helmethair/scalatest/descriptor/ScalatestDescriptor.java
@@ -16,6 +16,7 @@ public abstract class ScalatestDescriptor implements TestDescriptor {
     private final UniqueId id;
     private TestDescriptor parentDescriptor = null;
     private Set<TestDescriptor> childDescriptors = new HashSet<TestDescriptor>();
+    protected Set<TestTag> tags = new HashSet<>();
 
     protected ScalatestDescriptor(UniqueId id) {
         this.id = id;
@@ -75,7 +76,7 @@ public Set<TestDescriptor> getChildren() {
 
     @Override
     public Set<TestTag> getTags() {
-        return new HashSet<>();
+        return tags;
     }
 
     @Override

diff --git a/src/main/java/co/helmethair/scalatest/descriptor/ScalatestFailedInitDescriptor.java b/src/main/java/co/helmethair/scalatest/descriptor/ScalatestFailedInitDescriptor.java
@@ -1,18 +1,21 @@
 package co.helmethair.scalatest.descriptor;
 
 import org.junit.platform.engine.TestSource;
+import org.junit.platform.engine.TestTag;
 import org.junit.platform.engine.support.descriptor.MethodSource;
 
 import java.util.Optional;
+import java.util.Set;
 
 public class ScalatestFailedInitDescriptor extends ScalatestDescriptor {
     private final Throwable cause;
     private final String suiteId;
 
-    public ScalatestFailedInitDescriptor(Throwable cause, String name) {
+    public ScalatestFailedInitDescriptor(Throwable cause, String name, Set<TestTag> tags) {
         super(ENGINE_ID.append("failed", name));
         this.cause = cause;
         this.suiteId = name;
+        this.tags = tags;
     }
 
     @Override