net.serenitybdd.junit5.JunitTestHierarchy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of serenity-junit5 Show documentation

There is a newer version: 4.2.9

package net.serenitybdd.junit5;

import org.junit.platform.engine.support.descriptor.ClassSource;
import org.junit.platform.launcher.TestIdentifier;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class JunitTestHierarchy {
    private static final JunitTestHierarchy INSTANCE = new JunitTestHierarchy();

    private final Map classPathToDisplayName = new ConcurrentHashMap<>();
    private final Map idToDisplayName = new ConcurrentHashMap<>();
    private final Map idToParent = new ConcurrentHashMap<>();

    public static JunitTestHierarchy getInstance() {
        return INSTANCE;
    }

    public void testSuiteStarted(TestIdentifier testIdentifier) {
        idToDisplayName.put(testIdentifier.getUniqueId(), testIdentifier.getDisplayName());
        if (testIdentifier.getSource().isPresent() && testIdentifier.getSource().get() instanceof ClassSource) {
            String className = ((ClassSource) testIdentifier.getSource().get()).getClassName();
            classPathToDisplayName.put(className, testIdentifier.getDisplayName());
        }
        testIdentifier.getParentId().ifPresent(
                parentId -> idToParent.put(testIdentifier.getUniqueId(), parentId)
        );

    }

}