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