use of org.kie.workbench.common.stunner.core.command.impl.CompositeCommand in project kie-wb-common by kiegroup.
the class PasteSelectionSessionCommandTest method executeWithMultiSelection.
@Test
@SuppressWarnings("unchecked")
public void executeWithMultiSelection() {
pasteSelectionSessionCommand.bind(session);
// Mock the callback of CloneNodeCommand
ArgumentCaptor<Consumer> consumerNode = ArgumentCaptor.forClass(Consumer.class);
ArgumentCaptor<Consumer> consumerNode2 = ArgumentCaptor.forClass(Consumer.class);
when(cloneNodeCommand.getCandidate()).thenReturn(node);
when(cloneNodeCommand2.getCandidate()).thenReturn(node2);
when(canvasCommandFactory.cloneNode(eq(node), any(), any(), consumerNode.capture())).thenReturn(cloneNodeCommand);
when(canvasCommandFactory.cloneNode(eq(node2), any(), any(), consumerNode2.capture())).thenReturn(cloneNodeCommand2);
Map<Node, ArgumentCaptor<Consumer>> consumerMap = new HashMap() {
{
put(node, consumerNode);
put(node2, consumerNode2);
}
};
// Mock the callback of CloneConnectorCommand
ArgumentCaptor<Consumer> consumerEdge = ArgumentCaptor.forClass(Consumer.class);
when(canvasCommandFactory.cloneConnector(any(), anyString(), anyString(), anyString(), consumerEdge.capture())).thenReturn(cloneConnectorCommand);
// apply callbacks mocks
when(sessionCommandManager.execute(eq(canvasHandler), any())).thenAnswer(param -> {
CompositeCommand argument = param.getArgumentAt(1, CompositeCommand.class);
// callback to nodes
argument.getCommands().stream().filter(c -> c instanceof CloneNodeCommand).forEach(c -> {
CloneNodeCommand cloneNodeCommand = (CloneNodeCommand) c;
Node candidate = cloneNodeCommand.getCandidate();
consumerMap.get(candidate).getValue().accept(cloneMap.get(candidate));
});
// callback to connectors
argument.getCommands().stream().filter(c -> c instanceof CloneConnectorCommand).forEach(c -> consumerEdge.getValue().accept(cloneEdge));
return commandResult;
});
// Executing the command
clipboardControl.set(graphInstance.startNode, graphInstance.edge1, graphInstance.intermNode);
when(selectionControl.getSelectedItems()).thenReturn(Arrays.asList(graphInstance.startNode.getUUID(), graphInstance.edge1.getUUID(), graphInstance.intermNode.getUUID()));
pasteSelectionSessionCommand.execute(callback);
verify(canvasCommandFactory, times(1)).cloneNode(eq(graphInstance.startNode), eq(graphInstance.parentNode.getUUID()), eq(new Point2D(X, DEFAULT_PADDING + Y + NODE_SIZE)), any());
verify(canvasCommandFactory, times(1)).cloneConnector(eq(graphInstance.edge1), anyString(), anyString(), anyString(), any());
// check command registry update after execution to allow a single undo/redo
verify(commandRegistry, times(2)).pop();
ArgumentCaptor<Command> commandArgumentCaptor = ArgumentCaptor.forClass(Command.class);
verify(commandRegistry, times(1)).register(commandArgumentCaptor.capture());
assertTrue(commandArgumentCaptor.getValue() instanceof CompositeCommand);
assertEquals(((CompositeCommand) commandArgumentCaptor.getValue()).size(), 2);
}
use of org.kie.workbench.common.stunner.core.command.impl.CompositeCommand in project kie-wb-common by kiegroup.
the class BPMNGraphFactoryImplTest method testBuild.
@Test
@SuppressWarnings("unchecked")
public void testBuild() {
final Node diagramNode = mock(Node.class);
final Node startEventNode = mock(Node.class);
when(factoryManager.newElement(anyString(), eq(BPMNDiagramImpl.class))).thenReturn(diagramNode);
when(factoryManager.newElement(anyString(), eq(StartNoneEvent.class))).thenReturn(startEventNode);
final Graph<DefinitionSet, Node> graph = tested.build("uuid1", "defSetId");
assertNotNull(graph);
assertEquals("uuid1", graph.getUUID());
assertEquals(1, graph.getLabels().size());
assertTrue(graph.getLabels().contains("defSetId"));
final ArgumentCaptor<Command> commandCaptor = ArgumentCaptor.forClass(Command.class);
verify(graphCommandFactory, times(1)).addNode(eq(diagramNode));
verify(graphCommandFactory, times(1)).addChildNode(eq(diagramNode), eq(startEventNode), eq(new Point2D(BPMNGraphFactoryImpl.START_X, BPMNGraphFactoryImpl.START_Y)));
verify(graphCommandManager, times(1)).execute(any(GraphCommandExecutionContext.class), commandCaptor.capture());
final Command command = commandCaptor.getValue();
assertTrue(command instanceof CompositeCommand);
final CompositeCommand compositeCommand = (CompositeCommand) command;
assertEquals(2, compositeCommand.size());
}
use of org.kie.workbench.common.stunner.core.command.impl.CompositeCommand in project kie-wb-common by kiegroup.
the class CloneConnectorCommandTest method newCanvasCommand.
@Test
@SuppressWarnings("unchecked")
public void newCanvasCommand() {
Command<AbstractCanvasHandler, CanvasViolation> command = cloneConnectorCommand.newCanvasCommand(context);
assertTrue(command instanceof CompositeCommand);
Consumer<Edge> cloneCallback = cloneConnectorCommand.getCloneCallback();
cloneCallback.accept(clone);
CompositeCommand compositeCommand = (CompositeCommand) command;
assertTrue(compositeCommand.getCommands().stream().anyMatch(c -> c instanceof AddCanvasConnectorCommand));
assertEquals(((AddCanvasConnectorCommand) compositeCommand.getCommands().stream().filter(c -> c instanceof AddCanvasConnectorCommand).findFirst().get()).getCandidate(), clone);
assertTrue(compositeCommand.getCommands().stream().anyMatch(c -> c instanceof SetCanvasConnectionCommand));
assertEquals(((SetCanvasConnectionCommand) compositeCommand.getCommands().stream().filter(c -> c instanceof SetCanvasConnectionCommand).findFirst().get()).getEdge(), clone);
}
use of org.kie.workbench.common.stunner.core.command.impl.CompositeCommand in project kie-wb-common by kiegroup.
the class GridFactoryCommandUtils method assertCommands.
public static void assertCommands(final org.kie.workbench.common.stunner.core.command.Command command, final Class... commands) {
assertThat(command).isInstanceOf(CompositeCommand.class);
final CompositeCommand compositeCommand = (CompositeCommand) command;
assertThat(compositeCommand.getCommands()).hasSize(commands.length);
for (int i = 0; i < commands.length; i++) {
assertThat(compositeCommand.getCommands().get(i)).isInstanceOf(commands[i]);
}
}
Aggregations