eu.drus.jpa.unit.neo4j.dataset.Node Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jpa-unit-neo4j Show documentation

JUnit extension for simple testing of JPA entities and components

The newest version!

package eu.drus.jpa.unit.neo4j.dataset;

import static java.util.stream.Collectors.toList;
import static org.neo4j.cypherdsl.CypherQuery.node;
import static org.neo4j.cypherdsl.CypherQuery.value;

import java.util.List;

import org.apache.commons.lang3.builder.ToStringBuilder;
import org.apache.commons.lang3.builder.ToStringStyle;
import org.neo4j.cypherdsl.CypherQuery;
import org.neo4j.cypherdsl.Path;

public class Node extends GraphElement {

    Node(final String id, final List labels, final List attributes) {
        super(id, labels, attributes);
    }

    public PathBuilder toPath() {
        return new PathBuilder();
    }

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

    public String asString() {
        final ToStringBuilder builder = new ToStringBuilder(this, ToStringStyle.SHORT_PREFIX_STYLE);
        builder.append("id", getId());
        builder.append("labels", getLabels());
        builder.append("attributes", getAttributes());
        return builder.build();
    }

    public class PathBuilder {

        private Path path;

        private PathBuilder() {
            path = node(getId()).labels(getLabels().stream().map(CypherQuery::label).collect(toList()));
        }

        public PathBuilder withId(final String id) {
            path = node(id).labels(getLabels().stream().map(CypherQuery::label).collect(toList()));
            return this;
        }

        public PathBuilder withAllAttributes() {
            path = path.values(getAttributes().stream().map(a -> value(a.getName(), a.getValue())).collect(toList()));
            return this;
        }

        public PathBuilder withIdAttributes() {
            path = path
                    .values(getAttributes().stream().filter(Attribute::isId).map(a -> value(a.getName(), a.getValue())).collect(toList()));
            return this;
        }

        public Path build() {
            return path;
        }
    }
}