/* * Copyright 2016 Red Hat, Inc. and/or its affiliates. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at *   *     http://www.apache.org/licenses/LICENSE-2.0 *   * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.kie.workbench.common.stunner.core.client.canvas.command; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.kie.workbench.common.stunner.core.client.command.CanvasViolation; import org.kie.workbench.common.stunner.core.client.shape.EdgeShape; import org.kie.workbench.common.stunner.core.client.shape.MutationContext; import org.kie.workbench.common.stunner.core.client.shape.Shape; import org.kie.workbench.common.stunner.core.client.shape.view.ShapeView; import org.kie.workbench.common.stunner.core.command.CommandResult; import org.kie.workbench.common.stunner.core.graph.Edge; import org.kie.workbench.common.stunner.core.graph.Node; import org.mockito.Mock; import org.mockito.runners.MockitoJUnitRunner; import static org.junit.Assert.*; import static org.mockito.Matchers.any; import static org.mockito.Matchers.eq; import static org.mockito.Mockito.*; @RunWith(MockitoJUnitRunner.class) public class AddCanvasConnectorCommandTest extends AbstractCanvasCommandTest { private static final String EDGE_ID = "e1"; private static final String SOURCE_ID = "s1"; @Mock private Edge candidate; @Mock private Node source; @Mock private EdgeShape candidateShape; @Mock private Shape sourceShape; @Mock private ShapeView sourceShapeView; private AddCanvasConnectorCommand tested; @Before public void setup() throws Exception { super.setup(); when(candidate.getUUID()).thenReturn(EDGE_ID); when(source.getUUID()).thenReturn(SOURCE_ID); when(candidate.getSourceNode()).thenReturn(source); when(canvas.getShape(eq(EDGE_ID))).thenReturn(candidateShape); when(canvas.getShape(eq(SOURCE_ID))).thenReturn(sourceShape); when(sourceShape.getShapeView()).thenReturn(sourceShapeView); this.tested = new AddCanvasConnectorCommand(candidate, SHAPE_SET_ID); } @Test @SuppressWarnings("unchecked") public void testExecute() { final CommandResult<CanvasViolation> result = tested.execute(canvasHandler); assertNotEquals(CommandResult.Type.ERROR, result.getType()); verify(canvasHandler, times(1)).register(eq(SHAPE_SET_ID), eq(candidate)); verify(canvasHandler, times(1)).applyElementMutation(eq(candidate), any(MutationContext.class)); verify(candidateShape, times(1)).applyConnections(eq(candidate), eq(sourceShapeView), eq(null), any(MutationContext.class)); verify(canvasHandler, times(1)).notifyCanvasElementUpdated(eq(source)); } }