ru.tinkoff.kora.test.extension.junit5.GraphReplacementNoDeps Maven / Gradle / Ivy

Go to download

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

Kora test-junit5 module

The newest version!

package ru.tinkoff.kora.test.extension.junit5;

import org.junit.jupiter.api.extension.ExtensionConfigurationException;
import ru.tinkoff.kora.application.graph.ApplicationGraphDraw;
import ru.tinkoff.kora.application.graph.Node;

import java.util.function.Function;

record GraphReplacementNoDeps(Function function,
                                 GraphCandidate candidate) implements GraphModification {

    @Override
    public void accept(ApplicationGraphDraw graphDraw) {
        var nodesToReplace = GraphUtils.findNodeByTypeOrAssignable(graphDraw, candidate());
        if (nodesToReplace.isEmpty()) {
            throw new ExtensionConfigurationException("Can't find Nodes to Replace: " + candidate());
        }

        for (var nodeToReplace : nodesToReplace) {
            @SuppressWarnings("unchecked")
            var casted = (Node) nodeToReplace;
            graphDraw.replaceNode(casted, g -> function.apply(new DefaultKoraAppGraph(graphDraw, g)));
        }
    }
}