networkmodel/src/test/java/org/opendaylight/transportpce/networkmodel/service/NetworkModelWavelengthServiceTest.java

   1 /*
   2  * Copyright © 2020 Orange, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8
   9 package org.opendaylight.transportpce.networkmodel.service;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.assertNull;
  13 import static org.junit.Assert.fail;
  14
  15 import java.io.FileNotFoundException;
  16 import java.io.FileReader;
  17 import java.io.IOException;
  18 import java.io.Reader;
  19 import java.nio.charset.StandardCharsets;
  20 import java.util.BitSet;
  21 import java.util.Optional;
  22 import java.util.concurrent.ExecutionException;
  23 import java.util.concurrent.TimeUnit;
  24 import java.util.concurrent.TimeoutException;
  25 import org.junit.BeforeClass;
  26 import org.junit.Test;
  27 import org.opendaylight.mdsal.binding.api.ReadTransaction;
  28 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  29 import org.opendaylight.transportpce.common.InstanceIdentifiers;
  30 import org.opendaylight.transportpce.common.NetworkUtils;
  31 import org.opendaylight.transportpce.common.Timeouts;
  32 import org.opendaylight.transportpce.common.fixedflex.GridConstant;
  33 import org.opendaylight.transportpce.networkmodel.util.OpenRoadmTopology;
  34 import org.opendaylight.transportpce.test.AbstractTest;
  35 import org.opendaylight.transportpce.test.converter.DataObjectConverter;
  36 import org.opendaylight.transportpce.test.converter.JSONDataObjectConverter;
  37 import org.opendaylight.transportpce.test.utils.TopologyDataUtils;
  38 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.Node1;
  39 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.TerminationPoint1;
  40 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev200529.available.freq.map.AvailFreqMapsKey;
  41 import org.opendaylight.yang.gen.v1.pathdescription.stub.rev201211.PathDescription;
  42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  43 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  44 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  45 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  46 import org.slf4j.Logger;
  47 import org.slf4j.LoggerFactory;
  48
  49 public class NetworkModelWavelengthServiceTest extends AbstractTest {
  50     private static final Logger LOG = LoggerFactory.getLogger(NetworkModelWavelengthServiceTest.class);
  51     private static final String OPENROADM_TOPOLOGY_FILE = "src/test/resources/openroadm-topology.xml";
  52     private static final String PATH_DESCRIPTION_FILE = "src/test/resources/path_description.json";
  53     private static PathDescription pathDescription;
  54     private final AvailFreqMapsKey availFreqMapKey = new AvailFreqMapsKey(GridConstant.C_BAND);
  55     private final BitSet usedBits = new BitSet(8);
  56     private static BitSet availableBits = new BitSet(8);
  57
  58     @BeforeClass
  59     public static void setUp() throws InterruptedException, ExecutionException, FileNotFoundException {
  60         availableBits.set(0, 8, true);
  61         TopologyDataUtils.writeTopologyFromFileToDatastore(getDataStoreContextUtil(), OPENROADM_TOPOLOGY_FILE,
  62                 InstanceIdentifiers.OVERLAY_NETWORK_II);
  63         DataObjectConverter dataObjectConverter = JSONDataObjectConverter
  64                 .createWithDataStoreUtil(getDataStoreContextUtil());
  65         try (Reader reader = new FileReader(PATH_DESCRIPTION_FILE, StandardCharsets.UTF_8)) {
  66             NormalizedNode<? extends PathArgument, ?> normalizedNode = dataObjectConverter
  67                     .transformIntoNormalizedNode(reader).get();
  68             pathDescription = (PathDescription) getDataStoreContextUtil()
  69                     .getBindingDOMCodecServices().fromNormalizedNode(YangInstanceIdentifier
  70                             .of(PathDescription.QNAME), normalizedNode).getValue();
  71         } catch (IOException e) {
  72             LOG.error("Cannot load path description ", e);
  73             fail("Cannot load path description ");
  74         }
  75     }
  76
  77     @Test
  78     public void allocateFrequenciesTest() throws IOException {
  79         NetworkModelWavelengthService service = new NetworkModelWavelengthServiceImpl(getDataBroker());
  80         service.allocateFrequencies(pathDescription.getAToZDirection(), pathDescription.getZToADirection());
  81         TerminationPoint1 terminationPoint = getNetworkTerminationPointFromDatastore("ROADM-A1-DEG2", "DEG2-CTP-TXRX");
  82         assertEquals("Lambda 1 should not be available for ctp-attributes",
  83                 BitSet.valueOf(terminationPoint.getCtpAttributes().getAvailFreqMaps().get(availFreqMapKey)
  84                 .getFreqMap()).get(760, 768),usedBits);
  85         assertNull("cp-attributes should be null", terminationPoint.getCpAttributes());
  86         terminationPoint = getNetworkTerminationPointFromDatastore("ROADM-A1-SRG1", "SRG1-PP1-TXRX");
  87         assertEquals("Lambda 1 should not be available for pp-attributes",
  88                 BitSet.valueOf(terminationPoint.getPpAttributes().getAvailFreqMaps().get(availFreqMapKey)
  89                 .getFreqMap()).get(760, 768),usedBits);
  90         Node1 node = getNetworkNodeFromDatastore("ROADM-A1-SRG1");
  91         assertEquals("Lambda 1 should not be available for srg-attributes",
  92                 BitSet.valueOf(node.getSrgAttributes().getAvailFreqMaps().get(availFreqMapKey)
  93                 .getFreqMap()).get(760, 768),usedBits);
  94     }
  95
  96     @Test
  97     public void releaseFrequenciesTest() throws IOException {
  98         NetworkModelWavelengthService service = new NetworkModelWavelengthServiceImpl(getDataBroker());
  99         service.allocateFrequencies(pathDescription.getAToZDirection(), pathDescription.getZToADirection());
 100         service.releaseFrequencies(pathDescription.getAToZDirection(), pathDescription.getZToADirection());
 101         TerminationPoint1 terminationPoint = getNetworkTerminationPointFromDatastore("ROADM-A1-DEG2", "DEG2-CTP-TXRX");
 102         assertEquals("Lambda 1 should be available for ctp-attributes",
 103                 BitSet.valueOf(terminationPoint.getCtpAttributes().getAvailFreqMaps().get(availFreqMapKey)
 104                 .getFreqMap()).get(760, 768),availableBits);
 105         terminationPoint = getNetworkTerminationPointFromDatastore("ROADM-A1-SRG1", "SRG1-PP1-TXRX");
 106         assertEquals("Lambda 1 should be available for pp-attributes",
 107                 BitSet.valueOf(terminationPoint.getPpAttributes().getAvailFreqMaps().get(availFreqMapKey)
 108                 .getFreqMap()).get(760, 768),availableBits);
 109         Node1 node = getNetworkNodeFromDatastore("ROADM-A1-SRG1");
 110         assertEquals("Lambda 1 should be available for srg-attributes",
 111                 BitSet.valueOf(node.getSrgAttributes().getAvailFreqMaps().get(availFreqMapKey)
 112                 .getFreqMap()).get(760, 768),availableBits);
 113     }
 114
 115     private TerminationPoint1 getNetworkTerminationPointFromDatastore(String nodeId, String tpId) {
 116         InstanceIdentifier<TerminationPoint1> tpIID = OpenRoadmTopology
 117                 .createNetworkTerminationPointIIDBuilder(nodeId, tpId).build();
 118         try (ReadTransaction readTx = getDataBroker().newReadOnlyTransaction()) {
 119             Optional<TerminationPoint1> optionalTerminationPoint = readTx
 120                     .read(LogicalDatastoreType.CONFIGURATION, tpIID)
 121                     .get(Timeouts.DATASTORE_READ, TimeUnit.MILLISECONDS);
 122             if (optionalTerminationPoint.isPresent()) {
 123                 return optionalTerminationPoint.get();
 124             } else {
 125                 return null;
 126             }
 127         } catch (ExecutionException | TimeoutException e) {
 128             LOG.warn("Exception while getting termination {} for node id {} point from {} topology", tpId, nodeId,
 129                     NetworkUtils.OVERLAY_NETWORK_ID, e);
 130             return null;
 131         } catch (InterruptedException e) {
 132             LOG.warn("Getting termination {} for node id {} point from {} topology was interrupted", tpId, nodeId,
 133                     NetworkUtils.OVERLAY_NETWORK_ID, e);
 134             Thread.currentThread().interrupt();
 135             return null;
 136         }
 137     }
 138
 139     private Node1 getNetworkNodeFromDatastore(String nodeId) {
 140         InstanceIdentifier<Node1> nodeIID = OpenRoadmTopology.createNetworkNodeIID(nodeId);
 141         try (ReadTransaction nodeReadTx = getDataBroker().newReadOnlyTransaction()) {
 142             Optional<Node1> optionalNode = nodeReadTx.read(LogicalDatastoreType.CONFIGURATION, nodeIID)
 143                     .get(Timeouts.DATASTORE_READ, TimeUnit.MILLISECONDS);
 144             if (optionalNode.isPresent()) {
 145                 return optionalNode.get();
 146             } else {
 147                 LOG.error("Unable to get network node for node id {}from topology {}", nodeId,
 148                         NetworkUtils.OVERLAY_NETWORK_ID);
 149                 return null;
 150             }
 151         } catch (ExecutionException | TimeoutException e) {
 152             LOG.warn("Exception while getting network node for node id {} from {} topology", nodeId,
 153                     NetworkUtils.OVERLAY_NETWORK_ID, e);
 154             return null;
 155         } catch (InterruptedException e) {
 156             LOG.warn("Getting network node for node id {} from {} topology was interrupted", nodeId,
 157                     NetworkUtils.OVERLAY_NETWORK_ID, e);
 158             Thread.currentThread().interrupt();
 159             return null;
 160         }
 161     }
 162
 163 }