2 * Copyright © 2020 Orange Labs, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.transportpce.pce.networkanalyzer;
10 import static org.junit.jupiter.api.Assertions.assertEquals;
11 import static org.junit.jupiter.api.Assertions.assertNotNull;
12 import static org.junit.jupiter.api.Assertions.assertNull;
13 import static org.mockito.ArgumentMatchers.anyString;
14 import static org.mockito.Mockito.lenient;
15 import static org.mockito.Mockito.when;
17 import java.util.concurrent.ExecutionException;
18 import org.junit.jupiter.api.BeforeEach;
19 import org.junit.jupiter.api.Test;
20 import org.junit.jupiter.api.extension.ExtendWith;
21 import org.mockito.Mock;
22 import org.mockito.junit.jupiter.MockitoExtension;
23 import org.opendaylight.transportpce.common.StringConstants;
24 import org.opendaylight.transportpce.common.mapping.PortMapping;
25 import org.opendaylight.transportpce.common.network.NetworkTransactionImpl;
26 import org.opendaylight.transportpce.pce.constraints.PceConstraintsCalc;
27 import org.opendaylight.transportpce.pce.utils.PceTestData;
28 import org.opendaylight.transportpce.pce.utils.PceTestUtils;
29 import org.opendaylight.transportpce.pce.utils.TransactionUtils;
30 import org.opendaylight.transportpce.test.AbstractTest;
31 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.pce.rev240205.PathComputationRequestInput;
32 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev231221.mapping.Mapping;
33 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev231221.mapping.MappingBuilder;
34 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev231221.network.Nodes;
35 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev231221.network.NodesBuilder;
36 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev231221.network.NodesKey;
37 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev231221.network.nodes.NodeInfo;
38 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev231221.network.nodes.NodeInfoBuilder;
39 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.types.rev191129.NodeTypes;
41 @ExtendWith(MockitoExtension.class)
42 public class PceCalculationTest extends AbstractTest {
44 private PceCalculation pceCalculation;
45 private PceConstraintsCalc pceConstraintsCalc;
46 private PceResult pceResult = new PceResult();
47 private Mapping mapping;
50 private PortMapping portMapping;
53 void setUp() throws ExecutionException, InterruptedException {
54 pceResult.setRC("200");
55 PceTestUtils.writeNetworkIntoDataStore(getDataBroker(), getDataStoreContextUtil(),
56 TransactionUtils.getNetworkForSpanLoss());
57 pceConstraintsCalc = new PceConstraintsCalc(PceTestData.getPCERequest(),
58 new NetworkTransactionImpl(getDataBroker()));
59 mapping = new MappingBuilder().setLogicalConnectionPoint("logicalConnectionPoint").setPortQual("xpdr-client")
61 NodeInfo info = new NodeInfoBuilder().setNodeType(NodeTypes.Xpdr).build();
62 Nodes node = new NodesBuilder().withKey(new NodesKey("node")).setNodeId("node").setNodeInfo(info).build();
63 lenient().when(portMapping.getMapping(anyString(), anyString())).thenReturn(mapping);
64 when(portMapping.getNode(anyString())).thenReturn(node);
68 void testPceCalculationValues() {
69 pceCalculation = new PceCalculation(
70 PceTestData.getPCERequest(),
71 new NetworkTransactionImpl(getDataBroker()),
72 pceConstraintsCalc.getPceHardConstraints(),
73 pceConstraintsCalc.getPceSoftConstraints(),
76 pceCalculation.retrievePceNetwork();
77 assertEquals(StringConstants.SERVICE_TYPE_100GE_T, pceCalculation.getServiceType());
78 assertNotNull(pceCalculation.getReturnStructure());
79 assertNull(pceCalculation.getaendPceNode());
80 assertNull(pceCalculation.getzendPceNode());
84 void testPceCalculationValues2() {
85 pceCalculation = new PceCalculation(
86 PceTestData.getPathComputationRequestInputWithCoRoutingOrGeneral(),
87 new NetworkTransactionImpl(getDataBroker()),
88 pceConstraintsCalc.getPceHardConstraints(),
89 pceConstraintsCalc.getPceSoftConstraints(),
90 pceResult, portMapping);
91 pceCalculation.retrievePceNetwork();
92 assertEquals(StringConstants.SERVICE_TYPE_100GE_T, pceCalculation.getServiceType());
93 assertNotNull(pceCalculation.getReturnStructure());
94 assertNull(pceCalculation.getaendPceNode());
95 assertNull(pceCalculation.getzendPceNode());
99 void testPceCalculationValues42() {
100 PathComputationRequestInput input = PceTestData.getPathComputationRequestInputWithCoRoutingOrGeneral2();
101 pceConstraintsCalc = new PceConstraintsCalc(input,
102 new NetworkTransactionImpl(getDataBroker()));
103 pceCalculation = new PceCalculation(
104 PceTestData.getPCE_test3_request_54(),
105 new NetworkTransactionImpl(getDataBroker()),
106 pceConstraintsCalc.getPceHardConstraints(),
107 pceConstraintsCalc.getPceSoftConstraints(),
108 pceResult, portMapping);
109 pceCalculation.retrievePceNetwork();
110 assertEquals(StringConstants.SERVICE_TYPE_100GE_T, pceCalculation.getServiceType());
111 assertNotNull(pceCalculation.getReturnStructure());
112 assertNull(pceCalculation.getaendPceNode());
113 assertNull(pceCalculation.getzendPceNode());