コード例 #1
0
  private EntityTreeNode mockOperationComponent(
      final Long id,
      final String nodeNumber,
      final Collection<Entity> inputProducts,
      final Collection<Entity> outputProducts,
      final List<EntityTreeNode> subOperations) {
    EntityTreeNode operationComponent = mock(EntityTreeNode.class);

    given(operationComponent.getId()).willReturn(id);

    EntityList inputProductsList = mockProductComponentsList(inputProducts);
    given(operationComponent.getHasManyField(OPERATION_PRODUCT_IN_COMPONENTS))
        .willReturn(inputProductsList);
    given(operationComponent.getField(OPERATION_PRODUCT_IN_COMPONENTS))
        .willReturn(inputProductsList);

    EntityList outputProductsList = mockProductComponentsList(outputProducts);
    given(operationComponent.getHasManyField(OPERATION_PRODUCT_OUT_COMPONENTS))
        .willReturn(outputProductsList);
    given(operationComponent.getField(OPERATION_PRODUCT_OUT_COMPONENTS))
        .willReturn(outputProductsList);

    given(operationComponent.getField(NODE_NUMBER)).willReturn(nodeNumber);
    given(operationComponent.getStringField(NODE_NUMBER)).willReturn(nodeNumber);

    given(operationComponent.getChildren()).willReturn(subOperations);

    return operationComponent;
  }
コード例 #2
0
  @Test
  public final void shouldReturnNotEmptyMapIfManyParentOpConsumesManyOutputsFromOneSubOp() {
    // given
    Entity product1 = mockProductComponent(1L);
    Entity product2 = mockProductComponent(2L);
    Entity product3 = mockProductComponent(3L);
    Entity product4 = mockProductComponent(4L);
    Entity product5 = mockProductComponent(5L);
    Entity product6 = mockProductComponent(6L);
    Entity product7 = mockProductComponent(7L);
    Entity product8 = mockProductComponent(8L);
    Entity product9 = mockProductComponent(9L);
    Entity product10 = mockProductComponent(10L);
    Entity product11 = mockProductComponent(11L);
    Entity product12 = mockProductComponent(12L);

    EntityTreeNode node1A3 =
        mockOperationComponent(
            "1.A.3.", newArrayList(product11, product12), newArrayList(product10));
    EntityTreeNode node1A2 =
        mockOperationComponent(
            "1.A.2.",
            newArrayList(product10),
            newArrayList(product7, product8, product9),
            newArrayList(node1A3));
    EntityTreeNode node1A1 =
        mockOperationComponent(
            "1.A.1.",
            newArrayList(product7, product8, product9),
            newArrayList(product2, product3),
            newArrayList(node1A2));
    EntityTreeNode node1B1 =
        mockOperationComponent("1.B.1.", newArrayList(product6), newArrayList(product4, product5));
    EntityTreeNode node1 =
        mockOperationComponent(
            "1.",
            newArrayList(product2, product3, product4, product5),
            newArrayList(product1),
            newArrayList(node1A1, node1B1));

    given(tree.getRoot()).willReturn(node1);

    // when
    resultMap = technologyTreeValidationService.checkConsumingManyProductsFromOneSubOp(tree);

    // then
    assertNotNull(resultMap);
    assertFalse(resultMap.isEmpty());
    assertEquals(2, resultMap.size());
    assertEquals(2, resultMap.get(node1.getStringField(NODE_NUMBER)).size());
    hasNodeNumbersFor(node1, node1A1);
    hasNodeNumbersFor(node1, node1B1);
    hasNodeNumbersFor(node1A1, node1A2);
  }