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
  81   private void SetupMocks() {
  82     Dpn = BigInteger.valueOf(100000L);
  83     identifier = buildVrfEntryId(testRd, prefix);
  84     vrfEntry = buildVrfEntry(testRd, prefix, nexthop, label);
  85     fibmgr.setMdsalManager(mdsalManager);
  86     fibmgr.setVpnmanager(vpnmanager);
  87     when(vrfTableKey.getRouteDistinguisher()).thenReturn(testRd);
  88   }
  89
  90   @Before
  91   public void setUp() throws Exception {
  92     when(
  93         dataBroker.registerDataChangeListener(any(LogicalDatastoreType.class),
  94             any(InstanceIdentifier.class), any(DataChangeListener.class),
  95             any(DataChangeScope.class))).thenReturn(dataChangeListenerRegistration);
  96     dataChangeEvent = new MockDataChangedEvent();
  97     vrfbuilder = new VrfEntryBuilder();
  98     fibmgr = new FibManager(dataBroker) {
  99
 100       protected VpnInstanceOpDataEntry getVpnInstance(String rd) {
 101         return new VpnInstanceOpDataEntry() {
 102
 103           @Override
 104           public <E extends Augmentation<VpnInstanceOpDataEntry>> E getAugmentation(Class<E> aClass) {
 105             return null;
 106           }
 107
 108           @Override
 109           public Long getVpnId() {
 110             return vpnId;
 111           }
 112
 113           @Override
 114           public String getVrfId() {
 115             return testRd;
 116           }
 117
 118           @Override
 119           public List<VpnToDpnList> getVpnToDpnList() {
 120             List <VpnToDpnList> vpnToDpnLists =  new ArrayList<>();
 121             vpnToDpnLists.add(new VpnToDpnList() {
 122               @Override
 123               public BigInteger getDpnId() {
 124                 return Dpn;
 125               }
 126
 127               @Override
 128               public List<VpnInterfaces> getVpnInterfaces() {
 129                 return null;
 130               }
 131
 132               @Override
 133               public VpnToDpnListKey getKey() {
 134                 return new VpnToDpnListKey(Dpn);
 135               }
 136
 137               @Override
 138               public <E extends Augmentation<VpnToDpnList>> E getAugmentation(
 139                   Class<E> augmentationType) {
 140                 return null;
 141               }
 142
 143               @Override
 144               public Class<? extends DataContainer> getImplementedInterface() {
 145                 return null;
 146               }
 147             });
 148             return vpnToDpnLists;
 149           }
 150
 151           @Override
 152           public VpnInstanceOpDataEntryKey getKey() {
 153             return new VpnInstanceOpDataEntryKey(testRd);
 154           }
 155
 156           @Override
 157           public List<Long> getRouteEntryId() {
 158             return null;
 159           }
 160
 161           @Override
 162           public Class<? extends DataContainer> getImplementedInterface() {
 163             return null;
 164           }
 165         };
 166       }
 167     };
 168     SetupMocks();
 169   }
 170
 171   @Test
 172   public void testAdd() {
 173     dataChangeEvent.created.put(identifier, vrfEntry);
 174     //fibmgr.onDataChanged(dataChangeEvent);
 175     //Mockito.verify(mdsalManager, Mockito.times(2)).installFlow(any(FlowEntity.class));
 176   }
 177
 178   private VrfEntry buildVrfEntry(String rd, String prefix, String nexthop, int label) {
 179     return new VrfEntryBuilder().setDestPrefix(prefix).setNextHopAddress(nexthop)
 180         .setLabel((long) label).build();
 181   }
 182
 183   public static InstanceIdentifier<VrfTables> buildVrfTableId(String rd) {
 184     InstanceIdentifierBuilder<VrfTables> idBuilder =
 185         InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd));
 186     InstanceIdentifier<VrfTables> vrfTableId = idBuilder.build();
 187     return vrfTableId;
 188   }
 189
 190   public static InstanceIdentifier<VrfEntry> buildVrfEntryId(String rd, String prefix) {
 191     InstanceIdentifierBuilder<VrfEntry> idBuilder =
 192         InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd))
 193             .child(VrfEntry.class, new VrfEntryKey(prefix));
 194     InstanceIdentifier<VrfEntry> vrfEntryId = idBuilder.build();
 195     return vrfEntryId;
 196   }
 197 }