pcep/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/ParserToSalTest.java

   1 /*
   2  * Copyright (c) 2013 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 org.mockito.Matchers.any;
  11 import static org.mockito.Mockito.doAnswer;
  12 import static org.mockito.Mockito.doReturn;
  13 import static org.mockito.Mockito.mock;
  14 import io.netty.channel.Channel;
  15 import io.netty.channel.ChannelFuture;
  16 import io.netty.channel.ChannelHandler;
  17 import io.netty.channel.ChannelPipeline;
  18 import io.netty.util.HashedWheelTimer;
  19 import io.netty.util.concurrent.Promise;
  20
  21 import java.io.IOException;
  22 import java.net.InetSocketAddress;
  23 import java.net.SocketAddress;
  24 import java.util.HashMap;
  25 import java.util.List;
  26 import java.util.concurrent.ExecutionException;
  27 import java.util.concurrent.Future;
  28 import java.util.concurrent.TimeUnit;
  29 import java.util.concurrent.TimeoutException;
  30
  31 import org.junit.After;
  32 import org.junit.Before;
  33 import org.junit.Test;
  34 import org.mockito.Matchers;
  35 import org.mockito.Mock;
  36 import org.mockito.Mockito;
  37 import org.mockito.MockitoAnnotations;
  38 import org.mockito.invocation.InvocationOnMock;
  39 import org.mockito.stubbing.Answer;
  40 import org.opendaylight.controller.md.sal.common.api.TransactionStatus;
  41 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
  42 import org.opendaylight.controller.sal.binding.api.data.DataProviderService;
  43 import org.opendaylight.protocol.pcep.impl.DefaultPCEPSessionNegotiator;
  44 import org.opendaylight.protocol.pcep.impl.PCEPSessionImpl;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Tlvs1;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Tlvs1Builder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.stateful.capability.tlv.StatefulBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Pcrpt;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PcrptBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PlspId;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SymbolicPathName;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.LspBuilder;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.PcrptMessageBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.Reports;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcrpt.message.pcrpt.message.ReportsBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.symbolic.path.name.tlv.SymbolicPathNameBuilder;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder;
  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.DataObject;
  65 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  66 import org.opendaylight.yangtools.yang.binding.Notification;
  67 import org.opendaylight.yangtools.yang.common.RpcResult;
  68 import org.slf4j.Logger;
  69 import org.slf4j.LoggerFactory;
  70
  71 import com.google.common.collect.Lists;
  72
  73 public class ParserToSalTest {
  74
  75         private static final Logger LOG = LoggerFactory.getLogger(ParserToSalTest.class);
  76
  77         private List<Notification> receivedMsgs;
  78
  79         private PCEPSessionImpl session;
  80
  81         @Mock
  82         private Channel clientListener;
  83
  84         @Mock
  85         private ChannelPipeline pipeline;
  86
  87         @Mock
  88         DataProviderService providerService;
  89
  90         @Mock
  91         DataModificationTransaction mockedTransaction;
  92
  93         private final Open localPrefs = new OpenBuilder().setDeadTimer((short) 30).setKeepalive((short) 10).setTlvs(
  94                         new TlvsBuilder().addAugmentation(Tlvs1.class, new Tlvs1Builder().setStateful(new StatefulBuilder().build()).build()).build()).build();
  95
  96         private Pcrpt rptmsg;
  97
  98         private ServerSessionManager manager;
  99
 100         @Before
 101         public void setUp() throws IOException {
 102                 MockitoAnnotations.initMocks(this);
 103
 104                 doAnswer(new Answer<Object>() {
 105                         @Override
 106                         public Object answer(final InvocationOnMock invocation) {
 107                                 final Object[] args = invocation.getArguments();
 108                                 ParserToSalTest.this.receivedMsgs.add((Notification) args[0]);
 109                                 return mock(ChannelFuture.class);
 110                         }
 111                 }).when(this.clientListener).writeAndFlush(any(Notification.class));
 112                 doReturn("TestingChannel").when(this.clientListener).toString();
 113                 doReturn(this.pipeline).when(this.clientListener).pipeline();
 114                 doReturn(this.pipeline).when(this.pipeline).replace(any(ChannelHandler.class), any(String.class), any(ChannelHandler.class));
 115                 doReturn(true).when(this.clientListener).isActive();
 116                 final SocketAddress ra = new InetSocketAddress("127.0.0.1", 4189);
 117                 doReturn(ra).when(this.clientListener).remoteAddress();
 118                 final SocketAddress la = new InetSocketAddress("127.0.0.1", 30000);
 119                 doReturn(la).when(this.clientListener).localAddress();
 120
 121                 doReturn(mock(ChannelFuture.class)).when(this.clientListener).close();
 122
 123                 Mockito.doReturn(this.mockedTransaction).when(this.providerService).beginTransaction();
 124                 Mockito.doReturn(new Future<RpcResult<TransactionStatus>>() {
 125                         int i = 0;
 126
 127                         @Override
 128                         public boolean cancel(final boolean mayInterruptIfRunning) {
 129                                 LOG.debug("Cancel.");
 130                                 return false;
 131                         }
 132
 133                         @Override
 134                         public boolean isCancelled() {
 135                                 LOG.debug("Is cancelled.");
 136                                 return false;
 137                         }
 138
 139                         @Override
 140                         public boolean isDone() {
 141                                 this.i++;
 142                                 LOG.debug("Done. {}", this.i);
 143                                 return true;
 144                         }
 145
 146                         @Override
 147                         public RpcResult<TransactionStatus> get() throws InterruptedException, ExecutionException {
 148                                 return null;
 149                         }
 150
 151                         @Override
 152                         public RpcResult<TransactionStatus> get(final long timeout, final TimeUnit unit) throws InterruptedException,
 153                                         ExecutionException, TimeoutException {
 154                                 return null;
 155                         }
 156                 }).when(this.mockedTransaction).commit();
 157
 158                 final HashMap<Object, Object> data = new HashMap<>();
 159
 160                 Mockito.doAnswer(new Answer<Object>() {
 161                         @Override
 162                         public Object answer(final InvocationOnMock invocation) throws Throwable {
 163                                 final Object[] args = invocation.getArguments();
 164                                 LOG.debug("Get key {}", args[0]);
 165                                 return data.get(args[0]);
 166                         }
 167
 168                 }).when(this.mockedTransaction).readOperationalData(Matchers.any(InstanceIdentifier.class));
 169
 170                 Mockito.doAnswer(new Answer<Object>() {
 171                         @Override
 172                         public Object answer(final InvocationOnMock invocation) throws Throwable {
 173                                 final Object[] args = invocation.getArguments();
 174                                 LOG.debug("Get key {}", args[0]);
 175                                 return data.get(args[0]);
 176                         }
 177
 178                 }).when(this.providerService).readOperationalData(Matchers.any(InstanceIdentifier.class));
 179
 180                 Mockito.doAnswer(new Answer<Object>() {
 181
 182                         @Override
 183                         public Object answer(final InvocationOnMock invocation) throws Throwable {
 184                                 data.remove(invocation.getArguments()[0]);
 185                                 return null;
 186                         }
 187
 188                 }).when(this.mockedTransaction).removeOperationalData(Matchers.any(InstanceIdentifier.class));
 189
 190                 Mockito.doAnswer(new Answer<String>() {
 191                         @Override
 192                         public String answer(final InvocationOnMock invocation) throws Throwable {
 193                                 final Object[] args = invocation.getArguments();
 194                                 LOG.debug("Put key {} value {}", args[0]);
 195                                 LOG.debug("Put value {}", args[1]);
 196                                 data.put(args[0], args[1]);
 197                                 return null;
 198                         }
 199
 200                 }).when(this.mockedTransaction).putOperationalData(Matchers.any(InstanceIdentifier.class), Matchers.any(DataObject.class));
 201
 202                 this.manager = new ServerSessionManager(this.providerService, InstanceIdentifier.builder(NetworkTopology.class).child(
 203                                 Topology.class, new TopologyKey(new TopologyId("testtopo"))).toInstance(), new Stateful07TopologySessionListenerFactory());
 204                 final DefaultPCEPSessionNegotiator neg = new DefaultPCEPSessionNegotiator(new HashedWheelTimer(), mock(Promise.class), this.clientListener, this.manager.getSessionListener(), (short) 1, 5, this.localPrefs);
 205                 this.session = neg.createSession(new HashedWheelTimer(), this.clientListener, this.localPrefs, this.localPrefs);
 206
 207                 final List<Reports> reports = Lists.newArrayList(new ReportsBuilder().setLsp(
 208                                 new LspBuilder().setPlspId(new PlspId(5L)).setSync(false).setRemove(false).setTlvs(
 209                                                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.TlvsBuilder().setSymbolicPathName(
 210                                                                 new SymbolicPathNameBuilder().setPathName(new SymbolicPathName(new byte[] { 22, 34 })).build()).build()).build()).build());
 211                 this.rptmsg = new PcrptBuilder().setPcrptMessage(new PcrptMessageBuilder().setReports(reports).build()).build();
 212         }
 213
 214         @After
 215         public void tearDown() throws InterruptedException, ExecutionException {
 216                 this.manager.close();
 217         }
 218
 219         @Test
 220         public void testUnknownLsp() {
 221                 this.session.sessionUp();
 222                 this.session.handleMessage(this.rptmsg);
 223                 Mockito.verify(this.mockedTransaction, Mockito.times(4)).putOperationalData(Matchers.any(InstanceIdentifier.class),
 224                                 Matchers.any(DataObject.class));
 225                 Mockito.verify(this.mockedTransaction, Mockito.times(3)).commit();
 226         }
 227 }