use of org.drools.impact.analysis.graph.ReactivityType in project drools by kiegroup.
the class AbstractGraphTest method assertLink.
/**
* Assert that there are exact links with the types between source node and target node.
* If no expectedTypes, it means there is no link
*/
protected void assertLink(Graph graph, String sourceFqdn, String targetFqdn, ReactivityType... expectedTypes) {
Node source = graph.getNodeMap().get(sourceFqdn);
Node target = graph.getNodeMap().get(targetFqdn);
List<Link> outgoingLinks = source.getOutgoingLinks().stream().filter(l -> l.getTarget().equals(target)).collect(Collectors.toList());
List<Link> incomingLinks = target.getIncomingLinks().stream().filter(l -> l.getSource().equals(source)).collect(Collectors.toList());
assertThat(outgoingLinks).hasSameElementsAs(incomingLinks);
List<ReactivityType> outgoingLinkTypelist = outgoingLinks.stream().map(l -> l.getReactivityType()).collect(Collectors.toList());
List<ReactivityType> expectedTypeList = Arrays.asList(expectedTypes);
assertThat(outgoingLinkTypelist).hasSameElementsAs(expectedTypeList);
}
Aggregations