Clean JUnit tests
[transportpce.git] / pce / src / test / java / org / opendaylight / transportpce / pce / utils / TransactionUtils.java
1 /*
2  * Copyright © 2020 Orange Systems, 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
10 package org.opendaylight.transportpce.pce.utils;
11
12 import java.math.BigDecimal;
13 import java.util.HashMap;
14 import java.util.Map;
15 import java.util.Optional;
16 import org.eclipse.jdt.annotation.NonNull;
17 import org.opendaylight.mdsal.binding.api.DataBroker;
18 import org.opendaylight.mdsal.binding.api.WriteTransaction;
19 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
20 import org.opendaylight.transportpce.common.fixedflex.GridUtils;
21 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev220114.mapping.Mapping;
22 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev220114.mapping.MappingBuilder;
23 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.link.types.rev191129.FiberPmd;
24 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.link.types.rev191129.RatioDB;
25 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.Node1Builder;
26 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.interfaces.grp.Interface;
27 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.interfaces.grp.InterfaceKey;
28 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.org.openroadm.device.container.OrgOpenroadmDevice;
29 import org.opendaylight.yang.gen.v1.http.org.openroadm.equipment.states.types.rev191129.AdminStates;
30 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.amplified.link.attributes.AmplifiedLink;
31 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.amplified.link.attributes.AmplifiedLinkKey;
32 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.amplified.link.attributes.amplified.link.SectionElementBuilder;
33 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.span.attributes.LinkConcatenation;
34 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.span.attributes.LinkConcatenationBuilder;
35 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.span.attributes.LinkConcatenationKey;
36 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.Link1Builder;
37 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.networks.network.link.OMSAttributesBuilder;
38 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.networks.network.link.oms.attributes.AmplifiedLinkBuilder;
39 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.networks.network.link.oms.attributes.SpanBuilder;
40 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.networks.network.node.DegreeAttributes;
41 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.networks.network.node.DegreeAttributesBuilder;
42 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.networks.network.node.SrgAttributes;
43 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.networks.network.node.SrgAttributesBuilder;
44 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev200529.OpenroadmLinkType;
45 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev200529.OpenroadmNodeType;
46 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.CurrentPmList;
47 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.CurrentPmListBuilder;
48 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.current.pm.group.CurrentPm;
49 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.current.pm.group.CurrentPmBuilder;
50 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.current.pm.list.CurrentPmEntry;
51 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.current.pm.list.CurrentPmEntryBuilder;
52 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.current.pm.val.group.Measurement;
53 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.current.pm.val.group.MeasurementBuilder;
54 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.current.pm.val.group.MeasurementKey;
55 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.types.rev171215.PmDataType;
56 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.types.rev171215.Validity;
57 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
58 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NodeId;
59 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.NetworkBuilder;
60 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.Node;
61 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.NodeBuilder;
62 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.NodeKey;
63 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.node.SupportingNode;
64 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.node.SupportingNodeBuilder;
65 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.node.SupportingNodeKey;
66 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.LinkId;
67 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Network1;
68 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Network1Builder;
69 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.TpId;
70 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.Link;
71 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.LinkBuilder;
72 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.link.DestinationBuilder;
73 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.link.SourceBuilder;
74 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.DateAndTime;
75 import org.opendaylight.yangtools.yang.binding.Augmentation;
76 import org.opendaylight.yangtools.yang.binding.DataObject;
77 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
78 import org.opendaylight.yangtools.yang.common.Uint16;
79 import org.opendaylight.yangtools.yang.common.Uint32;
80
81 public final class TransactionUtils {
82
83     private TransactionUtils() {
84     }
85
86     @SuppressWarnings({"unchecked","rawtypes"})
87     // FIXME check if the InstanceIdentifier raw type can be avoided
88     // Raw types use are discouraged since they lack type safety.
89     // Resulting Problems are observed at run time and not at compile time
90     public static void writeTransaction(DataBroker dataBroker, InstanceIdentifier instanceIdentifier,
91                                         DataObject object) {
92         @NonNull
93         WriteTransaction transaction = dataBroker.newWriteOnlyTransaction();
94         transaction.put(LogicalDatastoreType.CONFIGURATION, instanceIdentifier, object);
95         transaction.commit();
96     }
97
98     public static org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks
99             .Network getNetworkForSpanLoss() {
100
101         Map<LinkConcatenationKey,LinkConcatenation> linkConcentationValues = new HashMap<>();
102         LinkConcatenation linkConcatenation = new LinkConcatenationBuilder()
103                 .setFiberType(LinkConcatenation.FiberType.Truewave)
104                 .setPmd(new FiberPmd(BigDecimal.ONE))
105                 .setSRLGId(Uint32.valueOf(1))
106                 .setSRLGLength(Uint32.valueOf(1))
107                 .build();
108         LinkConcatenation linkConcatenation2 = new LinkConcatenationBuilder()
109                 .setFiberType(LinkConcatenation.FiberType.Truewave)
110                 .setPmd(new FiberPmd(BigDecimal.ONE))
111                 .setSRLGId(Uint32.valueOf(1))
112                 .setSRLGLength(Uint32.valueOf(1))
113                 .build();
114         linkConcentationValues.put(linkConcatenation.key(),linkConcatenation);
115         linkConcentationValues.put(linkConcatenation2.key(),linkConcatenation2);
116         // create 2 openroadm-topology degree nodes, end points of the link to be
117         // measured
118         SupportingNode supportingNodeA = new SupportingNodeBuilder().setNetworkRef(new NetworkId("openroadm-network"))
119                 .setNodeRef(new NodeId("ROADM-A1")).build();
120         Node ietfNodeA = new NodeBuilder().setNodeId(new NodeId("ROADM-A1-DEG2"))
121                 .setSupportingNode(Map.of(supportingNodeA.key(),supportingNodeA))
122                 .addAugmentation(getNode1AugImpl()).build();
123         Map<SupportingNodeKey,SupportingNode> supportingNodeListC = new HashMap<>();
124         SupportingNode supportingNode = new SupportingNodeBuilder().setNetworkRef(new NetworkId("openroadm-network"))
125                 .setNodeRef(new NodeId("ROADM-C1")).build();
126         supportingNodeListC.put(supportingNode.key(),supportingNode);
127         SupportingNode supportingNode2 = new SupportingNodeBuilder().setNetworkRef(new NetworkId("clli-network"))
128                 .setNodeRef(new NodeId("ROADM-C2")).build();
129         supportingNodeListC.put(supportingNode2.key(),supportingNode2);
130
131         Node ietfNodeC = new NodeBuilder().setNodeId(new NodeId("ROADM-C1-DEG1")).setSupportingNode(supportingNodeListC)
132                 .addAugmentation(getNode1AugImpl())
133                 .addAugmentation(getNode1())
134                 .build();
135         Map<NodeKey,Node> ietfNodeList = new HashMap<>();
136         ietfNodeList.put(ietfNodeA.key(),ietfNodeA);
137         ietfNodeList.put(ietfNodeC.key(),ietfNodeC);
138
139         Map<AmplifiedLinkKey,AmplifiedLink>
140                 amplifiedLinkValues = new HashMap<>();
141         org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.amplified.link.attributes.AmplifiedLink al =
142                 new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.amplified.link.attributes
143                         .AmplifiedLinkBuilder().setSectionElement(new SectionElementBuilder()
144                         .setSectionElement(new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529
145                                 .amplified.link.attributes.amplified.link.section.element.section.element
146                                 .SpanBuilder()
147                                 .setSpan(
148                                         new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.amplified
149                                                 .link.attributes.amplified.link.section.element.section
150                                                 .element.span.SpanBuilder()
151                                                 .setAdministrativeState(AdminStates.InService)
152                                                 .setAutoSpanloss(true)
153                                                 .setEngineeredSpanloss(new RatioDB(BigDecimal.ONE))
154                                                 .setLinkConcatenation(linkConcentationValues)
155                                                 .setSpanlossBase(new RatioDB(BigDecimal.ONE))
156                                                 .setSpanlossCurrent(new RatioDB(BigDecimal.ONE))
157                                                 .build())
158                                 .build())
159                         .build())
160                         .setSectionEltNumber(Uint16.valueOf(1)).build();
161         org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.amplified.link.attributes.AmplifiedLink al2 =
162                 new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.amplified.link.attributes
163                         .AmplifiedLinkBuilder().setSectionElement(new SectionElementBuilder()
164                         .setSectionElement(
165                                 new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529.amplified.link
166                                         .attributes.amplified.link.section.element.section.element.SpanBuilder()
167                                         .setSpan(
168                                                 new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev200529
169                                                         .amplified.link
170                                                         .attributes.amplified.link.section.element.section.element.span
171                                                         .SpanBuilder()
172                                                         .setAdministrativeState(AdminStates.InService)
173                                                         .setAutoSpanloss(true)
174                                                         .setEngineeredSpanloss(new RatioDB(BigDecimal.ONE))
175                                                         .setLinkConcatenation(linkConcentationValues)
176                                                         .setSpanlossBase(new RatioDB(BigDecimal.ONE))
177                                                         .setSpanlossCurrent(new RatioDB(BigDecimal.ONE))
178                                                         .build())
179                                         .build())
180                         .build())
181                         .setSectionEltNumber(Uint16.valueOf(1)).build();
182         amplifiedLinkValues.put(al.key(),al);
183         amplifiedLinkValues.put(al2.key(),al2);
184         Augmentation<Link> aug11 = new Link1Builder()
185                 .setAmplified(true)
186                 .setOMSAttributes(new OMSAttributesBuilder()
187                         .setAmplifiedLink(new AmplifiedLinkBuilder().setAmplifiedLink(amplifiedLinkValues).build())
188                         .setSpan(new SpanBuilder().build())
189                         .build())
190                 .build();
191         org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.Link1 aug12 =
192                 new org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.Link1Builder()
193                         .setOppositeLink(new LinkId("link 1"))
194                         .setLinkType(OpenroadmLinkType.ROADMTOROADM).build();
195         Augmentation<Link> aug21 = new Link1Builder()
196                 .setAmplified(true)
197                 .setOMSAttributes(new OMSAttributesBuilder()
198                         .setAmplifiedLink(new AmplifiedLinkBuilder().setAmplifiedLink(amplifiedLinkValues).build())
199                         .setSpan(new SpanBuilder().build())
200                         .build())
201                 .build();
202         org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.Link1 aug22 =
203                 new org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.Link1Builder()
204                         .setLinkType(OpenroadmLinkType.ROADMTOROADM).build();
205         // create the roadm-to-roadm link to be measured
206         Link roadm2roadmLink = new LinkBuilder().setLinkId(new LinkId("ROADM-A1-to-ROADM-C1"))
207                 .setSource(new SourceBuilder().setSourceNode(ietfNodeA.getNodeId())
208                         .setSourceTp(new TpId("DEG2-TTP-TXRX")).build())
209                 .setDestination(new DestinationBuilder().setDestNode(ietfNodeC.getNodeId())
210                         .setDestTp(new TpId("DEG1-TTP-TXRX")).build())
211                 .addAugmentation(aug11)
212                 .addAugmentation(aug12)
213                 .addAugmentation(aug21)
214                 .addAugmentation(aug22)
215                 .build();
216         // create the ietf network
217         Network1 openroadmAugmToIetfNetwork = new Network1Builder()
218                 .setLink(Map.of(roadm2roadmLink.key(),roadm2roadmLink)).build();
219         // openroadm Topology builder
220         NetworkBuilder ietfNetworkBldr = new NetworkBuilder()
221                 .setNetworkId(new NetworkId("openroadm-topology"))
222                 .setNode(ietfNodeList)
223                 .addAugmentation(openroadmAugmToIetfNetwork);
224
225         return ietfNetworkBldr.build();
226     }
227
228     private static Augmentation<Node> getNode1AugImpl() {
229         return  new Node1Builder().setNodeType(OpenroadmNodeType.DEGREE).build();
230     }
231
232     public static Mapping getMapping1() {
233         return new MappingBuilder().setLogicalConnectionPoint("DEG2-TTP-TXRX").setSupportingOts("OTS-DEG2-TTP-TXRX")
234                 .build();
235     }
236
237     public static Mapping getMapping2() {
238         return new MappingBuilder().setLogicalConnectionPoint("DEG1-TTP-TXRX").setSupportingOts("OTS-DEG1-TTP-TXRX")
239                 .build();
240     }
241
242     public static Optional<CurrentPmList> getCurrentPmListA() {
243         Measurement measurementA = new MeasurementBuilder()
244                 .setGranularity(org.opendaylight.yang.gen.v1.http.org.openroadm.pm.types.rev171215.PmGranularity._15min)
245                 .setPmParameterValue(new PmDataType(new BigDecimal("-3.5")))
246                 .setValidity(Validity.Complete)
247                 .build();
248         CurrentPm cpA = new CurrentPmBuilder()
249                 .setType(org.opendaylight.yang.gen.v1.http.org.openroadm.pm.types.rev171215.PmNamesEnum
250                         .OpticalPowerOutput)
251                 .setMeasurement(Map.of(measurementA.key(),measurementA))
252                 .build();
253         InstanceIdentifier<Interface> interfaceIIDA = InstanceIdentifier.create(OrgOpenroadmDevice.class)
254                 .child(Interface.class, new InterfaceKey("OTS-DEG2-TTP-TXRX"));
255         CurrentPmEntry currentPmEntryA = new CurrentPmEntryBuilder()
256                 .setCurrentPm(Map.of(cpA.key(),cpA))
257                 .setPmResourceInstance(interfaceIIDA)
258                 .setPmResourceType(
259                         org.opendaylight.yang.gen.v1.http.org.openroadm.resource.types.rev181019.ResourceTypeEnum
260                                 .Interface)
261                 .setPmResourceTypeExtension("")
262                 .setRetrievalTime(new DateAndTime("2018-06-07T13:22:58+00:00"))
263                 .build();
264         return Optional.of(new CurrentPmListBuilder()
265                 .setCurrentPmEntry(Map.of(currentPmEntryA.key(),currentPmEntryA)).build());
266     }
267
268     public static Optional<CurrentPmList> getCurrentPmListC() {
269         Measurement measurementC = new MeasurementBuilder()
270                 .setGranularity(org.opendaylight.yang.gen.v1.http.org.openroadm.pm.types.rev171215.PmGranularity._15min)
271                 .setPmParameterValue(new PmDataType(new BigDecimal("-18.1")))
272                 .setValidity(Validity.Complete)
273                 .build();
274         Map<MeasurementKey,Measurement> measurementListC = new HashMap<>();
275         measurementListC.put(measurementC.key(),measurementC);
276         CurrentPm cpC = new CurrentPmBuilder()
277                 .setType(org.opendaylight.yang.gen.v1.http.org.openroadm.pm.types.rev171215.PmNamesEnum
278                         .OpticalPowerInput)
279                 .setMeasurement(measurementListC)
280                 .build();
281         InstanceIdentifier<Interface> interfaceIIDC = InstanceIdentifier.create(OrgOpenroadmDevice.class)
282                 .child(Interface.class, new InterfaceKey("OTS-DEG1-TTP-TXRX"));
283         CurrentPmEntry currentPmEntryC = new CurrentPmEntryBuilder()
284                 .setCurrentPm(Map.of(cpC.key(),cpC))
285                 .setPmResourceInstance(interfaceIIDC)
286                 .setPmResourceType(
287                         org.opendaylight.yang.gen.v1.http.org.openroadm.resource.types.rev181019.ResourceTypeEnum
288                                 .Interface)
289                 .setPmResourceTypeExtension("")
290                 .setRetrievalTime(new DateAndTime("2018-06-07T13:22:58+00:00"))
291                 .build();
292         return Optional.of(new CurrentPmListBuilder()
293                 .setCurrentPmEntry(Map.of(currentPmEntryC.key(),currentPmEntryC)).build());
294     }
295
296     private static org.opendaylight
297             .yang.gen.v1.http.org.openroadm.network.topology.rev200529.Node1 getNode1() {
298         return new org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.Node1Builder()
299                 .setSrgAttributes(getSrgAttributes())
300                 .setDegreeAttributes(getDegAttributes())
301                 .build();
302     }
303
304     private static DegreeAttributes getDegAttributes() {
305         return new DegreeAttributesBuilder()
306                 .setAvailFreqMaps(GridUtils.initFreqMaps4FixedGrid2Available())
307                 .build();
308     }
309
310     private static  SrgAttributes getSrgAttributes() {
311         return new SrgAttributesBuilder().setAvailFreqMaps(GridUtils.initFreqMaps4FixedGrid2Available()).build();
312     }
313
314
315 }