munit.internal.junitinterface.TagFilter Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of junit-interface Show documentation

A Java implementation of sbt's test interface for JUnit 4

There is a newer version: 1.0.0

package munit.internal.junitinterface;

import org.junit.runner.Description;
import org.junit.runner.manipulation.Filter;

import java.lang.annotation.Annotation;
import java.util.Set;

public class TagFilter extends Filter {
    final Set includeTags, excludeTags;

    public TagFilter(Set includeTags, Set excludeTags) {
        this.includeTags = includeTags;
        this.excludeTags = excludeTags;
    }

    @Override
    public boolean shouldRun(Description description) {
        if (includeTags.isEmpty() && excludeTags.isEmpty()) return true;
        boolean isIncluded = includeTags.isEmpty();
        for (Annotation annotation : description.getAnnotations()) {
            if (annotation instanceof Tag) {
                Tag tag = (Tag) annotation;
                isIncluded = isIncluded || includeTags.contains(tag.value());
                boolean isExcluded = excludeTags.contains(tag.value());
                if (isExcluded) {
                    return false;
                }
            }
        }
        return isIncluded;
    }

    @Override
    public String toString() {
        return "TagFilter{" +
                "includeTags=" + includeTags +
                ", excludeTags=" + excludeTags +
                '}';
    }

    @Override
    public String describe() {
        return toString();
    }
}