interfacemgr/interfacemgr-impl/src/test/java/org/opendaylight/vpnservice/interfacemgr/test/TopologyStateInterfaceTest.java

   1 /**
   2  * Created by eranjsu on 28-Dec-15.
   3  */
   4 package org.opendaylight.vpnservice.interfacemgr.test;
   5
   6 import com.google.common.base.Optional;
   7 import com.google.common.util.concurrent.Futures;
   8 import org.junit.Before;
   9 import org.junit.Test;
  10 import org.junit.runner.RunWith;
  11 import org.mockito.Mock;
  12 import org.mockito.runners.MockitoJUnitRunner;
  13 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  14 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  15 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  16 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  17 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
  18 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  19 import org.opendaylight.vpnservice.interfacemgr.IfmUtil;
  20 import org.opendaylight.vpnservice.interfacemgr.commons.InterfaceMetaUtils;
  21 import org.opendaylight.vpnservice.interfacemgr.renderer.hwvtep.utilities.SouthboundUtils;
  22 import org.opendaylight.vpnservice.interfacemgr.renderer.ovs.statehelpers.OvsInterfaceTopologyStateAddHelper;
  23 import org.opendaylight.vpnservice.interfacemgr.renderer.ovs.statehelpers.OvsInterfaceTopologyStateRemoveHelper;
  24 import org.opendaylight.vpnservice.interfacemgr.renderer.ovs.statehelpers.OvsInterfaceTopologyStateUpdateHelper;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.InterfaceBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.InterfaceKey;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.*;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.InterfaceBfdStatus;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.port._interface.attributes.InterfaceBfdStatusBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.bridge._interface.info.BridgeEntry;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.bridge._interface.info.BridgeEntryKey;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.bridge._interface.info.bridge.entry.BridgeInterfaceEntry;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.bridge._interface.info.bridge.entry.BridgeInterfaceEntryBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.bridge._interface.info.bridge.entry.BridgeInterfaceEntryKey;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.bridge.ref.info.BridgeRefEntry;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.bridge.ref.info.BridgeRefEntryBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.bridge.ref.info.BridgeRefEntryKey;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rev150331.ParentRefs;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rev150331.TunnelTypeGre;
  41 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  42 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
  43 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  44 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  45
  46 import java.math.BigInteger;
  47 import java.util.ArrayList;
  48 import java.util.List;
  49
  50 import static org.mockito.Mockito.*;
  51
  52
  53 @RunWith(MockitoJUnitRunner.class)
  54 public class TopologyStateInterfaceTest {
  55
  56     BigInteger dpId = BigInteger.valueOf(1);
  57     InstanceIdentifier<OvsdbBridgeAugmentation>bridgeIid = null;
  58     InstanceIdentifier<BridgeEntry> bridgeEntryIid = null;
  59     OvsdbBridgeAugmentation bridgeNew;
  60     OvsdbBridgeAugmentation bridgeOld;
  61     BridgeEntry bridgeEntry = null;
  62     ParentRefs parentRefs = null;
  63     InstanceIdentifier<Interface> interfaceInstanceIdentifier = null;
  64     InstanceIdentifier<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface> interfaceStateIdentifier = null;
  65     Interface tunnelInterfaceEnabled = null;
  66     BridgeInterfaceEntry bridgeInterfaceEntry;
  67     BridgeInterfaceEntryKey bridgeInterfaceEntryKey;
  68
  69     @Mock DataBroker dataBroker;
  70     @Mock ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
  71     @Mock ReadOnlyTransaction mockReadTx;
  72     @Mock WriteTransaction mockWriteTx;
  73     @Mock OvsdbTerminationPointAugmentation newTerminationPoint;
  74
  75     OvsInterfaceTopologyStateAddHelper addHelper;
  76     OvsInterfaceTopologyStateRemoveHelper removeHelper;
  77     OvsInterfaceTopologyStateUpdateHelper updateHelper;
  78
  79     @Before
  80     public void setUp() throws Exception {
  81         when(dataBroker.registerDataChangeListener(
  82                 any(LogicalDatastoreType.class),
  83                 any(InstanceIdentifier.class),
  84                 any(DataChangeListener.class),
  85                 any(DataChangeScope.class)))
  86                 .thenReturn(dataChangeListenerRegistration);
  87         setupMocks();
  88     }
  89
  90     private void setupMocks()
  91     {
  92         Node node = new NodeBuilder().setKey(null).setNodeId(null).build();
  93         tunnelInterfaceEnabled = InterfaceManagerTestUtil.buildTunnelInterface(dpId, InterfaceManagerTestUtil.tunnelInterfaceName, "Test Interface1", true, TunnelTypeGre.class
  94                 , "192.168.56.101", "192.168.56.102");
  95         bridgeIid = InterfaceManagerTestUtil.getOvsdbAugmentationInstanceIdentifier(InterfaceManagerTestUtil.interfaceName, node);
  96         bridgeNew = InterfaceManagerTestUtil.getOvsdbBridgeRef("s1");
  97         bridgeOld = InterfaceManagerTestUtil.getOvsdbBridgeRef("s1");
  98         bridgeEntryIid = InterfaceMetaUtils.getBridgeEntryIdentifier(new BridgeEntryKey(dpId));
  99         interfaceInstanceIdentifier = IfmUtil.buildId(InterfaceManagerTestUtil.tunnelInterfaceName);
 100         interfaceStateIdentifier = IfmUtil.buildStateInterfaceId(newTerminationPoint.getName());
 101         bridgeInterfaceEntryKey = new BridgeInterfaceEntryKey(InterfaceManagerTestUtil.tunnelInterfaceName);
 102         bridgeInterfaceEntry =
 103                 new BridgeInterfaceEntryBuilder().setKey(bridgeInterfaceEntryKey)
 104                         .setInterfaceName(tunnelInterfaceEnabled.getName()).build();
 105         bridgeEntry = InterfaceManagerTestUtil.buildBridgeEntry(dpId, bridgeInterfaceEntry);
 106
 107         when(dataBroker.newReadOnlyTransaction()).thenReturn(mockReadTx);
 108         when(dataBroker.newWriteOnlyTransaction()).thenReturn(mockWriteTx);
 109     }
 110
 111     @Test
 112     public void testAddTopologyStateInterface()
 113     {
 114         Optional<OvsdbBridgeAugmentation>expectedOvsdbBridgeAugmentation = Optional.of(bridgeNew);
 115         Optional<BridgeEntry> expectedBridgeEntry = Optional.of(bridgeEntry);
 116         Optional<Interface> expectedInterface = Optional.of(tunnelInterfaceEnabled);
 117
 118         doReturn(Futures.immediateCheckedFuture(expectedInterface)).when(mockReadTx).read(
 119                 LogicalDatastoreType.CONFIGURATION, interfaceInstanceIdentifier);
 120         doReturn(Futures.immediateCheckedFuture(expectedBridgeEntry)).when(mockReadTx).read(
 121                 LogicalDatastoreType.CONFIGURATION,bridgeEntryIid);
 122
 123         OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentation = new OvsdbBridgeAugmentationBuilder(bridgeNew);
 124         ovsdbBridgeAugmentation.setDatapathId(DatapathId.getDefaultInstance("00:00:00:00:00:00:00:01"));
 125         ovsdbBridgeAugmentation.setBridgeName(OvsdbBridgeName.getDefaultInstance("a"));
 126         bridgeNew = ovsdbBridgeAugmentation.build();
 127
 128         addHelper.addPortToBridge(bridgeIid,bridgeNew,dataBroker);
 129
 130         BigInteger ovsdbDpId = BigInteger.valueOf(1);
 131         BridgeRefEntryKey bridgeRefEntryKey = new BridgeRefEntryKey(ovsdbDpId);
 132         InstanceIdentifier<BridgeRefEntry> bridgeEntryId =
 133                 InterfaceMetaUtils.getBridgeRefEntryIdentifier(bridgeRefEntryKey);
 134         BridgeRefEntryBuilder tunnelDpnBridgeEntryBuilder =
 135                 new BridgeRefEntryBuilder().setKey(bridgeRefEntryKey).setDpid(ovsdbDpId)
 136                         .setBridgeReference(new OvsdbBridgeRef(bridgeIid));
 137
 138         verify(mockWriteTx).put(LogicalDatastoreType.OPERATIONAL, bridgeEntryId, tunnelDpnBridgeEntryBuilder.build(), true);
 139     }
 140
 141     @Test
 142     public void testDeleteTopologyStateInterface()
 143     {
 144         doReturn(Futures.immediateCheckedFuture(Optional.absent())).when(mockReadTx).read(
 145                 LogicalDatastoreType.OPERATIONAL, bridgeIid);
 146         OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentation = new OvsdbBridgeAugmentationBuilder(bridgeOld);
 147         ovsdbBridgeAugmentation.setDatapathId(DatapathId.getDefaultInstance("00:00:00:00:00:00:00:01"));
 148         ovsdbBridgeAugmentation.setBridgeName(OvsdbBridgeName.getDefaultInstance("b"));
 149         bridgeOld = ovsdbBridgeAugmentation.build();
 150
 151         removeHelper.removePortFromBridge(bridgeIid,bridgeOld,dataBroker);
 152
 153         BigInteger ovsdbDpId = BigInteger.valueOf(1);
 154         BridgeRefEntryKey bridgeRefEntryKey = new BridgeRefEntryKey(ovsdbDpId);
 155         InstanceIdentifier<BridgeRefEntry> bridgeEntryId =
 156                 InterfaceMetaUtils.getBridgeRefEntryIdentifier(bridgeRefEntryKey);
 157
 158         verify(mockWriteTx).delete(LogicalDatastoreType.OPERATIONAL,bridgeEntryId);
 159     }
 160
 161     @Test
 162     public void testUpdateBridgeReferenceEntry()
 163     {
 164         Optional<OvsdbBridgeAugmentation>expectedOvsdbBridgeAugmentation = Optional.of(bridgeNew);
 165         Optional<BridgeEntry> expectedBridgeEntry = Optional.of(bridgeEntry);
 166         Optional<Interface> expectedInterface = Optional.of(tunnelInterfaceEnabled);
 167
 168         doReturn(Futures.immediateCheckedFuture(expectedInterface)).when(mockReadTx).read(
 169                 LogicalDatastoreType.CONFIGURATION, interfaceInstanceIdentifier);
 170         doReturn(Futures.immediateCheckedFuture(expectedBridgeEntry)).when(mockReadTx).read(
 171                 LogicalDatastoreType.CONFIGURATION,bridgeEntryIid);
 172         doReturn(Futures.immediateCheckedFuture(expectedOvsdbBridgeAugmentation)).when(mockReadTx).read(
 173                 LogicalDatastoreType.OPERATIONAL, bridgeIid);
 174
 175         OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentation = new OvsdbBridgeAugmentationBuilder(bridgeNew);
 176         ovsdbBridgeAugmentation.setDatapathId(DatapathId.getDefaultInstance("00:00:00:00:00:00:00:01"));
 177         ovsdbBridgeAugmentation.setBridgeName(OvsdbBridgeName.getDefaultInstance("a"));
 178         bridgeNew = ovsdbBridgeAugmentation.build();
 179
 180         updateHelper.updateBridgeRefEntry(bridgeIid, bridgeNew, bridgeOld, dataBroker);
 181     }
 182
 183     @Test
 184     public void testUpdateTunnelState(){
 185
 186         List<InterfaceBfdStatus> interfaceBfdStatus = new ArrayList<InterfaceBfdStatus>();
 187         interfaceBfdStatus.add(new InterfaceBfdStatusBuilder().setBfdStatusKey(SouthboundUtils.BFD_OP_STATE).setBfdStatusValue(SouthboundUtils.BFD_STATE_UP).build());
 188         List bfdStatusSpy = spy(interfaceBfdStatus);
 189         when(newTerminationPoint.getInterfaceBfdStatus()).thenReturn(bfdStatusSpy);
 190
 191         org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface ifState = new InterfaceBuilder().setKey(new InterfaceKey(InterfaceManagerTestUtil.interfaceName)).build();
 192         Optional<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface> expectedInterface = Optional.of(ifState);
 193
 194         doReturn(Futures.immediateCheckedFuture(expectedInterface)).when(mockReadTx).read(
 195                 LogicalDatastoreType.OPERATIONAL, interfaceStateIdentifier);
 196         updateHelper.updateTunnelState(dataBroker, newTerminationPoint, null);
 197
 198         //verify
 199         InstanceIdentifier<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface> ifStateId =
 200                 IfmUtil.buildStateInterfaceId(null);
 201         InterfaceBuilder ifaceBuilder = new InterfaceBuilder().setOperStatus(org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface.OperStatus.Up);
 202         ifaceBuilder.setKey(IfmUtil.getStateInterfaceKeyFromName(null));
 203         verify(mockWriteTx).merge(LogicalDatastoreType.OPERATIONAL, ifStateId, ifaceBuilder.build());
 204
 205     }
 206 }