add unit test for graph package.
[transportpce.git] / pce / src / test / java / org / opendaylight / transportpce / pce / utils / NodeUtils.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.pce.utils;
10
11 import com.google.common.collect.ImmutableList;
12 import java.math.BigDecimal;
13 import java.util.ArrayList;
14 import java.util.List;
15 import org.opendaylight.transportpce.common.NetworkUtils;
16 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.link.types.rev181130.FiberPmd;
17 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.link.types.rev181130.RatioDB;
18 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev181130.Link1Builder;
19 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.state.types.rev181130.State;
20 import org.opendaylight.yang.gen.v1.http.org.openroadm.equipment.states.types.rev181130.AdminStates;
21 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.amplified.link.attributes.AmplifiedLink;
22 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.amplified.link.attributes.amplified.link.SectionElementBuilder;
23 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.span.attributes.LinkConcatenation;
24 import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.span.attributes.LinkConcatenationBuilder;
25 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130.Link1;
26 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130.TerminationPoint1;
27 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130.TerminationPoint1Builder;
28 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130.networks.network.link.OMSAttributesBuilder;
29 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130.networks.network.link.oms.attributes.AmplifiedLinkBuilder;
30 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130.networks.network.link.oms.attributes.SpanBuilder;
31 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev181130.OpenroadmLinkType;
32 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev181130.OpenroadmTpType;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NodeId;
35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.NodeBuilder;
36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.NodeKey;
37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.node.SupportingNode;
38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.node.SupportingNodeBuilder;
39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.LinkId;
40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Node1;
41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Node1Builder;
42 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.Link;
43 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.LinkBuilder;
44 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.LinkKey;
45 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.link.DestinationBuilder;
46 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.link.SourceBuilder;
47 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.node.TerminationPointBuilder;
48 import org.opendaylight.yangtools.yang.binding.Augmentation;
49
50 public class NodeUtils {
51
52     private static final String LINK_ID_FORMAT = "%1$s-%2$sto%3$s-%4$s";
53
54     public static LinkBuilder createLinkBuilder(
55             String srcNode, String destNode, String srcTp, String destTp, Link1Builder link1Builder) {
56         SourceBuilder ietfSrcLinkBldr =
57                 new SourceBuilder().setSourceNode(new NodeId(srcNode)).setSourceTp(srcTp);
58         //create destination link
59         DestinationBuilder ietfDestLinkBldr =
60                 new DestinationBuilder().setDestNode(new NodeId(destNode)).setDestTp(destTp);
61         LinkId linkId = new LinkId(String.format(LINK_ID_FORMAT, srcNode, srcTp, destNode, destTp));
62
63         LinkId oppositeLinkId = new LinkId("OpenROADM-3-2-DEG1-to-OpenROADM-3-1-DEG1");
64         //For setting up attributes for openRoadm augment
65         OMSAttributesBuilder omsAttributesBuilder =
66                 new OMSAttributesBuilder().setOppositeLink(oppositeLinkId);
67
68         // Augementation
69         Augmentation<Link> aug11 = new org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130
70                 .Link1Builder()
71                 .setAdministrativeGroup(Long.valueOf(123))
72                 .setAdministrativeState(State.InService)
73                 .setAmplified(true)
74                 .setLinkLatency(Long.valueOf(123))
75                 .setLinkLength(BigDecimal.valueOf(123))
76                 .setOMSAttributes(new OMSAttributesBuilder()
77                         .setOppositeLink(new LinkId("OpenROADM-3-2-DEG1-to-OpenROADM-3-1-DEG1"))
78                         .setSpan(new SpanBuilder().build())
79                         .setTEMetric(Long.valueOf(123)).build())
80                 .setOperationalState(State.InService).build();
81
82         LinkBuilder linkBuilder = new LinkBuilder()
83                 .setSource(ietfSrcLinkBldr.build())
84                 .setDestination(ietfDestLinkBldr.build())
85                 .setLinkId(linkId)
86                 .addAugmentation(Link1.class, aug11)
87                 .withKey(new LinkKey(linkId));
88
89         linkBuilder.addAugmentation(
90                 org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev181130.Link1.class,
91                 link1Builder.build());
92         return linkBuilder;
93     }
94
95     public static LinkBuilder createRoadmToRoadm(String srcNode, String destNode, String srcTp, String destTp) {
96         Link1Builder link1Builder = new Link1Builder()
97                 .setLinkLatency(30L)
98                 .setLinkType(OpenroadmLinkType.ROADMTOROADM);
99         return createLinkBuilder(srcNode, destNode, srcTp, destTp, link1Builder);
100
101     }
102
103     public static List<SupportingNode> geSupportingNodes() {
104         List<SupportingNode> supportingNodes1 = new ArrayList<>();
105 //
106         supportingNodes1
107                 .add(new SupportingNodeBuilder()
108                         .setNodeRef(new NodeId("node 1"))
109                         .setNetworkRef(new NetworkId(NetworkUtils.CLLI_NETWORK_ID))
110                         .build());
111
112         supportingNodes1
113                 .add(new SupportingNodeBuilder()
114                         .setNodeRef(new NodeId("node 2"))
115                         .setNetworkRef(new NetworkId(NetworkUtils.UNDERLAY_NETWORK_ID))
116                         .build());
117         return supportingNodes1;
118     }
119
120     public static NodeBuilder getNodeBuilder(List<SupportingNode> supportingNodes1) {
121
122
123         //update tp of nodes
124         TerminationPointBuilder xpdrTpBldr = new TerminationPointBuilder();
125         TerminationPoint1Builder tp1Bldr = new TerminationPoint1Builder();
126
127         tp1Bldr.setTpType(OpenroadmTpType.XPONDERNETWORK);
128         xpdrTpBldr.addAugmentation(TerminationPoint1.class, tp1Bldr.build());
129         org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Node1 node1 =
130                 new Node1Builder().setTerminationPoint(ImmutableList.of(xpdrTpBldr.build())).build();
131
132
133         return new NodeBuilder()
134                 .setNodeId(new NodeId("node 1"))
135                 .withKey(new NodeKey(new NodeId("node 1")))
136                 .addAugmentation(
137                         Node1.class, node1)
138                 .setSupportingNode(supportingNodes1);
139     }
140
141     private Link genereateLinkBuilder() {
142
143         List<LinkConcatenation> linkConcentationValues = new ArrayList<>();
144         LinkConcatenation linkConcatenation = new LinkConcatenationBuilder()
145                 .setFiberType(LinkConcatenation.FiberType.Truewave)
146                 .setPmd(new FiberPmd(BigDecimal.ONE))
147                 .setSRLGId(Long.valueOf(1))
148                 .setSRLGLength(Long.valueOf(1))
149                 .build();
150         LinkConcatenation linkConcatenation2 = new LinkConcatenationBuilder()
151                 .setFiberType(LinkConcatenation.FiberType.Truewave)
152                 .setPmd(new FiberPmd(BigDecimal.ONE))
153                 .setSRLGId(Long.valueOf(1))
154                 .setSRLGLength(Long.valueOf(1))
155                 .build();
156         linkConcentationValues.add(linkConcatenation);
157         linkConcentationValues.add(linkConcatenation2);
158
159         List<AmplifiedLink>
160                 amplifiedLinkValues = new ArrayList<>();
161         org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.amplified.link.attributes.AmplifiedLink al =
162                 new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.amplified.link.attributes
163                         .AmplifiedLinkBuilder().setSectionElement(new SectionElementBuilder()
164                         .setSectionElement(new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130
165                                 .amplified.link.attributes.amplified.link.section.element.section.element
166                                 .SpanBuilder()
167                                 .setSpan(
168                                         new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.amplified
169                                                 .link.attributes.amplified.link.section.element.section
170                                                 .element.span.SpanBuilder()
171                                                 .setAdministrativeState(AdminStates.InService)
172                                                 .setAutoSpanloss(true)
173                                                 .setClfi("clfi")
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(Integer.valueOf(1)).build();
182         org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.amplified.link.attributes.AmplifiedLink al2 =
183                 new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.amplified.link.attributes
184                         .AmplifiedLinkBuilder().setSectionElement(new SectionElementBuilder()
185                         .setSectionElement(
186                                 new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130.amplified.link
187                                         .attributes.amplified.link.section.element.section.element.SpanBuilder()
188                                         .setSpan(
189                                                 new org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev181130
190                                                         .amplified.link
191                                                         .attributes.amplified.link.section.element.section.element.span
192                                                         .SpanBuilder()
193                                                         .setAdministrativeState(AdminStates.InService)
194                                                         .setAutoSpanloss(true)
195                                                         .setClfi("clfi")
196                                                         .setEngineeredSpanloss(new RatioDB(BigDecimal.ONE))
197                                                         .setLinkConcatenation(linkConcentationValues)
198                                                         .setSpanlossBase(new RatioDB(BigDecimal.ONE))
199                                                         .setSpanlossCurrent(new RatioDB(BigDecimal.ONE))
200                                                         .build())
201                                         .build())
202                         .build())
203                         .setSectionEltNumber(Integer.valueOf(1)).build();
204
205         amplifiedLinkValues.add(al);
206         amplifiedLinkValues.add(al2);
207         Augmentation<Link> aug11 =
208                 new org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130.Link1Builder()
209                         .setAdministrativeGroup(Long.valueOf(123))
210                         .setAdministrativeState(State.InService)
211                         .setAmplified(true)
212                         .setLinkLatency(Long.valueOf(123))
213                         .setLinkLength(BigDecimal.valueOf(123))
214                         .setOMSAttributes(new OMSAttributesBuilder()
215                                 .setAmplifiedLink(new AmplifiedLinkBuilder().setAmplifiedLink(amplifiedLinkValues)
216                                         .build())
217                                 .setOppositeLink(new LinkId("link 1"))
218                                 .setSpan(new SpanBuilder().build())
219                                 .setTEMetric(Long.valueOf(123)).build())
220                         .setOperationalState(State.InService).build();
221
222         TransactionUtils.getNetworkForSpanLoss();
223         return new LinkBuilder().setLinkId(new LinkId("OpenROADM-3-1-DEG1-to-OpenROADM-3-2-DEG1"))
224                 .setSource(
225                         new SourceBuilder().setSourceNode(new NodeId("OpenROADM-3-2-DEG1"))
226                         .setSourceTp("DEG1-TTP-TX").build())
227                 .setDestination(
228                         new DestinationBuilder().setDestNode(new NodeId("OpenROADM-3-1-DEG1"))
229                                 .setDestTp("DEG1-TTP-RX").build())
230                 .addAugmentation(Link1.class, aug11)
231                 .build();
232
233
234     }
235 }