Add new revision for pcep types model
[bgpcep.git] / pcep / tunnel / tunnel-provider / src / test / java / org / opendaylight / bgpcep / pcep / tunnel / provider / TunnelProgrammingTest.java
1 /*
2  * Copyright (c) 2014 Cisco 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 package org.opendaylight.bgpcep.pcep.tunnel.provider;
10
11 import com.google.common.collect.Lists;
12 import com.google.common.util.concurrent.ListenableFuture;
13 import java.util.Collections;
14 import java.util.List;
15 import java.util.concurrent.ExecutionException;
16 import java.util.concurrent.Executor;
17 import org.junit.Assert;
18 import org.junit.Before;
19 import org.junit.Test;
20 import org.mockito.Mock;
21 import org.mockito.Mockito;
22 import org.mockito.MockitoAnnotations;
23 import org.opendaylight.bgpcep.programming.spi.Instruction;
24 import org.opendaylight.bgpcep.programming.spi.InstructionScheduler;
25 import org.opendaylight.bgpcep.programming.spi.SchedulerException;
26 import org.opendaylight.mdsal.binding.api.RpcConsumerRegistry;
27 import org.opendaylight.mdsal.binding.api.RpcProviderService;
28 import org.opendaylight.mdsal.binding.api.WriteTransaction;
29 import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractConcurrentDataBrokerTest;
30 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
31 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.network.concepts.rev131125.Bandwidth;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.network.topology.rev140113.NetworkTopologyRef;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.AdministrativeStatus;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.PcepCreateP2pTunnelInput1;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.PcepCreateP2pTunnelInput1Builder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.PcepUpdateTunnelInput1;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.PcepUpdateTunnelInput1Builder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.ClassType;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.endpoints.address.family.Ipv4Case;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.endpoints.address.family.ipv4._case.Ipv4;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.explicit.route.object.ero.Subobject;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.programming.rev150720.InstructionStatus;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.programming.rev150720.SubmitInstructionInput;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.basic.explicit.route.subobjects.subobject.type.IpPrefixCase;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.basic.explicit.route.subobjects.subobject.type.IpPrefixCaseBuilder;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.basic.explicit.route.subobjects.subobject.type.ip.prefix._case.IpPrefixBuilder;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.AddLspInput;
53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.AddLspOutput;
54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.NetworkTopologyPcepService;
55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.RemoveLspInput;
56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.RemoveLspOutput;
57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.UpdateLspInput;
58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.UpdateLspOutput;
59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.add.lsp.args.Arguments;
60 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.p2p.rev130819.tunnel.p2p.path.cfg.attributes.ExplicitHops;
61 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.p2p.rev130819.tunnel.p2p.path.cfg.attributes.ExplicitHopsBuilder;
62 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepCreateP2pTunnelInputBuilder;
63 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepDestroyTunnelInputBuilder;
64 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepUpdateTunnelInputBuilder;
65 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.rev181109.ExplicitHops1;
66 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.rev181109.ExplicitHops1Builder;
67 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.rev181109.Link1;
68 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.rev181109.Link1Builder;
69 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.rev181109.SupportingNode1;
70 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.rev181109.SupportingNode1Builder;
71 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.rev181109.tunnel.pcep.supporting.node.attributes.PathComputationClientBuilder;
72 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.programming.rev130930.create.p2p.tunnel.input.DestinationBuilder;
73 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.programming.rev130930.create.p2p.tunnel.input.SourceBuilder;
74 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.LinkId;
75 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
76 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
77 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
78 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TpId;
79 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
80 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
81 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
82 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Link;
83 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.LinkBuilder;
84 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.LinkKey;
85 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
86 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
87 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
88 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointBuilder;
89 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointKey;
90 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.node.attributes.SupportingNode;
91 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.node.attributes.SupportingNodeBuilder;
92 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.node.attributes.SupportingNodeKey;
93 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.nt.l3.unicast.igp.topology.rev131021.TerminationPoint1;
94 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.nt.l3.unicast.igp.topology.rev131021.TerminationPoint1Builder;
95 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.nt.l3.unicast.igp.topology.rev131021.igp.termination.point.attributes.IgpTerminationPointAttributesBuilder;
96 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.nt.l3.unicast.igp.topology.rev131021.igp.termination.point.attributes.igp.termination.point.attributes.termination.point.type.IpBuilder;
97 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
98 import org.opendaylight.yangtools.yang.common.RpcResult;
99 import org.osgi.framework.BundleContext;
100
101 public class TunnelProgrammingTest extends AbstractConcurrentDataBrokerTest {
102
103     private static final TopologyId TOPOLOGY_ID = new TopologyId("tunnel-topo");
104     private static final InstanceIdentifier<Topology> TOPO_IID = InstanceIdentifier.builder(NetworkTopology.class)
105             .child(Topology.class, new TopologyKey(TOPOLOGY_ID)).build();
106
107     private static final String NODE1_IPV4 = "127.0.0.1";
108     private static final NodeId NODE1_ID = new NodeId("pcc://" + NODE1_IPV4);
109     private static final TpId TP1_ID = new TpId(NODE1_IPV4);
110
111     private static final String NODE2_IPV4 = "127.0.0.10";
112     private static final NodeId NODE2_ID = new NodeId("pcc://" + NODE2_IPV4);
113     private static final TpId TP2_ID = new TpId(NODE2_IPV4);
114
115     private static final LinkId LINK1_ID = new LinkId("link1");
116
117     private static final String IPV4_PREFIX1 = "195.20.160.40/32";
118     private static final String IPV4_PREFIX2 = "201.20.160.43/32";
119
120     @Mock
121     private NetworkTopologyPcepService topologyService;
122     @Mock
123     private InstructionScheduler scheduler;
124     @Mock
125     private ListenableFuture<Instruction> instructionFuture;
126     @Mock
127     private Instruction instruction;
128     @Mock
129     private ClusterSingletonServiceProvider cssp;
130     @Mock
131     private RpcProviderService rpr;
132     @Mock
133     private RpcConsumerRegistry rpcs;
134     @Mock
135     private BundleContext bundleContext;
136
137     private TunnelProgramming tunnelProgramming;
138
139     private AddLspInput addLspInput;
140     private UpdateLspInput updateLspInput;
141     private RemoveLspInput removeLspInput;
142
143     @Mock
144     private ListenableFuture<RpcResult<AddLspOutput>> futureAddLspOutput;
145     @Mock
146     private ListenableFuture<RpcResult<UpdateLspOutput>> futureUpdateLspOutput;
147     @Mock
148     private ListenableFuture<RpcResult<RemoveLspOutput>> futureRemoveLspOutput;
149
150     private static Node createNode(final NodeId nodeId, final TpId tpId, final String ipv4Address) {
151         final TerminationPointBuilder tpBuilder = new TerminationPointBuilder();
152         tpBuilder.setTpId(tpId);
153         tpBuilder.withKey(new TerminationPointKey(tpId));
154         tpBuilder.addAugmentation(TerminationPoint1.class, new TerminationPoint1Builder()
155                 .setIgpTerminationPointAttributes(new IgpTerminationPointAttributesBuilder()
156                         .setTerminationPointType(new IpBuilder()
157                                 .setIpAddress(Collections.singletonList(new IpAddress(new Ipv4Address(ipv4Address))))
158                                 .build()).build()).build());
159         final NodeBuilder nodeBuilder = new NodeBuilder();
160         nodeBuilder.setNodeId(nodeId);
161         nodeBuilder.withKey(new NodeKey(nodeId));
162         nodeBuilder.setTerminationPoint(Lists.newArrayList(tpBuilder.build()));
163         final SupportingNode supportingNode = new SupportingNodeBuilder()
164                 .withKey(new SupportingNodeKey(nodeId, new TopologyId("dummy")))
165                 .addAugmentation(SupportingNode1.class, new SupportingNode1Builder()
166                         .setPathComputationClient(new PathComputationClientBuilder()
167                                 .setControlling(true).build()).build()).build();
168         nodeBuilder.setSupportingNode(Lists.newArrayList(supportingNode));
169         return nodeBuilder.build();
170     }
171
172     private static ExplicitHops createExplicitHop(final String ipv4Prefix) {
173         final ExplicitHopsBuilder explcitHopsBuilder = new ExplicitHopsBuilder();
174         explcitHopsBuilder.addAugmentation(ExplicitHops1.class, new ExplicitHops1Builder()
175                 .setSubobjectType(new IpPrefixCaseBuilder().setIpPrefix(new IpPrefixBuilder()
176                         .setIpPrefix(new IpPrefix(new Ipv4Prefix(ipv4Prefix))).build()).build()).build());
177         return explcitHopsBuilder.build();
178     }
179
180     @Before
181     public void setUp() throws SchedulerException, InterruptedException, ExecutionException {
182         MockitoAnnotations.initMocks(this);
183         Mockito.doReturn(true).when(this.instruction).checkedExecutionStart();
184         Mockito.doNothing().when(this.instruction).executionCompleted(InstructionStatus.Failed, null);
185         Mockito.doAnswer(invocation -> {
186             final Runnable callback = (Runnable) invocation.getArguments()[0];
187             callback.run();
188             return null;
189         }).when(this.instructionFuture).addListener(Mockito.any(Runnable.class), Mockito.any(Executor.class));
190         Mockito.doAnswer(invocation -> {
191             final Runnable callback = (Runnable) invocation.getArguments()[0];
192             callback.run();
193             return null;
194         }).when(this.futureAddLspOutput).addListener(Mockito.any(Runnable.class), Mockito.any(Executor.class));
195         Mockito.doAnswer(invocation -> {
196             final Runnable callback = (Runnable) invocation.getArguments()[0];
197             callback.run();
198             return null;
199         }).when(this.futureUpdateLspOutput).addListener(Mockito.any(Runnable.class), Mockito.any(Executor.class));
200         Mockito.doAnswer(invocation -> {
201             final Runnable callback = (Runnable) invocation.getArguments()[0];
202             callback.run();
203             return null;
204         }).when(this.futureRemoveLspOutput).addListener(Mockito.any(Runnable.class), Mockito.any(Executor.class));
205         Mockito.doAnswer(invocation -> {
206             TunnelProgrammingTest.this.addLspInput = (AddLspInput) invocation.getArguments()[0];
207             return TunnelProgrammingTest.this.futureAddLspOutput;
208         }).when(this.topologyService).addLsp(Mockito.any(AddLspInput.class));
209         Mockito.doAnswer(invocation -> {
210             TunnelProgrammingTest.this.updateLspInput = (UpdateLspInput) invocation.getArguments()[0];
211             return TunnelProgrammingTest.this.futureUpdateLspOutput;
212         }).when(this.topologyService).updateLsp(Mockito.any(UpdateLspInput.class));
213         Mockito.doAnswer(invocation -> {
214             TunnelProgrammingTest.this.removeLspInput = (RemoveLspInput) invocation.getArguments()[0];
215             return TunnelProgrammingTest.this.futureRemoveLspOutput;
216         }).when(this.topologyService).removeLsp(Mockito.any(RemoveLspInput.class));
217         Mockito.doReturn(this.instruction).when(this.instructionFuture).get();
218         Mockito.doReturn(true).when(this.instructionFuture).isDone();
219         Mockito.doReturn(this.instructionFuture).when(this.scheduler)
220                 .scheduleInstruction(Mockito.any(SubmitInstructionInput.class));
221
222         Mockito.doReturn(this.topologyService).when(this.rpcs)
223                 .getRpcService(NetworkTopologyPcepService.class);
224
225         createInitialTopology();
226         final TunnelProviderDependencies dependencies = new TunnelProviderDependencies(getDataBroker(), this.cssp,
227                 this.rpr, this.rpcs, this.bundleContext);
228         this.tunnelProgramming = new TunnelProgramming(this.scheduler, dependencies);
229     }
230
231     @Test
232     public void testTunnelProgramming() throws InterruptedException, ExecutionException {
233         final Bandwidth bwd = new Bandwidth(new byte[]{0x00, 0x00, 0x00, (byte) 0xff});
234         final ClassType classType = new ClassType((short) 1);
235         final String tunnelName = "create-tunnel";
236         final NetworkTopologyRef topologyRef = new NetworkTopologyRef(TOPO_IID);
237         // create tunnel
238         final PcepCreateP2pTunnelInputBuilder createInputBuilder = new PcepCreateP2pTunnelInputBuilder();
239         createInputBuilder.setDestination(new DestinationBuilder().setNode(NODE2_ID).setTp(TP2_ID).build());
240         createInputBuilder.setSource(new SourceBuilder().setNode(NODE1_ID).setTp(TP1_ID).build());
241         createInputBuilder.setNetworkTopologyRef(topologyRef);
242         createInputBuilder.setBandwidth(bwd);
243         createInputBuilder.setClassType(classType);
244         createInputBuilder.setSymbolicPathName(tunnelName);
245         createInputBuilder.setExplicitHops(Lists.newArrayList());
246         createInputBuilder.addAugmentation(PcepCreateP2pTunnelInput1.class, new PcepCreateP2pTunnelInput1Builder()
247                 .setAdministrativeStatus(AdministrativeStatus.Active).build());
248         this.tunnelProgramming.pcepCreateP2pTunnel(createInputBuilder.build());
249         //check add-lsp input
250         Assert.assertNotNull(this.addLspInput);
251         Assert.assertEquals(tunnelName, this.addLspInput.getName());
252         final Arguments agrs = this.addLspInput.getArguments();
253         Assert.assertNotNull(agrs);
254         Assert.assertEquals(bwd, agrs.getBandwidth().getBandwidth());
255         Assert.assertEquals(classType, agrs.getClassType().getClassType());
256         final Ipv4 ipv4Endpoints = ((Ipv4Case) agrs.getEndpointsObj().getAddressFamily()).getIpv4();
257         Assert.assertEquals(NODE1_IPV4, ipv4Endpoints.getSourceIpv4Address().getValue());
258         Assert.assertEquals(NODE2_IPV4, ipv4Endpoints.getDestinationIpv4Address().getValue());
259         Assert.assertEquals(NODE1_ID.getValue(), this.addLspInput.getNode().getValue());
260         createLink();
261
262         // update tunnel
263         final PcepUpdateTunnelInputBuilder updateInputBuilder = new PcepUpdateTunnelInputBuilder();
264         updateInputBuilder.setNetworkTopologyRef(topologyRef);
265         updateInputBuilder.setBandwidth(bwd);
266         updateInputBuilder.setClassType(classType);
267         updateInputBuilder.setExplicitHops(Lists.newArrayList(createExplicitHop(IPV4_PREFIX1),
268                 createExplicitHop(IPV4_PREFIX2)));
269         updateInputBuilder.setLinkId(LINK1_ID);
270         updateInputBuilder.addAugmentation(PcepUpdateTunnelInput1.class, new PcepUpdateTunnelInput1Builder()
271                 .setAdministrativeStatus(AdministrativeStatus.Active).build());
272         this.tunnelProgramming.pcepUpdateTunnel(updateInputBuilder.build());
273         //check update-lsp input
274         Assert.assertNotNull(this.updateLspInput);
275         Assert.assertEquals(LINK1_ID.getValue(), this.updateLspInput.getName());
276         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.update.lsp
277                 .args.Arguments updArgs = this.updateLspInput.getArguments();
278         Assert.assertEquals(2, updArgs.getEro().getSubobject().size());
279         final List<Subobject> subObjects = updArgs.getEro().getSubobject();
280         final IpPrefixCase prefix1 = (IpPrefixCase) subObjects.get(0).getSubobjectType();
281         final IpPrefixCase prefix2 = (IpPrefixCase) subObjects.get(1).getSubobjectType();
282         Assert.assertEquals(IPV4_PREFIX1, prefix1.getIpPrefix().getIpPrefix().getIpv4Prefix().getValue());
283         Assert.assertEquals(IPV4_PREFIX2, prefix2.getIpPrefix().getIpPrefix().getIpv4Prefix().getValue());
284
285         // delete tunnel
286         final PcepDestroyTunnelInputBuilder destroyInputBuilder = new PcepDestroyTunnelInputBuilder();
287         destroyInputBuilder.setLinkId(LINK1_ID);
288         destroyInputBuilder.setNetworkTopologyRef(topologyRef);
289         this.tunnelProgramming.pcepDestroyTunnel(destroyInputBuilder.build());
290         Assert.assertNotNull(this.removeLspInput);
291         Assert.assertEquals(LINK1_ID.getValue(), this.removeLspInput.getName());
292         Assert.assertEquals(NODE1_ID.getValue(), this.removeLspInput.getNode().getValue());
293     }
294
295     private void createInitialTopology() throws InterruptedException, ExecutionException {
296         final TopologyBuilder topologyBuilder = new TopologyBuilder();
297         topologyBuilder.withKey(new TopologyKey(TOPOLOGY_ID));
298         topologyBuilder.setServerProvided(true);
299         topologyBuilder.setTopologyId(TOPOLOGY_ID);
300         topologyBuilder.setNode(Lists.newArrayList(createNode(NODE1_ID, TP1_ID, NODE1_IPV4),
301                 createNode(NODE2_ID, TP2_ID, NODE2_IPV4)));
302         final WriteTransaction wTx = getDataBroker().newWriteOnlyTransaction();
303         wTx.put(LogicalDatastoreType.OPERATIONAL, TOPO_IID, topologyBuilder.build(), true);
304         wTx.commit().get();
305     }
306
307     private void createLink() throws InterruptedException, ExecutionException {
308         final LinkBuilder linkBuilder = new LinkBuilder();
309         linkBuilder.setSource(new org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology
310                 .rev131021.link.attributes.SourceBuilder().setSourceNode(NODE1_ID).setSourceTp(TP1_ID).build());
311         linkBuilder.setDestination(new org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology
312                 .rev131021.link.attributes.DestinationBuilder().setDestNode(NODE2_ID).setDestTp(TP2_ID).build());
313         linkBuilder.setLinkId(LINK1_ID);
314         linkBuilder.withKey(new LinkKey(LINK1_ID));
315         linkBuilder.addAugmentation(Link1.class, new Link1Builder().setSymbolicPathName(LINK1_ID.getValue()).build());
316         final WriteTransaction wTx = getDataBroker().newWriteOnlyTransaction();
317         wTx.put(LogicalDatastoreType.OPERATIONAL, TOPO_IID.builder().child(Link.class, new LinkKey(LINK1_ID)).build(),
318                 linkBuilder.build(), true);
319         wTx.commit().get();
320     }
321
322 }