Examples with DirectGraphCommandExecutionContext org.kie.workbench.common.stunner.core.graph.command.DirectGraphCommandExecutionContext used on opensource projects

Example 1 with DirectGraphCommandExecutionContext

use of org.kie.workbench.common.stunner.core.graph.command.DirectGraphCommandExecutionContext in project kie-wb-common by kiegroup.

the class NodeProxyTest method testCreateTargetNodeInSomeParent.

@Test
@SuppressWarnings("all")
public void testCreateTargetNodeInSomeParent() {
    Node<View<?>, Edge> parentNode = new NodeImpl<>(PARENT_NODE_ID);
    parentNode.setContent(new ViewImpl<>(mock(Object.class), Bounds.create()));
    DirectGraphCommandExecutionContext context = new DirectGraphCommandExecutionContext(definitionManager, factoryManager, new MapIndexBuilder().build(graph));
    new AddNodeCommand(parentNode).execute(context);
    new SetChildrenCommand(parentNode, sourceNode).execute(context);
    CanvasCommand<AbstractCanvasHandler> addConnector = mock(CanvasCommand.class);
    CanvasCommand<AbstractCanvasHandler> addNode = mock(CanvasCommand.class);
    CanvasCommand<AbstractCanvasHandler> setTargetNode = mock(CanvasCommand.class);
    doReturn(addConnector).when(commandFactory).addConnector(eq(sourceNode), eq(edge), Mockito.<MagnetConnection>any(), eq(SHAPE_SET_ID));
    doReturn(addNode).when(commandFactory).addChildNode(eq(parentNode), eq(targetNode), eq(SHAPE_SET_ID));
    doReturn(setTargetNode).when(commandFactory).setTargetNode(eq(targetNode), eq(edge), any());
    verifyCreateTargetNode(addConnector, addNode, setTargetNode);
}