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

   1 /*
   2  * Copyright (c) 2015 Ericsson India Global Services Pvt Ltd. 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
   9 package org.opendaylight.vpnservice.interfacemgr.test;
  10
  11 import com.google.common.base.Optional;
  12 import com.google.common.util.concurrent.Futures;
  13 import org.junit.Before;
  14 import org.junit.Test;
  15 import org.junit.runner.RunWith;
  16 import org.mockito.Mock;
  17 import org.mockito.runners.MockitoJUnitRunner;
  18 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  19 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  20 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  21 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  22 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
  23 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  24 import org.opendaylight.vpnservice.interfacemgr.IfmConstants;
  25 import org.opendaylight.vpnservice.interfacemgr.IfmUtil;
  26 import org.opendaylight.vpnservice.interfacemgr.commons.InterfaceManagerCommonUtils;
  27 import org.opendaylight.vpnservice.interfacemgr.renderer.ovs.statehelpers.OvsInterfaceStateAddHelper;
  28 import org.opendaylight.vpnservice.interfacemgr.renderer.ovs.statehelpers.OvsInterfaceStateRemoveHelper;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceKey;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface.OperStatus;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnector;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.idmanager.rev150403.*;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.IfIndexesInterfaceMap;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007._if.indexes._interface.map.IfIndexInterface;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007._if.indexes._interface.map.IfIndexInterfaceKey;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.meta.rev151007.bridge._interface.info.BridgeEntry;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.interfacemgr.rev150331.TunnelTypeGre;
  41 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  43 import org.opendaylight.yangtools.yang.common.RpcResult;
  44 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  45
  46 import java.math.BigInteger;
  47 import java.util.ArrayList;
  48 import java.util.List;
  49 import java.util.concurrent.Future;
  50
  51 import static org.mockito.Mockito.*;
  52
  53
  54
  55 @RunWith(MockitoJUnitRunner.class)
  56 public class StateInterfaceTest {
  57     BigInteger dpId = BigInteger.valueOf(1);
  58     NodeConnectorId nodeConnectorId = null;
  59     NodeConnector nodeConnector = null;
  60     FlowCapableNodeConnector fcNodeConnectorNew = null;
  61     Interface interfaceEnabled = null;
  62     Interface interfaceDisabled = null;
  63     IfIndexInterface IfindexInterface = null;
  64     InstanceIdentifier<BridgeEntry> bridgeEntryIid = null;
  65     InstanceIdentifier<Interface> interfaceInstanceIdentifier = null;
  66     InstanceIdentifier<FlowCapableNodeConnector> fcNodeConnectorId = null;
  67     InstanceIdentifier<IfIndexInterface> ifIndexId =null;
  68     InstanceIdentifier<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface> interfaceStateIdentifier = null;
  69     org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface stateInterface;
  70
  71     @Mock DataBroker dataBroker;
  72     @Mock IdManagerService idManager;
  73     @Mock ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
  74     @Mock ReadOnlyTransaction mockReadTx;
  75     @Mock WriteTransaction mockWriteTx;
  76
  77     OvsInterfaceStateAddHelper addHelper;
  78     OvsInterfaceStateRemoveHelper removeHelper;
  79
  80     @Before
  81     public void setUp() throws Exception {
  82         when(dataBroker.registerDataChangeListener(
  83                 any(LogicalDatastoreType.class),
  84                 any(InstanceIdentifier.class),
  85                 any(DataChangeListener.class),
  86                 any(DataChangeScope.class)))
  87                 .thenReturn(dataChangeListenerRegistration);
  88         setupMocks();
  89     }
  90
  91     private void setupMocks() {
  92         nodeConnectorId = InterfaceManagerTestUtil.buildNodeConnectorId(BigInteger.valueOf(1), 2);
  93         nodeConnector = InterfaceManagerTestUtil.buildFlowCapableNodeConnector(nodeConnectorId);
  94         fcNodeConnectorNew = nodeConnector.getAugmentation(FlowCapableNodeConnector.class);
  95         fcNodeConnectorId = InterfaceManagerTestUtil.getFlowCapableNodeConnectorIdentifier("openflow:1", nodeConnectorId);
  96         IfindexInterface = InterfaceManagerTestUtil.buildIfIndexInterface(100,InterfaceManagerTestUtil.interfaceName);
  97         ifIndexId = InstanceIdentifier.builder(IfIndexesInterfaceMap.class).child(IfIndexInterface.class, new IfIndexInterfaceKey(100)).build();
  98         interfaceInstanceIdentifier = InterfaceManagerCommonUtils.getInterfaceIdentifier(new InterfaceKey(InterfaceManagerTestUtil.interfaceName));
  99         interfaceStateIdentifier = IfmUtil.buildStateInterfaceId(InterfaceManagerTestUtil.interfaceName);
 100         interfaceEnabled = InterfaceManagerTestUtil.buildTunnelInterface(dpId, InterfaceManagerTestUtil.interfaceName, "Test Interface1", true, TunnelTypeGre.class
 101                 , "192.168.56.101", "192.168.56.102");
 102         interfaceDisabled = InterfaceManagerTestUtil.buildTunnelInterface(dpId, InterfaceManagerTestUtil.interfaceName, "Test Interface1", false, TunnelTypeGre.class
 103                 , "192.168.56.101", "192.168.56.102");
 104         org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.InterfaceBuilder ifaceBuilder = new org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.InterfaceBuilder();
 105         List<String> lowerLayerIfList = new ArrayList<>();
 106         lowerLayerIfList.add(nodeConnectorId.getValue());
 107         ifaceBuilder.setOperStatus(OperStatus.Up).setAdminStatus(org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface.AdminStatus.Up)
 108                 .setPhysAddress(org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.PhysAddress.getDefaultInstance("AA:AA:AA:AA:AA:AA"))
 109                 .setIfIndex(100)
 110                 .setLowerLayerIf(lowerLayerIfList);
 111         ifaceBuilder.setKey(IfmUtil.getStateInterfaceKeyFromName(InterfaceManagerTestUtil.interfaceName));
 112         stateInterface = ifaceBuilder.build();
 113         when(dataBroker.newReadOnlyTransaction()).thenReturn(mockReadTx);
 114         when(dataBroker.newWriteOnlyTransaction()).thenReturn(mockWriteTx);
 115     }
 116
 117     @Test
 118     public void testAddStateInterface() {
 119         Optional<Interface> expectedInterface = Optional.of(interfaceEnabled);
 120         AllocateIdOutput expectedId = new AllocateIdOutputBuilder().setIdValue(Long.valueOf("100")).build();
 121
 122         Future<RpcResult<AllocateIdOutput>> idOutputOptional = RpcResultBuilder.success(expectedId).buildFuture();
 123
 124         doReturn(Futures.immediateCheckedFuture(expectedInterface)).when(mockReadTx).read(
 125                 LogicalDatastoreType.CONFIGURATION, interfaceInstanceIdentifier);
 126         AllocateIdInput getIdInput = new AllocateIdInputBuilder()
 127                 .setPoolName(IfmConstants.IFM_IDPOOL_NAME)
 128                 .setIdKey(InterfaceManagerTestUtil.interfaceName).build();
 129         doReturn(idOutputOptional).when(idManager).allocateId(getIdInput);
 130
 131         addHelper.addState(dataBroker, idManager, nodeConnectorId, InterfaceManagerTestUtil.interfaceName, fcNodeConnectorNew);
 132
 133         //Add some verifications
 134         verify(mockWriteTx).put(LogicalDatastoreType.OPERATIONAL, interfaceStateIdentifier,
 135               stateInterface,true);
 136     }
 137     @Test
 138     public void testDeleteStateInterface() {
 139         Optional<org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.state.Interface>
 140                 expectedStateInterface = Optional.of(stateInterface);
 141         Optional<IfIndexInterface> expectedIfindexInterface = Optional.of(IfindexInterface);
 142         doReturn(Futures.immediateCheckedFuture(expectedIfindexInterface)).when(mockReadTx).read(
 143                 LogicalDatastoreType.OPERATIONAL, ifIndexId);
 144         doReturn(Futures.immediateCheckedFuture(expectedStateInterface)).when(mockReadTx).read(
 145                 LogicalDatastoreType.OPERATIONAL, interfaceStateIdentifier);
 146         doReturn(Futures.immediateCheckedFuture(Optional.absent())).when(mockReadTx).read(
 147                 LogicalDatastoreType.CONFIGURATION, interfaceInstanceIdentifier);
 148
 149         ReleaseIdInput getIdInput = new ReleaseIdInputBuilder()
 150                 .setPoolName(IfmConstants.IFM_IDPOOL_NAME)
 151                 .setIdKey(InterfaceManagerTestUtil.interfaceName).build();
 152
 153         doReturn(Futures.immediateFuture(RpcResultBuilder.<Void>success().build())).when(idManager).releaseId(getIdInput);
 154
 155         removeHelper.removeState(idManager, fcNodeConnectorId, dataBroker, InterfaceManagerTestUtil.interfaceName, fcNodeConnectorNew);
 156
 157         verify(mockWriteTx).delete(LogicalDatastoreType.OPERATIONAL, interfaceStateIdentifier);
 158         verify(mockWriteTx).delete(LogicalDatastoreType.OPERATIONAL, ifIndexId);
 159
 160     }
 161 }