2 * Copyright © 2017 AT&T, 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.utils;
10 import java.io.InputStream;
11 import java.util.List;
12 import java.util.Optional;
13 import java.util.concurrent.ExecutionException;
14 import java.util.stream.Collectors;
15 import org.junit.Assert;
16 import org.opendaylight.mdsal.binding.api.DataBroker;
17 import org.opendaylight.mdsal.binding.api.WriteTransaction;
18 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
19 import org.opendaylight.transportpce.common.DataStoreContext;
20 import org.opendaylight.transportpce.common.NetworkUtils;
21 import org.opendaylight.transportpce.common.converter.XMLDataObjectConverter;
22 import org.opendaylight.transportpce.common.converter.api.DataObjectConverter;
23 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.pce.rev200128.PathComputationRequestOutput;
24 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev171017.path.description.atoz.direction.AToZ;
25 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev171017.pce.resource.resource.resource.Node;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.Networks;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.Network;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.NetworkKey;
30 import org.opendaylight.yangtools.yang.binding.DataObject;
31 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
32 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
33 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
34 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
35 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
36 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
37 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
41 public final class PceTestUtils {
43 private static final Logger LOG = LoggerFactory.getLogger(PceTestUtils.class);
45 private PceTestUtils() {
48 public static void writeTopologyIntoDataStore(DataBroker dataBroker,
49 DataStoreContext dataStoreContext, String topologyDataPath)
50 throws ExecutionException, InterruptedException {
52 DataObjectConverter dataObjectConverter = XMLDataObjectConverter.createWithDataStoreUtil(dataStoreContext);
53 InputStream resourceAsStream = PceTestUtils.class.getClassLoader().getResourceAsStream(topologyDataPath);
54 Optional<NormalizedNode<? extends YangInstanceIdentifier.PathArgument, ?>> normalizedNode
55 = dataObjectConverter.transformIntoNormalizedNode(resourceAsStream);
56 DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> next
57 = ((ContainerNode) normalizedNode.get()).getValue().iterator().next();
58 MapEntryNode mapNode = ((MapNode) next).getValue().iterator().next();
59 Optional<DataObject> dataObject = dataObjectConverter.getDataObject(mapNode, Network.QNAME);
60 InstanceIdentifier<Network> nwInstanceIdentifier = InstanceIdentifier.builder(Networks.class)
61 .child(Network.class, new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
63 WriteTransaction dataWriteTransaction = dataBroker.newWriteOnlyTransaction();
64 dataWriteTransaction.put(LogicalDatastoreType.CONFIGURATION, nwInstanceIdentifier, (Network) dataObject.get());
65 dataWriteTransaction.commit().get();
68 public static void writeNetworkIntoDataStore(DataBroker dataBroker,
69 DataStoreContext dataStoreContext, Network network)
70 throws ExecutionException, InterruptedException {
72 InstanceIdentifier<Network> nwInstanceIdentifier = InstanceIdentifier.builder(Networks.class)
73 .child(Network.class, new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
75 WriteTransaction dataWriteTransaction = dataBroker.newWriteOnlyTransaction();
76 dataWriteTransaction.put(LogicalDatastoreType.CONFIGURATION, nwInstanceIdentifier, network);
77 dataWriteTransaction.commit().get();
80 public static void checkConfigurationResponse(PathComputationRequestOutput output,
81 PathComputationRequestOutput expectedOutput) {
82 Assert.assertNotNull(output);
84 expectedOutput.getConfigurationResponseCommon().getAckFinalIndicator(),
85 output.getConfigurationResponseCommon().getAckFinalIndicator());
87 expectedOutput.getConfigurationResponseCommon().getResponseMessage(),
88 output.getConfigurationResponseCommon().getResponseMessage());
90 expectedOutput.getConfigurationResponseCommon().getResponseCode(),
91 output.getConfigurationResponseCommon().getResponseCode());
94 public static void checkCalculatedPath(PathComputationRequestOutput output,
95 PathComputationRequestOutput expectedOutput) {
96 Assert.assertNotNull(output.getResponseParameters().getPathDescription());
97 Assert.assertNotNull(output.getResponseParameters().getPathDescription().getAToZDirection());
98 Assert.assertNotNull(output.getResponseParameters().getPathDescription().getZToADirection());
99 Assert.assertNotNull(output.getResponseParameters().getPathDescription().getAToZDirection().getAToZ());
100 Assert.assertNotNull(output.getResponseParameters().getPathDescription().getZToADirection().getZToA());
102 int atozSize = output.getResponseParameters().getPathDescription().getAToZDirection().getAToZ().size();
103 int ztoaSize = output.getResponseParameters().getPathDescription().getZToADirection().getZToA().size();
104 Assert.assertEquals(atozSize, ztoaSize);
106 Long actualAToZWavel = output.getResponseParameters().getPathDescription().getAToZDirection()
107 .getAToZWavelengthNumber().toJava();
108 Long expectedAToZWavel = expectedOutput.getResponseParameters().getPathDescription().getAToZDirection()
109 .getAToZWavelengthNumber().toJava();
110 Assert.assertEquals(actualAToZWavel, expectedAToZWavel);
112 Long actualZtoAWavel = output.getResponseParameters().getPathDescription().getZToADirection()
113 .getZToAWavelengthNumber().toJava();
114 Long expectedZtoAWavel = expectedOutput.getResponseParameters().getPathDescription().getZToADirection()
115 .getZToAWavelengthNumber().toJava();
116 Assert.assertEquals(actualZtoAWavel, expectedZtoAWavel);
118 Long actualAToZRate = output.getResponseParameters().getPathDescription().getAToZDirection().getRate().toJava();
119 Long expectedAToZRate = expectedOutput.getResponseParameters().getPathDescription().getAToZDirection()
121 Assert.assertEquals(expectedAToZRate, actualAToZRate);
123 Long actualZToARate = output.getResponseParameters().getPathDescription().getZToADirection().getRate().toJava();
124 Long expectedZToARate = expectedOutput.getResponseParameters().getPathDescription().getZToADirection()
126 Assert.assertEquals(actualZToARate, expectedZToARate);
129 private static List<String> getNodesFromPath(PathComputationRequestOutput output) {
130 List<AToZ> atozList = output.getResponseParameters().getPathDescription().getAToZDirection().getAToZ();
131 return atozList.stream()
133 if ((aToZ.getResource() == null) || (aToZ.getResource().getResource() == null)) {
134 LOG.debug("Diversity constraint: Resource of AToZ node {} is null! Skipping this node!",
138 return aToZ.getResource().getResource() instanceof Node;
140 Node node = (Node) aToZ.getResource().getResource();
141 if (node.getNodeId() == null) {
142 LOG.warn("Node in AToZ node {} contains null! Skipping this node!", aToZ.getId());
145 return node.getNodeId().toString();
146 }).collect(Collectors.toList());
149 public static boolean comparePath(PathComputationRequestOutput output1, PathComputationRequestOutput output2) {
150 // return true if paths are different
151 List<String> nodes1 = getNodesFromPath(output1);
152 LOG.info("nodes1: {}", nodes1.toString());
153 List<String> nodes2 = getNodesFromPath(output2);
154 LOG.info("nodes2: {}", nodes2.toString());
155 nodes1.retainAll(nodes2);
156 LOG.info("nodes after intersection: {}", nodes1.toString());
157 return nodes1.isEmpty();
160 public static boolean checkPCECalculationConflicts(PathComputationRequestOutput[] outputs,
161 int iterationOrdinal, DataBroker dataBroker) {