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.mockito.ArgumentMatchers.anyString;
11 import static org.mockito.Mockito.when;
13 import java.util.concurrent.ExecutionException;
14 import org.junit.Assert;
15 import org.junit.Before;
16 import org.junit.Test;
17 import org.junit.runner.RunWith;
18 import org.mockito.Mock;
19 import org.mockito.junit.MockitoJUnitRunner;
20 import org.opendaylight.transportpce.common.StringConstants;
21 import org.opendaylight.transportpce.common.mapping.PortMapping;
22 import org.opendaylight.transportpce.common.network.NetworkTransactionImpl;
23 import org.opendaylight.transportpce.common.network.RequestProcessor;
24 import org.opendaylight.transportpce.pce.constraints.PceConstraintsCalc;
25 import org.opendaylight.transportpce.pce.utils.PceTestData;
26 import org.opendaylight.transportpce.pce.utils.PceTestUtils;
27 import org.opendaylight.transportpce.pce.utils.TransactionUtils;
28 import org.opendaylight.transportpce.test.AbstractTest;
29 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.pce.rev210701.PathComputationRequestInput;
30 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev210927.mapping.Mapping;
31 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev210927.mapping.MappingBuilder;
32 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev210927.network.Nodes;
33 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev210927.network.NodesBuilder;
34 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev210927.network.NodesKey;
35 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev210927.network.nodes.NodeInfo;
36 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev210927.network.nodes.NodeInfoBuilder;
37 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.types.rev191129.NodeTypes;
38 //import org.opendaylight.yang.gen.v1.http.org.openroadm.device.types.rev191129.NodeTypes;
40 @RunWith(MockitoJUnitRunner.StrictStubs.class)
41 public class PceCalculationTest extends AbstractTest {
43 private PceCalculation pceCalculation;
44 private PceConstraintsCalc pceConstraintsCalc;
45 private PceResult pceResult = new PceResult();
46 private Mapping mapping;
49 private PortMapping portMapping;
53 public 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(new RequestProcessor(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 when(portMapping.getMapping(anyString(), anyString())).thenReturn(mapping);
64 when(portMapping.getNode(anyString())).thenReturn(node);
68 public void testPceCalculationValues() {
69 pceCalculation = new PceCalculation(
70 PceTestData.getPCERequest(),
71 new NetworkTransactionImpl(new RequestProcessor(getDataBroker())),
72 pceConstraintsCalc.getPceHardConstraints(),
73 pceConstraintsCalc.getPceSoftConstraints(),
76 pceCalculation.retrievePceNetwork();
77 Assert.assertEquals(StringConstants.SERVICE_TYPE_100GE_T, pceCalculation.getServiceType());
78 Assert.assertNotNull(pceCalculation.getReturnStructure());
80 Assert.assertNull(pceCalculation.getaendPceNode());
81 Assert.assertNull(pceCalculation.getzendPceNode());
85 public void testPceCalculationValues2() {
86 pceCalculation = new PceCalculation(
87 PceTestData.getPathComputationRequestInputWithCoRoutingOrGeneral(),
88 new NetworkTransactionImpl(new RequestProcessor(getDataBroker())),
89 pceConstraintsCalc.getPceHardConstraints(),
90 pceConstraintsCalc.getPceSoftConstraints(),
91 pceResult, portMapping);
92 pceCalculation.retrievePceNetwork();
93 Assert.assertEquals(StringConstants.SERVICE_TYPE_100GE_T, pceCalculation.getServiceType());
94 Assert.assertNotNull(pceCalculation.getReturnStructure());
96 Assert.assertNull(pceCalculation.getaendPceNode());
97 Assert.assertNull(pceCalculation.getzendPceNode());
101 public void testPceCalculationValues42() {
102 PathComputationRequestInput input = PceTestData.getPathComputationRequestInputWithCoRoutingOrGeneral2();
103 pceConstraintsCalc = new PceConstraintsCalc(input,
104 new NetworkTransactionImpl(new RequestProcessor(getDataBroker())));
105 pceCalculation = new PceCalculation(
106 PceTestData.getPCE_test3_request_54(),
107 new NetworkTransactionImpl(new RequestProcessor(getDataBroker())),
108 pceConstraintsCalc.getPceHardConstraints(),
109 pceConstraintsCalc.getPceSoftConstraints(),
110 pceResult, portMapping);
111 pceCalculation.retrievePceNetwork();
112 Assert.assertEquals(StringConstants.SERVICE_TYPE_100GE_T, pceCalculation.getServiceType());
113 Assert.assertNotNull(pceCalculation.getReturnStructure());
115 Assert.assertNull(pceCalculation.getaendPceNode());
116 Assert.assertNull(pceCalculation.getzendPceNode());