Upgrade openroadm network models to 10.1
[transportpce.git] / networkmodel / src / test / java / org / opendaylight / transportpce / networkmodel / service / FrequenciesServiceTest.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.rev211210.Node1;
39 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev211210.TerminationPoint1;
40 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev211210.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.schema.NormalizedNode;
45 import org.slf4j.Logger;
46 import org.slf4j.LoggerFactory;
47
48 public class FrequenciesServiceTest extends AbstractTest {
49     private static final Logger LOG = LoggerFactory.getLogger(FrequenciesServiceTest.class);
50     private static final String OPENROADM_TOPOLOGY_FILE = "src/test/resources/openroadm-topology.xml";
51     private static final String PATH_DESCRIPTION_FILE = "src/test/resources/path_description.json";
52     private static PathDescription pathDescription;
53     private final AvailFreqMapsKey availFreqMapKey = new AvailFreqMapsKey(GridConstant.C_BAND);
54     private final BitSet usedBits = new BitSet(8);
55     private static BitSet availableBits = new BitSet(8);
56
57     @BeforeClass
58     public static void setUp() throws InterruptedException, ExecutionException, FileNotFoundException {
59         availableBits.set(0, 8, true);
60         TopologyDataUtils.writeTopologyFromFileToDatastore(getDataStoreContextUtil(), OPENROADM_TOPOLOGY_FILE,
61                 InstanceIdentifiers.OVERLAY_NETWORK_II);
62         DataObjectConverter dataObjectConverter = JSONDataObjectConverter
63                 .createWithDataStoreUtil(getDataStoreContextUtil());
64         try (Reader reader = new FileReader(PATH_DESCRIPTION_FILE, StandardCharsets.UTF_8)) {
65             NormalizedNode normalizedNode = dataObjectConverter
66                     .transformIntoNormalizedNode(reader).get();
67             pathDescription = (PathDescription) getDataStoreContextUtil()
68                     .getBindingDOMCodecServices().fromNormalizedNode(YangInstanceIdentifier
69                             .of(PathDescription.QNAME), normalizedNode).getValue();
70         } catch (IOException e) {
71             LOG.error("Cannot load path description ", e);
72             fail("Cannot load path description ");
73         }
74     }
75
76     @Test
77     public void allocateFrequenciesTest() throws IOException {
78         FrequenciesService service = new FrequenciesServiceImpl(getDataBroker());
79         service.allocateFrequencies(pathDescription.getAToZDirection(), pathDescription.getZToADirection());
80         TerminationPoint1 terminationPoint = getNetworkTerminationPointFromDatastore("ROADM-A1-DEG2", "DEG2-CTP-TXRX");
81         assertEquals("Lambda 1 should not be available for ctp-attributes",
82                 BitSet.valueOf(terminationPoint.getCtpAttributes().getAvailFreqMaps().get(availFreqMapKey)
83                 .getFreqMap()).get(760, 768),usedBits);
84         assertNull("cp-attributes should be null", terminationPoint.getCpAttributes());
85         terminationPoint = getNetworkTerminationPointFromDatastore("ROADM-A1-SRG1", "SRG1-PP1-TXRX");
86         assertEquals("Lambda 1 should not be available for pp-attributes",
87                 BitSet.valueOf(terminationPoint.getPpAttributes().getAvailFreqMaps().get(availFreqMapKey)
88                 .getFreqMap()).get(760, 768),usedBits);
89         Node1 node = getNetworkNodeFromDatastore("ROADM-A1-SRG1");
90         assertEquals("Lambda 1 should not be available for srg-attributes",
91                 BitSet.valueOf(node.getSrgAttributes().getAvailFreqMaps().get(availFreqMapKey)
92                 .getFreqMap()).get(760, 768),usedBits);
93     }
94
95     @Test
96     public void releaseFrequenciesTest() throws IOException {
97         FrequenciesService service = new FrequenciesServiceImpl(getDataBroker());
98         service.allocateFrequencies(pathDescription.getAToZDirection(), pathDescription.getZToADirection());
99         service.releaseFrequencies(pathDescription.getAToZDirection(), pathDescription.getZToADirection());
100         TerminationPoint1 terminationPoint = getNetworkTerminationPointFromDatastore("ROADM-A1-DEG2", "DEG2-CTP-TXRX");
101         assertEquals("Lambda 1 should be available for ctp-attributes",
102                 BitSet.valueOf(terminationPoint.getCtpAttributes().getAvailFreqMaps().get(availFreqMapKey)
103                 .getFreqMap()).get(760, 768),availableBits);
104         terminationPoint = getNetworkTerminationPointFromDatastore("ROADM-A1-SRG1", "SRG1-PP1-TXRX");
105         assertEquals("Lambda 1 should be available for pp-attributes",
106                 BitSet.valueOf(terminationPoint.getPpAttributes().getAvailFreqMaps().get(availFreqMapKey)
107                 .getFreqMap()).get(760, 768),availableBits);
108         Node1 node = getNetworkNodeFromDatastore("ROADM-A1-SRG1");
109         assertEquals("Lambda 1 should be available for srg-attributes",
110                 BitSet.valueOf(node.getSrgAttributes().getAvailFreqMaps().get(availFreqMapKey)
111                 .getFreqMap()).get(760, 768),availableBits);
112     }
113
114     private TerminationPoint1 getNetworkTerminationPointFromDatastore(String nodeId, String tpId) {
115         InstanceIdentifier<TerminationPoint1> tpIID = OpenRoadmTopology
116                 .createNetworkTerminationPointIIDBuilder(nodeId, tpId).build();
117         try (ReadTransaction readTx = getDataBroker().newReadOnlyTransaction()) {
118             Optional<TerminationPoint1> optionalTerminationPoint = readTx
119                     .read(LogicalDatastoreType.CONFIGURATION, tpIID)
120                     .get(Timeouts.DATASTORE_READ, TimeUnit.MILLISECONDS);
121             if (optionalTerminationPoint.isPresent()) {
122                 return optionalTerminationPoint.get();
123             } else {
124                 return null;
125             }
126         } catch (ExecutionException | TimeoutException e) {
127             LOG.warn("Exception while getting termination {} for node id {} point from {} topology", tpId, nodeId,
128                     NetworkUtils.OVERLAY_NETWORK_ID, e);
129             return null;
130         } catch (InterruptedException e) {
131             LOG.warn("Getting termination {} for node id {} point from {} topology was interrupted", tpId, nodeId,
132                     NetworkUtils.OVERLAY_NETWORK_ID, e);
133             Thread.currentThread().interrupt();
134             return null;
135         }
136     }
137
138     private Node1 getNetworkNodeFromDatastore(String nodeId) {
139         InstanceIdentifier<Node1> nodeIID = OpenRoadmTopology.createNetworkNodeIID(nodeId);
140         try (ReadTransaction nodeReadTx = getDataBroker().newReadOnlyTransaction()) {
141             Optional<Node1> optionalNode = nodeReadTx.read(LogicalDatastoreType.CONFIGURATION, nodeIID)
142                     .get(Timeouts.DATASTORE_READ, TimeUnit.MILLISECONDS);
143             if (optionalNode.isPresent()) {
144                 return optionalNode.get();
145             } else {
146                 LOG.error("Unable to get network node for node id {}from topology {}", nodeId,
147                         NetworkUtils.OVERLAY_NETWORK_ID);
148                 return null;
149             }
150         } catch (ExecutionException | TimeoutException e) {
151             LOG.warn("Exception while getting network node for node id {} from {} topology", nodeId,
152                     NetworkUtils.OVERLAY_NETWORK_ID, e);
153             return null;
154         } catch (InterruptedException e) {
155             LOG.warn("Getting network node for node id {} from {} topology was interrupted", nodeId,
156                     NetworkUtils.OVERLAY_NETWORK_ID, e);
157             Thread.currentThread().interrupt();
158             return null;
159         }
160     }
161
162 }