コード例 #1
0
  /** Переход от порта к методу, если поступило данное сообщение */
  @Ignore // TODO
  @Test
  public void portToMethod() throws Exception {

    final XShape method = Mockito.mock(XShape.class, "method");
    Mockito.when(shapeHelperWrapper.getNodeType(method)).thenReturn(Node.NodeType.MethodOfProcess);

    final XShape serverPort = Mockito.mock(XShape.class, "serverPort");
    Mockito.when(shapeHelperWrapper.getNodeType(serverPort)).thenReturn(Node.NodeType.ServerPort);

    // link start
    XShape link1Shape = Mockito.mock(XShape.class, "link1Shape");
    final XConnectorShape link1Connector =
        Mockito.mock(XConnectorShape.class, "link1ConnectorShape");
    XPropertySet link1PropertySet = Mockito.mock(XPropertySet.class, "link1PropertySet");

    XShape link2Shape = Mockito.mock(XShape.class, "link2Shape");
    final XConnectorShape link2Connector =
        Mockito.mock(XConnectorShape.class, "link2ConnectorShape");
    XPropertySet link2PropertySet = Mockito.mock(XPropertySet.class, "link2PropertySet");

    XShape linkTextShape = Mockito.mock(XShape.class, "linkTextShape");
    XPropertySet linkTextPropertySet = Mockito.mock(XPropertySet.class);

    Mockito.when(shapeHelperWrapper.isConnectorShape(link1Shape)).thenReturn(true);
    Mockito.when(shapeHelperWrapper.isConnectorShape(link2Shape)).thenReturn(true);

    Mockito.when(link1PropertySet.getPropertyValue(ConnectedShapes.START_SHAPE))
        .thenAnswer(
            new Answer<Object>() {
              @Override
              public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
                return serverPort;
              }
            });

    Mockito.when(link1PropertySet.getPropertyValue(ConnectedShapes.START_SHAPE))
        .thenReturn(serverPort);

    Mockito.when(link1PropertySet.getPropertyValue(ConnectedShapes.END_SHAPE))
        .thenReturn(linkTextShape);

    Mockito.when(link2PropertySet.getPropertyValue(ConnectedShapes.START_SHAPE))
        .thenReturn(linkTextShape);

    Mockito.when(link2PropertySet.getPropertyValue(ConnectedShapes.END_SHAPE)).thenReturn(method);

    Mockito.when(unoRuntimeWrapper.queryInterface(XPropertySet.class, link1Connector))
        .thenReturn(link1PropertySet);

    Mockito.when(unoRuntimeWrapper.queryInterface(XPropertySet.class, link2Connector))
        .thenReturn(link2PropertySet);

    Mockito.when(unoRuntimeWrapper.queryInterface(XConnectorShape.class, link1Shape))
        .thenReturn(link1Connector);
    Mockito.when(unoRuntimeWrapper.queryInterface(XConnectorShape.class, link2Shape))
        .thenReturn(link2Connector);

    Mockito.when(unoRuntimeWrapper.queryInterface(XShape.class, serverPort)).thenReturn(serverPort);
    Mockito.when(unoRuntimeWrapper.queryInterface(XShape.class, method)).thenReturn(method);
    Mockito.when(unoRuntimeWrapper.queryInterface(XShape.class, linkTextShape))
        .thenReturn(linkTextShape);

    Mockito.when(unoRuntimeWrapper.queryInterface(XConnectorShape.class, link2Shape))
        .thenAnswer(
            new Answer<Object>() {
              @Override
              public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
                return link2Connector;
              }
            });

    Mockito.when(unoRuntimeWrapper.queryInterface(XConnectorShape.class, link1Shape))
        .thenAnswer(
            new Answer<Object>() {
              @Override
              public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
                return (link1Connector);
              }
            });

    Mockito.when(unoRuntimeWrapper.queryInterface(XConnectorShape.class, link1Connector))
        .thenReturn(link1Connector);
    Mockito.when(unoRuntimeWrapper.queryInterface(XConnectorShape.class, link2Connector))
        .thenReturn(link2Connector);

    Mockito.when(shapeHelperWrapper.isTextShape(linkTextShape)).thenReturn(true);
    Mockito.when(linkTextPropertySet.getPropertyValue("Text")).thenReturn("msg");

    Mockito.when(unoRuntimeWrapper.queryInterface(XPropertySet.class, linkTextShape))
        .thenReturn(linkTextPropertySet);

    // link end

    diagramWalker.setDiagramType(DiagramType.Process);

    Graph walk1 =
        diagramWalker.walk(
            Sets.newHashSet(method, link1Shape, link2Shape, linkTextShape, serverPort));
    List<ConnectedShapesComplex> walk =
        new ArrayList<>(diagramWalker.getConnectedShapesComplexes());

    Assert.assertEquals(diagramWalker.visited.size(), 5);
    Assert.assertEquals(walk.size(), 1);
    ConnectedShapesComplex connectedShapesComplex = walk.get(0);
    Assert.assertEquals(connectedShapesComplex.fromShape, serverPort);
    Assert.assertEquals(connectedShapesComplex.toShape, method);
    Assert.assertEquals(connectedShapesComplex.textShape, linkTextShape);
  }
