ru.tinkoff.kora.test.extension.junit5.GraphAddition 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 ru.tinkoff.kora.application.graph.ApplicationGraphDraw;

import java.util.function.Function;

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

    private static final Class[] TAGS_EMPTY = new Class[]{};

    @Override
    public void accept(ApplicationGraphDraw graphDraw) {
        var tags = (candidate().tags().isEmpty())
                ? TAGS_EMPTY
                : candidate().tagsAsArray();

        graphDraw.addNode0(candidate().type(), tags, g -> function.apply(new DefaultKoraAppGraph(graphDraw, g)));
    }
}