Examples with ControlHandleList com.ait.lienzo.client.core.shape.wires.ControlHandleList used on opensource projects

Example 1 with ControlHandleList

use of com.ait.lienzo.client.core.shape.wires.ControlHandleList in project kie-wb-common by kiegroup.

the class WiresConnectorViewTest method testConnectMagnetsByLocation.

@Test
public void testConnectMagnetsByLocation() {
    // Mocks for head stuff.
    IControlHandleList headControlHandleList = new ControlHandleList(mock(Shape.class));
    WiresMagnet headMagnet1 = mock(WiresMagnet.class);
    when(headMagnet1.getX()).thenReturn(10d);
    when(headMagnet1.getY()).thenReturn(20d);
    when(headMagnet1.getIndex()).thenReturn(0);
    IPrimitive ch1 = mock(IPrimitive.class);
    when(ch1.getLocation()).thenReturn(new Point2D(10d, 20d));
    when(headMagnet1.getControl()).thenReturn(ch1);
    WiresMagnet headMagnet2 = mock(WiresMagnet.class);
    when(headMagnet2.getX()).thenReturn(100d);
    when(headMagnet2.getY()).thenReturn(200d);
    IPrimitive ch2 = mock(IPrimitive.class);
    when(ch2.getLocation()).thenReturn(new Point2D(100d, 200d));
    when(headMagnet2.getControl()).thenReturn(ch2);
    headControlHandleList.add(headMagnet1);
    headControlHandleList.add(headMagnet2);
    MagnetManager.Magnets headMagnets = mock(MagnetManager.Magnets.class);
    when(headMagnets.getMagnets()).thenReturn(headControlHandleList);
    Point2D headAbs = new Point2D(0, 0);
    MagnetConnection headConnection = MagnetConnection.Builder.at(10, 20);
    // Mocks for head tail.
    IControlHandleList tailControlHandleList = new ControlHandleList(mock(Shape.class));
    WiresMagnet tailMagnet1 = mock(WiresMagnet.class);
    when(tailMagnet1.getX()).thenReturn(10d);
    when(tailMagnet1.getY()).thenReturn(20d);
    IPrimitive th1 = mock(IPrimitive.class);
    when(th1.getLocation()).thenReturn(new Point2D(10d, 20d));
    when(tailMagnet1.getControl()).thenReturn(th1);
    WiresMagnet tailMagnet2 = mock(WiresMagnet.class);
    when(tailMagnet2.getX()).thenReturn(100d);
    when(tailMagnet2.getY()).thenReturn(200d);
    when(tailMagnet2.getIndex()).thenReturn(2);
    IPrimitive th2 = mock(IPrimitive.class);
    when(th2.getLocation()).thenReturn(new Point2D(100d, 200d));
    when(tailMagnet2.getControl()).thenReturn(th2);
    tailControlHandleList.add(tailMagnet1);
    tailControlHandleList.add(tailMagnet2);
    MagnetManager.Magnets tailMagnets = mock(MagnetManager.Magnets.class);
    when(tailMagnets.getMagnets()).thenReturn(tailControlHandleList);
    Point2D tailAbs = new Point2D(0, 0);
    MagnetConnection tailConnection = MagnetConnection.Builder.at(100, 200);
    // Mocks both source/target actual connections.
    WiresConnection headWiresConnection = mock(WiresConnection.class);
    WiresConnection tailWiresConnection = mock(WiresConnection.class);
    tested.setHeadConnection(headWiresConnection);
    tested.setTailConnection(tailWiresConnection);
    // Perform the connection.
    tested.connect(headMagnets, headAbs, headConnection, tailMagnets, tailAbs, tailConnection);
    // Verify it sets the right magnets (closest to the location):
    verify(headWiresConnection, never()).setXOffset(anyDouble());
    verify(headWiresConnection, never()).setYOffset(anyDouble());
    verify(headWiresConnection, times(1)).setAutoConnection(eq(false));
    verify(headWiresConnection, times(1)).setMagnet(eq(headMagnet1));
    verify(tailWiresConnection, times(1)).setXOffset(eq(0d));
    verify(tailWiresConnection, times(1)).setYOffset(eq(0d));
    verify(tailWiresConnection, times(1)).setAutoConnection(eq(false));
    verify(tailWiresConnection, times(1)).setMagnet(eq(tailMagnet2));
}