コード例 #2
0
  @Test
  public void methodToMethodSuccessfullCase() throws Exception {

    final XShape method1 = Mockito.mock(XShape.class, "method1");
    Mockito.when(shapeHelperWrapper.getNodeType(method1)).thenReturn(Node.NodeType.MethodOfProcess);

    final XShape method2 = Mockito.mock(XShape.class, "method2");
    Mockito.when(shapeHelperWrapper.getNodeType(method2)).thenReturn(Node.NodeType.ServerPort);

    // link start
    XShape link1Shape = Mockito.mock(XShape.class, "link1Shape");
    XConnectorShape link1Connector = Mockito.mock(XConnectorShape.class, "link1ConnectorShape");
    XPropertySet link1PropertySet = Mockito.mock(XPropertySet.class, "link1PropertySet");

    XShape link2Shape = Mockito.mock(XShape.class, "link2Shape");
    XConnectorShape link2Connector = Mockito.mock(XConnectorShape.class, "link2ConnectorShape");
    XPropertySet link2PropertySet = Mockito.mock(XPropertySet.class, "link2PropertySet");

    XShape linkTextShape = Mockito.mock(XShape.class, "linkTextShape");
    XPropertySet linkTextPropertySet = Mockito.mock(XPropertySet.class);

    Mockito.when(shapeHelperWrapper.isConnectorShape(link1Shape)).thenReturn(true);
    Mockito.when(shapeHelperWrapper.isConnectorShape(link2Shape)).thenReturn(true);

    Mockito.when(link1PropertySet.getPropertyValue(ConnectedShapes.START_SHAPE))
        .thenAnswer(
            new Answer<Object>() {
              @Override
              public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
                return method1;
              }
            });

    Mockito.when(link1PropertySet.getPropertyValue(ConnectedShapes.END_SHAPE))
        .thenReturn(linkTextShape);

    Mockito.when(link2PropertySet.getPropertyValue(ConnectedShapes.START_SHAPE))
        .thenReturn(linkTextShape);

    Mockito.when(link2PropertySet.getPropertyValue(ConnectedShapes.END_SHAPE)).thenReturn(method2);

    Mockito.when(unoRuntimeWrapper.queryInterface(XPropertySet.class, link1Connector))
        .thenReturn(link1PropertySet);

    Mockito.when(unoRuntimeWrapper.queryInterface(XPropertySet.class, link2Connector))
        .thenReturn(link2PropertySet);

    Mockito.when(unoRuntimeWrapper.queryInterface(XConnectorShape.class, link1Shape))
        .thenReturn(link1Connector);
    Mockito.when(unoRuntimeWrapper.queryInterface(XConnectorShape.class, link2Shape))
        .thenReturn(link2Connector);

    Mockito.when(unoRuntimeWrapper.queryInterface(XShape.class, method2)).thenReturn(method2);
    Mockito.when(unoRuntimeWrapper.queryInterface(XShape.class, method1)).thenReturn(method1);
    Mockito.when(unoRuntimeWrapper.queryInterface(XShape.class, linkTextShape))
        .thenReturn(linkTextShape);

    Mockito.when(shapeHelperWrapper.isTextShape(linkTextShape)).thenReturn(true);
    Mockito.when(linkTextPropertySet.getPropertyValue("Text")).thenReturn("+");

    Mockito.when(unoRuntimeWrapper.queryInterface(XPropertySet.class, linkTextShape))
        .thenReturn(linkTextPropertySet);

    // link end

    diagramWalker.setDiagramType(DiagramType.Process);

    diagramWalker.walk(Sets.newHashSet(method1, link1Shape, link2Shape, linkTextShape, method2));

    Assert.assertEquals(diagramWalker.visited.size(), 5);
  }