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