fibmanager/fibmanager-impl/src/test/java/org/opendaylight/vpnservice/fibmanager/test/FibManagerTest.java

   1 /*
   2  * Copyright (c) 2015 - 2016 Ericsson India Global Services Pvt Ltd. and others. All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the terms of the Eclipse
   5  * Public License v1.0 which accompanies this distribution, and is available at
   6  * http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.vpnservice.fibmanager.test;
   9
  10 import java.math.BigInteger;
  11
  12 import org.junit.Before;
  13 import org.junit.Test;
  14
  15 import static org.mockito.Matchers.any;
  16 import static org.mockito.Mockito.when;
  17
  18 import java.util.ArrayList;
  19 import java.util.List;
  20
  21 import org.junit.runner.RunWith;
  22 import org.mockito.Mock;
  23 import org.mockito.runners.MockitoJUnitRunner;
  24 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  25 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  26 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  27 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  28 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
  29 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  30 import org.opendaylight.vpnmanager.api.IVpnManager;
  31 import org.opendaylight.vpnservice.fibmanager.FibManager;
  32 import org.opendaylight.vpnservice.mdsalutil.interfaces.IMdsalApiManager;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.VpnInstanceOpDataEntry;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.VpnInstanceOpDataEntryKey;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.vpn.instance.op.data.entry.VpnToDpnList;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.vpn.instance.op.data.entry.VpnToDpnListKey;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.vpn.instance.op.data.entry.vpn.to.dpn.list.VpnInterfaces;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.FibEntries;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.fibentries.VrfTables;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.fibentries.VrfTablesKey;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntry;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntryBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntryKey;
  44 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  45 import org.opendaylight.yangtools.yang.binding.Augmentation;
  46 import org.opendaylight.yangtools.yang.binding.DataContainer;
  47 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  48 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
  49
  50 @RunWith(MockitoJUnitRunner.class)
  51 public class FibManagerTest {
  52
  53   @Mock
  54   DataBroker dataBroker;
  55   @Mock
  56   ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
  57   @Mock
  58   ReadOnlyTransaction mockReadTx;
  59   @Mock
  60   WriteTransaction mockWriteTx;
  61   @Mock
  62   IMdsalApiManager mdsalManager;
  63   @Mock
  64   IVpnManager vpnmanager;
  65   @Mock
  66   VrfTablesKey vrfTableKey;
  67
  68   MockDataChangedEvent dataChangeEvent;
  69   FibManager fibmgr;
  70   private static final Long EgressPointer = 11L;
  71   VrfEntry vrfEntry;
  72   InstanceIdentifier<VrfEntry> identifier;
  73   VrfEntryBuilder vrfbuilder;
  74   private static final String testRd = "100:1";
  75   private static final String prefix = "1.1.2.3";
  76   private static final String nexthop = "1.1.1.1";
  77   private static final int label = 10;
  78   BigInteger Dpn;
  79   private static final long vpnId = 101L;
  80   private static final long vpnIntfCnt = 2;
  81
  82   private void SetupMocks() {
  83     Dpn = BigInteger.valueOf(100000L);
  84     identifier = buildVrfEntryId(testRd, prefix);
  85     vrfEntry = buildVrfEntry(testRd, prefix, nexthop, label);
  86     fibmgr.setMdsalManager(mdsalManager);
  87     fibmgr.setVpnmanager(vpnmanager);
  88     when(vrfTableKey.getRouteDistinguisher()).thenReturn(testRd);
  89   }
  90
  91   @Before
  92   public void setUp() throws Exception {
  93     when(
  94         dataBroker.registerDataChangeListener(any(LogicalDatastoreType.class),
  95             any(InstanceIdentifier.class), any(DataChangeListener.class),
  96             any(DataChangeScope.class))).thenReturn(dataChangeListenerRegistration);
  97     dataChangeEvent = new MockDataChangedEvent();
  98     vrfbuilder = new VrfEntryBuilder();
  99     fibmgr = new FibManager(dataBroker) {
 100
 101       protected VpnInstanceOpDataEntry getVpnInstance(String rd) {
 102         return new VpnInstanceOpDataEntry() {
 103
 104           @Override
 105           public <E extends Augmentation<VpnInstanceOpDataEntry>> E getAugmentation(Class<E> aClass) {
 106             return null;
 107           }
 108
 109           @Override
 110           public Long getVpnId() {
 111             return vpnId;
 112           }
 113
 114           @Override
 115           public String getVrfId() {
 116             return testRd;
 117           }
 118
 119           @Override
 120           public Long getVpnInterfaceCount() { return vpnIntfCnt; }
 121
 122           @Override
 123           public List<VpnToDpnList> getVpnToDpnList() {
 124             List <VpnToDpnList> vpnToDpnLists =  new ArrayList<>();
 125             vpnToDpnLists.add(new VpnToDpnList() {
 126               @Override
 127               public BigInteger getDpnId() {
 128                 return Dpn;
 129               }
 130
 131               @Override
 132               public List<VpnInterfaces> getVpnInterfaces() {
 133                 return null;
 134               }
 135
 136               @Override
 137               public VpnToDpnListKey getKey() {
 138                 return new VpnToDpnListKey(Dpn);
 139               }
 140
 141               @Override
 142               public <E extends Augmentation<VpnToDpnList>> E getAugmentation(
 143                   Class<E> augmentationType) {
 144                 return null;
 145               }
 146
 147               @Override
 148               public Class<? extends DataContainer> getImplementedInterface() {
 149                 return null;
 150               }
 151             });
 152             return vpnToDpnLists;
 153           }
 154
 155           @Override
 156           public VpnInstanceOpDataEntryKey getKey() {
 157             return new VpnInstanceOpDataEntryKey(testRd);
 158           }
 159
 160           @Override
 161           public List<Long> getRouteEntryId() {
 162             return null;
 163           }
 164
 165           @Override
 166           public Class<? extends DataContainer> getImplementedInterface() {
 167             return null;
 168           }
 169         };
 170       }
 171     };
 172     SetupMocks();
 173   }
 174
 175   @Test
 176   public void testAdd() {
 177     dataChangeEvent.created.put(identifier, vrfEntry);
 178     //fibmgr.onDataChanged(dataChangeEvent);
 179     //Mockito.verify(mdsalManager, Mockito.times(2)).installFlow(any(FlowEntity.class));
 180   }
 181
 182   private VrfEntry buildVrfEntry(String rd, String prefix, String nexthop, int label) {
 183     return new VrfEntryBuilder().setDestPrefix(prefix).setNextHopAddress(nexthop)
 184         .setLabel((long) label).build();
 185   }
 186
 187   public static InstanceIdentifier<VrfTables> buildVrfTableId(String rd) {
 188     InstanceIdentifierBuilder<VrfTables> idBuilder =
 189         InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd));
 190     InstanceIdentifier<VrfTables> vrfTableId = idBuilder.build();
 191     return vrfTableId;
 192   }
 193
 194   public static InstanceIdentifier<VrfEntry> buildVrfEntryId(String rd, String prefix) {
 195     InstanceIdentifierBuilder<VrfEntry> idBuilder =
 196         InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd))
 197             .child(VrfEntry.class, new VrfEntryKey(prefix));
 198     InstanceIdentifier<VrfEntry> vrfEntryId = idBuilder.build();
 199     return vrfEntryId;
 200   }
 201 }