pcep/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/ParserToSalTest.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.topology.provider;
   9
  10 import static junit.framework.Assert.assertFalse;
  11 import static org.mockito.Matchers.any;
  12 import static org.mockito.Mockito.doAnswer;
  13 import static org.mockito.Mockito.doReturn;
  14 import static org.mockito.Mockito.mock;
  15
  16 import com.google.common.base.Optional;
  17 import com.google.common.collect.Lists;
  18 import io.netty.channel.Channel;
  19 import io.netty.channel.ChannelFuture;
  20 import io.netty.channel.ChannelHandler;
  21 import io.netty.channel.ChannelPipeline;
  22 import io.netty.channel.EventLoop;
  23 import io.netty.util.concurrent.Promise;
  24 import java.net.InetSocketAddress;
  25 import java.net.SocketAddress;
  26 import java.util.List;
  27 import java.util.concurrent.ExecutionException;
  28 import java.util.concurrent.TimeUnit;
  29 import junit.framework.Assert;
  30 import org.junit.After;
  31 import org.junit.Before;
  32 import org.junit.Test;
  33 import org.mockito.Mock;
  34 import org.mockito.MockitoAnnotations;
  35 import org.mockito.invocation.InvocationOnMock;
  36 import org.mockito.stubbing.Answer;
  37 import org.opendaylight.controller.md.sal.binding.test.AbstractDataBrokerTest;
  38 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  39 import org.opendaylight.protocol.pcep.impl.DefaultPCEPSessionNegotiator;
  40 import org.opendaylight.protocol.pcep.impl.PCEPSessionImpl;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Tlvs1;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Tlvs1Builder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.stateful.capability.tlv.StatefulBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Pcrpt;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PcrptBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PlspId;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SymbolicPathName;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.identifiers.tlv.LspIdentifiersBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.LspBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.PcrptMessageBuilder;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.Reports;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.ReportsBuilder;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.reports.PathBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.symbolic.path.name.tlv.SymbolicPathNameBuilder;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.EroBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.LspId;
  60 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  61 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  62 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  63 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  64 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  65 import org.opendaylight.yangtools.yang.binding.Notification;
  66
  67 public class ParserToSalTest extends AbstractDataBrokerTest {
  68
  69     private static final String TEST_TOPOLOGY_NAME = "testtopo";
  70
  71     private List<Notification> receivedMsgs;
  72
  73     private PCEPSessionImpl session;
  74
  75     @Mock
  76     private EventLoop eventLoop;
  77
  78     @Mock
  79     private Channel clientListener;
  80
  81     @Mock
  82     private ChannelPipeline pipeline;
  83
  84     private final Open localPrefs = new OpenBuilder().setDeadTimer((short) 30).setKeepalive((short) 10).setTlvs(
  85             new TlvsBuilder().addAugmentation(Tlvs1.class, new Tlvs1Builder().setStateful(new StatefulBuilder().build()).build()).build()).build();
  86
  87     private Pcrpt rptmsg;
  88
  89     private ServerSessionManager manager;
  90
  91     @Before
  92     public void setUp() throws Exception {
  93         MockitoAnnotations.initMocks(this);
  94
  95         doAnswer(new Answer<Object>() {
  96             @Override
  97             public Object answer(final InvocationOnMock invocation) {
  98                 final Object[] args = invocation.getArguments();
  99                 ParserToSalTest.this.receivedMsgs.add((Notification) args[0]);
 100                 return mock(ChannelFuture.class);
 101             }
 102         }).when(this.clientListener).writeAndFlush(any(Notification.class));
 103         doReturn("TestingChannel").when(this.clientListener).toString();
 104         doReturn(this.pipeline).when(this.clientListener).pipeline();
 105         doReturn(this.pipeline).when(this.pipeline).replace(any(ChannelHandler.class), any(String.class), any(ChannelHandler.class));
 106         doReturn(this.eventLoop).when(this.clientListener).eventLoop();
 107         doReturn(null).when(this.eventLoop).schedule(any(Runnable.class), any(long.class), any(TimeUnit.class));
 108         doReturn(true).when(this.clientListener).isActive();
 109         final SocketAddress ra = new InetSocketAddress("127.0.0.1", 4189);
 110         doReturn(ra).when(this.clientListener).remoteAddress();
 111         final SocketAddress la = new InetSocketAddress("127.0.0.1", 30000);
 112         doReturn(la).when(this.clientListener).localAddress();
 113
 114         doReturn(mock(ChannelFuture.class)).when(this.clientListener).close();
 115
 116         this.manager = new ServerSessionManager(getDataBroker(), InstanceIdentifier.builder(NetworkTopology.class).child(
 117                 Topology.class, new TopologyKey(new TopologyId(TEST_TOPOLOGY_NAME))).toInstance(), new Stateful07TopologySessionListenerFactory());
 118         final DefaultPCEPSessionNegotiator neg = new DefaultPCEPSessionNegotiator(mock(Promise.class), this.clientListener, this.manager.getSessionListener(), (short) 1, 5, this.localPrefs);
 119         this.session = neg.createSession(this.clientListener, this.localPrefs, this.localPrefs);
 120
 121         final List<Reports> reports = Lists.newArrayList(new ReportsBuilder().setPath(new PathBuilder().setEro(new EroBuilder().build()).build()).setLsp(
 122                 new LspBuilder().setPlspId(new PlspId(5L)).setSync(false).setRemove(false).setTlvs(
 123                         new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.TlvsBuilder().setLspIdentifiers(new LspIdentifiersBuilder().setLspId(new LspId(1L)).build()).setSymbolicPathName(
 124                                 new SymbolicPathNameBuilder().setPathName(new SymbolicPathName(new byte[] { 22, 34 })).build()).build()).build()).build());
 125         this.rptmsg = new PcrptBuilder().setPcrptMessage(new PcrptMessageBuilder().setReports(reports).build()).build();
 126
 127         Assert.assertTrue(getTopology(TEST_TOPOLOGY_NAME).isPresent());
 128     }
 129
 130     @After
 131     public void tearDown() throws InterruptedException, ExecutionException {
 132         this.manager.close();
 133     }
 134
 135     @Test
 136     public void testUnknownLsp() throws Exception {
 137         this.session.sessionUp();
 138         this.session.handleMessage(this.rptmsg);
 139         final Topology topology = getTopology(TEST_TOPOLOGY_NAME).get();
 140         assertFalse(topology.getNode().isEmpty());
 141     }
 142
 143     private Optional<Topology> getTopology(final String topologyId) throws InterruptedException, ExecutionException {
 144         return getDataBroker().newReadOnlyTransaction().read(LogicalDatastoreType.OPERATIONAL, InstanceIdentifier.builder(NetworkTopology.class).child(Topology.class,
 145                 new TopologyKey(new TopologyId(topologyId))).toInstance()).get();
 146     }
 147 }