vpnservice/vpnmanager/vpnmanager-impl/src/test/java/org/opendaylight/netvirt/vpnmanager/test/SubnetOpDpnManagerTest.java

   1 /*
   2  * Copyright (c) 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
   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.netvirt.vpnmanager.test;
  10
  11 import com.google.common.base.Optional;
  12 import com.google.common.util.concurrent.Futures;
  13
  14 import org.junit.Before;
  15 import org.junit.Test;
  16 import org.junit.runner.RunWith;
  17 import org.mockito.Mock;
  18 import org.mockito.runners.MockitoJUnitRunner;
  19 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  20 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  21 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  22 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  23 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker;
  24 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  25 import org.opendaylight.netvirt.vpnmanager.SubnetOpDpnManager;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.PortOpData;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.SubnetOpData;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.port.op.data.PortOpDataEntry;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.port.op.data.PortOpDataEntryBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.port.op.data.PortOpDataEntryKey;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.subnet.op.data.SubnetOpDataEntry;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.subnet.op.data.SubnetOpDataEntryKey;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.subnet.op.data.subnet.op.data.entry.SubnetToDpn;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.subnet.op.data.subnet.op.data.entry.SubnetToDpnBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.subnet.op.data.subnet.op.data.entry.SubnetToDpnKey;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.subnet.op.data.subnet.op.data.entry.subnet.to.dpn.VpnInterfaces;
  38 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  40
  41 import java.math.BigInteger;
  42 import java.util.ArrayList;
  43 import java.util.List;
  44
  45 import static org.mockito.Matchers.any;
  46 import static org.mockito.Mockito.verify;
  47 import static org.mockito.Mockito.doReturn;
  48 import static org.mockito.Mockito.when;
  49
  50
  51
  52 @RunWith(MockitoJUnitRunner.class)
  53 public class SubnetOpDpnManagerTest {
  54
  55     BigInteger dpId = BigInteger.valueOf(1);
  56     Uuid subnetId = Uuid.getDefaultInstance("067e6162-3b6f-4ae2-a171-2470b63dff00");
  57     String infName = "VPN";
  58     SubnetToDpn subnetToDpn = null;
  59     PortOpDataEntry portOp = null;
  60     String portId = "abc";
  61
  62     InstanceIdentifier<SubnetOpDataEntry> subOpIdentifier = InstanceIdentifier.builder(SubnetOpData.class).
  63             child(SubnetOpDataEntry.class, new SubnetOpDataEntryKey(subnetId)).build();
  64     InstanceIdentifier<SubnetToDpn> dpnOpId = subOpIdentifier.child(SubnetToDpn.class, new SubnetToDpnKey(dpId));
  65     InstanceIdentifier<PortOpDataEntry> portOpIdentifier = InstanceIdentifier.builder(PortOpData.class).
  66             child(PortOpDataEntry.class, new PortOpDataEntryKey(infName)).build();
  67
  68     @Mock DataBroker dataBroker;
  69     @Mock ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
  70     @Mock ReadOnlyTransaction mockReadTx;
  71     @Mock WriteTransaction mockWriteTx;
  72
  73     SubnetOpDpnManager subOpDpnManager;
  74
  75     Optional<SubnetToDpn> optionalSubDpn;
  76     Optional<PortOpDataEntry> optionalPortOp;
  77
  78     @Before
  79     public void setUp() throws Exception {
  80         when(dataBroker.registerDataChangeListener(
  81                 any(LogicalDatastoreType.class),
  82                 any(InstanceIdentifier.class),
  83                 any(DataChangeListener.class),
  84                 any(AsyncDataBroker.DataChangeScope.class)))
  85                 .thenReturn(dataChangeListenerRegistration);
  86         setupMocks();
  87
  88         subOpDpnManager = new SubnetOpDpnManager(dataBroker);
  89
  90         optionalSubDpn = Optional.of(subnetToDpn);
  91         optionalPortOp = Optional.of(portOp);
  92
  93         doReturn(Futures.immediateCheckedFuture(optionalPortOp)).when(mockReadTx).read(LogicalDatastoreType
  94                 .OPERATIONAL, portOpIdentifier);
  95         doReturn(Futures.immediateCheckedFuture(optionalSubDpn)).when(mockReadTx).read(LogicalDatastoreType
  96                 .OPERATIONAL, dpnOpId);
  97
  98     }
  99
 100     private void setupMocks() {
 101
 102         List<VpnInterfaces> vpnInterfaces = new ArrayList<>();
 103         subnetToDpn = new SubnetToDpnBuilder().setDpnId(dpId).setKey(new SubnetToDpnKey(dpId)).setVpnInterfaces
 104                 (vpnInterfaces).build();
 105         portOp = new PortOpDataEntryBuilder().setDpnId(dpId).setKey(new PortOpDataEntryKey(infName)).setSubnetId
 106                 (subnetId).setPortId(portId).build();
 107         doReturn(mockReadTx).when(dataBroker).newReadOnlyTransaction();
 108         doReturn(mockWriteTx).when(dataBroker).newWriteOnlyTransaction();
 109         doReturn(Futures.immediateCheckedFuture(null)).when(mockWriteTx).submit();
 110     }
 111
 112     @Test
 113     public void testAddInterfaceToDpn() {
 114
 115         subOpDpnManager.addInterfaceToDpn(subnetId, dpId, infName);
 116
 117         verify(mockWriteTx).put(LogicalDatastoreType.OPERATIONAL, dpnOpId, subnetToDpn, true);
 118
 119     }
 120
 121     @Test
 122     public void testAddPortOpDataEntryPortOpPresent() {
 123
 124         subOpDpnManager.addPortOpDataEntry(infName, subnetId, dpId);
 125
 126         verify(mockWriteTx).put(LogicalDatastoreType.OPERATIONAL, portOpIdentifier, portOp, true);
 127     }
 128
 129     @Test
 130     public void testAddPortOpDataEntryPortOpAbsent() {
 131
 132         doReturn(Futures.immediateCheckedFuture(Optional.absent())).when(mockReadTx).read(LogicalDatastoreType
 133                 .OPERATIONAL, portOpIdentifier);
 134
 135         subOpDpnManager.addPortOpDataEntry(infName, subnetId, dpId);
 136
 137         verify(mockWriteTx).put(LogicalDatastoreType.OPERATIONAL, portOpIdentifier, portOp, true);
 138     }
 139
 140     @Test
 141     public void testRemoveInterfaceFromDpn() {
 142
 143         subOpDpnManager.removeInterfaceFromDpn(subnetId, dpId, infName);
 144
 145         verify(mockWriteTx).delete(LogicalDatastoreType.OPERATIONAL, dpnOpId);
 146     }
 147
 148     @Test
 149     public void testRemovePortOpDataEntryPortOpPresent() {
 150
 151         subOpDpnManager.removePortOpDataEntry(infName);
 152
 153         verify(mockWriteTx).delete(LogicalDatastoreType.OPERATIONAL, portOpIdentifier);
 154     }
 155
 156     @Test
 157     public void testRemovePortOpDataEntryPortOpAbsent() {
 158
 159         doReturn(Futures.immediateCheckedFuture(Optional.absent())).when(mockReadTx).read(LogicalDatastoreType
 160                 .OPERATIONAL, portOpIdentifier);
 161
 162         subOpDpnManager.removePortOpDataEntry(infName);
 163
 164         verify(mockReadTx).read(LogicalDatastoreType.OPERATIONAL, portOpIdentifier);
 165
 166     }
 167
 168     @Test
 169     public void testGetPortOpDataEntryPortOpPresent() {
 170
 171         subOpDpnManager.getPortOpDataEntry(infName);
 172
 173         verify(mockReadTx).read(LogicalDatastoreType.OPERATIONAL, portOpIdentifier);
 174
 175     }
 176
 177     @Test
 178     public void testGetPortOpDataEntryPortOpAbsent() {
 179
 180         doReturn(Futures.immediateCheckedFuture(Optional.absent())).when(mockReadTx).read(LogicalDatastoreType
 181                 .OPERATIONAL, portOpIdentifier);
 182
 183         subOpDpnManager.getPortOpDataEntry(infName);
 184
 185         verify(mockReadTx).read(LogicalDatastoreType.OPERATIONAL, portOpIdentifier);
 186
 187     }
 188
 189 }