io.cucumber.junit.platform.engine.TestDescriptorOnLine Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cucumber-junit-platform-engine Show documentation
There is a newer version: 7.20.1
Show newest version
package io.cucumber.junit.platform.engine;

import io.cucumber.core.feature.FeatureWithLines;
import org.junit.platform.engine.TestDescriptor;
import org.junit.platform.engine.discovery.ClasspathResourceSelector;
import org.junit.platform.engine.discovery.FileSelector;
import org.junit.platform.engine.discovery.UriSelector;
import org.junit.platform.engine.support.descriptor.ClasspathResourceSource;
import org.junit.platform.engine.support.descriptor.FileSource;

import java.util.Optional;
import java.util.function.Predicate;

import static org.junit.platform.engine.support.descriptor.FilePosition.fromQuery;

@SuppressWarnings("Convert2MethodRef")
class TestDescriptorOnLine {

    static Predicate testDescriptorOnLine(int line) {
        return descriptor -> descriptor.getSource()
                .flatMap(testSource -> {
                    if (testSource instanceof FileSource) {
                        FileSource fileSystemSource = (FileSource) testSource;
                        return fileSystemSource.getPosition();
                    }
                    if (testSource instanceof ClasspathResourceSource) {
                        ClasspathResourceSource classpathResourceSource = (ClasspathResourceSource) testSource;
                        return classpathResourceSource.getPosition();
                    }
                    return Optional.empty();
                })
                .map(filePosition -> filePosition.getLine())
                .map(testSourceLine -> line == testSourceLine)
                .orElse(false);
    }

    private static boolean anyTestDescriptor(TestDescriptor testDescriptor) {
        return true;
    }

    private static Predicate eitherTestDescriptor(
            Predicate a, Predicate b
    ) {
        return a.or(b);
    }

    static Predicate from(FeatureWithLines selector) {
        return selector.lines().stream()
                .map(TestDescriptorOnLine::testDescriptorOnLine)
                .reduce(TestDescriptorOnLine::eitherTestDescriptor)
                .orElse(TestDescriptorOnLine::anyTestDescriptor);
    }

    static Predicate from(UriSelector selector) {
        String query = selector.getUri().getQuery();
        return fromQuery(query)
                .map(filePosition -> filePosition.getLine())
                .map(TestDescriptorOnLine::testDescriptorOnLine)
                .orElse(TestDescriptorOnLine::anyTestDescriptor);
    }

    static Predicate from(ClasspathResourceSelector selector) {
        return selector.getPosition()
                .map(filePosition -> filePosition.getLine())
                .map(TestDescriptorOnLine::testDescriptorOnLine)
                .orElse(TestDescriptorOnLine::anyTestDescriptor);
    }

    static Predicate from(FileSelector selector) {
        return selector.getPosition()
                .map(filePosition -> filePosition.getLine())
                .map(TestDescriptorOnLine::testDescriptorOnLine)
                .orElse(TestDescriptorOnLine::anyTestDescriptor);
    }

}