itm/itm-impl/src/test/java/org/opendaylight/genius/itm/impl/ItmManagerRpcServiceTest.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.genius.itm.impl;
  10
  11 import static org.mockito.Matchers.any;
  12 import static org.mockito.Mockito.doReturn;
  13 import static org.mockito.Mockito.verify;
  14 import static org.mockito.Mockito.when;
  15
  16 import com.google.common.base.Optional;
  17 import com.google.common.util.concurrent.Futures;
  18 import java.math.BigInteger;
  19 import java.util.ArrayList;
  20 import java.util.Collections;
  21 import java.util.List;
  22 import java.util.concurrent.Future;
  23 import org.junit.After;
  24 import org.junit.Before;
  25 import org.junit.Test;
  26 import org.junit.runner.RunWith;
  27 import org.mockito.Mock;
  28 import org.mockito.runners.MockitoJUnitRunner;
  29 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  30 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  31 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  32 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  33 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker;
  34 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  35 import org.opendaylight.genius.itm.globals.ITMConstants;
  36 import org.opendaylight.genius.itm.rpc.ItmManagerRpcService;
  37 import org.opendaylight.genius.mdsalutil.interfaces.IMdsalApiManager;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
  41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefixBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdInput;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdInputBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdOutput;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdOutputBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.IdManagerService;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.ReleaseIdInput;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.TunnelMonitoringTypeBase;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.TunnelTypeBase;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.TunnelTypeMplsOverGre;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rev160406.TunnelTypeVxlan;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.DpnEndpoints;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.DpnEndpointsBuilder;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.ExternalTunnelList;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.TunnelList;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.dpn.endpoints.DPNTEPsInfo;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.dpn.endpoints.DPNTEPsInfoBuilder;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.dpn.endpoints.DPNTEPsInfoKey;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.dpn.endpoints.dpn.teps.info.TunnelEndPoints;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.dpn.endpoints.dpn.teps.info.TunnelEndPointsBuilder;
  62 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.dpn.endpoints.dpn.teps.info.TunnelEndPointsKey;
  63 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.external.tunnel.list.ExternalTunnel;
  64 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.external.tunnel.list.ExternalTunnelBuilder;
  65 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.external.tunnel.list.ExternalTunnelKey;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.tunnel.list.InternalTunnel;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.tunnel.list.InternalTunnelBuilder;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.op.rev160406.tunnel.list.InternalTunnelKey;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.TransportZones;
  70 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.TransportZonesBuilder;
  71 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.TransportZone;
  72 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.TransportZoneBuilder;
  73 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.TransportZoneKey;
  74 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.Subnets;
  75 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.SubnetsBuilder;
  76 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.SubnetsKey;
  77 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.subnets.DeviceVteps;
  78 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.subnets.DeviceVtepsBuilder;
  79 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.subnets.DeviceVtepsKey;
  80 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.AddExternalTunnelEndpointInput;
  81 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.AddExternalTunnelEndpointInputBuilder;
  82 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.AddL2GwDeviceInput;
  83 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.AddL2GwDeviceInputBuilder;
  84 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.AddL2GwMlagDeviceInput;
  85 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.AddL2GwMlagDeviceInputBuilder;
  86 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.BuildExternalTunnelFromDpnsInput;
  87 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.BuildExternalTunnelFromDpnsInputBuilder;
  88 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.DeleteL2GwDeviceInput;
  89 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.DeleteL2GwDeviceInputBuilder;
  90 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.DeleteL2GwMlagDeviceInput;
  91 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.DeleteL2GwMlagDeviceInputBuilder;
  92 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.GetExternalTunnelInterfaceNameInput;
  93 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.GetExternalTunnelInterfaceNameInputBuilder;
  94 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.GetInternalOrExternalInterfaceNameInput;
  95 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.GetInternalOrExternalInterfaceNameInputBuilder;
  96 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.GetTunnelInterfaceNameInput;
  97 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.GetTunnelInterfaceNameInputBuilder;
  98 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.RemoveExternalTunnelEndpointInput;
  99 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.RemoveExternalTunnelEndpointInputBuilder;
 100 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.RemoveExternalTunnelFromDpnsInput;
 101 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.RemoveExternalTunnelFromDpnsInputBuilder;
 102 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.RemoveTerminatingServiceActionsInput;
 103 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.RemoveTerminatingServiceActionsInputBuilder;
 104 import org.opendaylight.yangtools.concepts.ListenerRegistration;
 105 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 106 import org.opendaylight.yangtools.yang.common.RpcResult;
 107 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
 108
 109 @RunWith(MockitoJUnitRunner.class)
 110 public class ItmManagerRpcServiceTest {
 111
 112     int vlanId = 100 ;
 113     String portName1 = "phy0";
 114     String sourceDevice = "abc";
 115     String destinationDevice = "xyz";
 116     String tepIp1 = "192.168.56.101";
 117     String gwyIp1 = "0.0.0.0";
 118     String subnetIp = "10.1.1.24";
 119     String tunnelInterfaceName =  "1:phy0:100" ;
 120     String transportZone1 = "TZA" ;
 121     String trunkInterfaceName = null;
 122     IpAddress ipAddress1 = IpAddressBuilder.getDefaultInstance(tepIp1);
 123     IpAddress gtwyIp1 = IpAddressBuilder.getDefaultInstance(gwyIp1);
 124     IpPrefix ipPrefixTest = IpPrefixBuilder.getDefaultInstance(subnetIp + "/24");
 125     BigInteger dpId1 = BigInteger.valueOf(1);
 126     BigInteger dpId2 = BigInteger.valueOf(2);
 127     ExternalTunnel externalTunnel = null;
 128     ExternalTunnel externalTunnelNew = null;
 129     InternalTunnel internalTunnel = null;
 130     DpnEndpoints dpnEndpoints= null;
 131     DPNTEPsInfo dpntePsInfoVxlan = null;
 132     TunnelEndPoints tunnelEndPointsVxlan = null;
 133     AllocateIdInput getIdInput1 = null;
 134     Interface iface = null;
 135     Subnets subnetsTest = null;
 136     TransportZones transportZones = null;
 137     TransportZone transportZone = null;
 138     DeviceVteps deviceVteps = null;
 139     List<DPNTEPsInfo> cfgdDpnListVxlan = new ArrayList<>() ;
 140     List<TunnelEndPoints> tunnelEndPointsListVxlan = new ArrayList<>();
 141     List<TransportZone> transportZoneList = new ArrayList<>() ;
 142     List<Subnets> subnetsList = new ArrayList<>() ;
 143     List<DeviceVteps> deviceVtepsList = new ArrayList<>();
 144     List<String> stringList = new ArrayList<>();
 145     List<BigInteger> dpId1List = new ArrayList<>();
 146     DeviceVtepsKey deviceVtepKey = new DeviceVtepsKey(ipAddress1, sourceDevice);
 147     AddExternalTunnelEndpointInput addExternalTunnelEndpointInput = null;
 148     GetInternalOrExternalInterfaceNameInput getInternalOrExternalInterfaceNameInput = null;
 149     BuildExternalTunnelFromDpnsInput buildExternalTunnelFromDpnsInput = null;
 150     RemoveExternalTunnelFromDpnsInput removeExternalTunnelFromDpnsInput = null;
 151     RemoveExternalTunnelEndpointInput removeExternalTunnelEndpointInput = null;
 152     RemoveTerminatingServiceActionsInput removeTerminatingServiceActionsInput = null;
 153     GetExternalTunnelInterfaceNameInput getExternalTunnelInterfaceNameInput = null;
 154     AddL2GwDeviceInput addL2GwDeviceInput = null;
 155     DeleteL2GwDeviceInput deleteL2GwDeviceInput = null;
 156     AddL2GwMlagDeviceInput addL2GwMlagDeviceInput = null;
 157     DeleteL2GwMlagDeviceInput deleteL2GwMlagDeviceInput = null;
 158     GetTunnelInterfaceNameInput getTunnelInterfaceNameInput = null;
 159     java.lang.Class<? extends TunnelTypeBase> tunnelType1 = TunnelTypeVxlan.class;
 160     AllocateIdOutput expectedId1 = new AllocateIdOutputBuilder().setIdValue(Long.valueOf("100")).build();
 161     Future<RpcResult<AllocateIdOutput>> idOutputOptional1 ;
 162     Class<? extends TunnelMonitoringTypeBase> monitorProtocol = ITMConstants.DEFAULT_MONITOR_PROTOCOL;
 163
 164     InstanceIdentifier<ExternalTunnel> externalTunnelIdentifier = InstanceIdentifier.create(ExternalTunnelList.class)
 165             .child(ExternalTunnel.class, new ExternalTunnelKey(String.valueOf(ipAddress1), dpId1.toString(),
 166                     TunnelTypeMplsOverGre.class));
 167     InstanceIdentifier<ExternalTunnel> externalTunnelIdentifierNew = InstanceIdentifier.create(ExternalTunnelList.class)
 168             .child(ExternalTunnel.class, new ExternalTunnelKey(String.valueOf(ipAddress1), dpId1.toString(),
 169                     tunnelType1));
 170     InstanceIdentifier<ExternalTunnel> externalTunnelIdentifier1 = InstanceIdentifier.create(ExternalTunnelList.class)
 171             .child(ExternalTunnel.class, new ExternalTunnelKey(destinationDevice, sourceDevice,
 172                     tunnelType1));
 173     InstanceIdentifier<InternalTunnel> internalTunnelIdentifier = InstanceIdentifier.create(TunnelList.class)
 174             .child(InternalTunnel.class, new InternalTunnelKey(dpId2, dpId1, tunnelType1));
 175     InstanceIdentifier<InternalTunnel> internalTunnelIdentifierNew = InstanceIdentifier.create(TunnelList.class)
 176             .child(InternalTunnel.class, new InternalTunnelKey(dpId1, dpId1, tunnelType1));
 177     InstanceIdentifier<DpnEndpoints> dpnEndpointsIdentifier = InstanceIdentifier.builder(DpnEndpoints.class).build();
 178     InstanceIdentifier<Interface> interfaceIdentifier = null;
 179     InstanceIdentifier<TransportZones> transportZonesIdentifier = InstanceIdentifier.create(TransportZones.class);
 180     InstanceIdentifier<DeviceVteps> deviceVtepsIdentifier = InstanceIdentifier.builder(TransportZones.class)
 181             .child(TransportZone.class, new TransportZoneKey(transportZone1))
 182             .child(Subnets.class, new SubnetsKey(ipPrefixTest)).child(DeviceVteps.class, deviceVtepKey).build();
 183
 184
 185     @Mock DataBroker dataBroker;
 186     @Mock ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
 187     @Mock ReadOnlyTransaction mockReadTx;
 188     @Mock WriteTransaction mockWriteTx;
 189     @Mock IdManagerService idManagerService;
 190     @Mock IMdsalApiManager mdsalApiManager;
 191
 192     ItmManagerRpcService itmManagerRpcService ;
 193
 194     Optional<ExternalTunnel> externalTunnelOptional ;
 195     Optional<InternalTunnel> internalTunnelOptional;
 196     Optional<DpnEndpoints> dpnEndpointsOptional ;
 197     Optional<TransportZones> transportZonesOptional ;
 198
 199     @Before
 200     public void setUp() {
 201         when(dataBroker.registerDataChangeListener(
 202                 any(LogicalDatastoreType.class),
 203                 any(InstanceIdentifier.class),
 204                 any(DataChangeListener.class),
 205                 any(AsyncDataBroker.DataChangeScope.class)))
 206                 .thenReturn(dataChangeListenerRegistration);
 207         setupMocks();
 208
 209         externalTunnelOptional = Optional.of(externalTunnel);
 210         internalTunnelOptional = Optional.of(internalTunnel);
 211         dpnEndpointsOptional = Optional.of(dpnEndpoints);
 212         transportZonesOptional = Optional.of(transportZones);
 213
 214         doReturn(Futures.immediateCheckedFuture(externalTunnelOptional)).when(mockReadTx).read(LogicalDatastoreType
 215                 .CONFIGURATION,externalTunnelIdentifier);
 216         doReturn(Futures.immediateCheckedFuture(externalTunnelOptional)).when(mockReadTx).read(LogicalDatastoreType
 217                 .CONFIGURATION,externalTunnelIdentifier1);
 218         doReturn(Futures.immediateCheckedFuture(internalTunnelOptional)).when(mockReadTx).read(LogicalDatastoreType
 219                 .CONFIGURATION,internalTunnelIdentifier);
 220         doReturn(Futures.immediateCheckedFuture(internalTunnelOptional)).when(mockReadTx).read(LogicalDatastoreType
 221                 .CONFIGURATION,internalTunnelIdentifierNew);
 222         doReturn(Futures.immediateCheckedFuture(dpnEndpointsOptional)).when(mockReadTx).read(LogicalDatastoreType
 223                 .CONFIGURATION,dpnEndpointsIdentifier);
 224         doReturn(Futures.immediateCheckedFuture(transportZonesOptional)).when(mockReadTx).read(LogicalDatastoreType
 225                 .CONFIGURATION,transportZonesIdentifier);
 226
 227         itmManagerRpcService = new ItmManagerRpcService(dataBroker,idManagerService, mdsalApiManager);
 228     }
 229
 230     @After
 231     public void cleanUp() {
 232     }
 233
 234     private void setupMocks(){
 235
 236         deviceVteps = new DeviceVtepsBuilder().setIpAddress(ipAddress1).setKey(new DeviceVtepsKey(ipAddress1,"abc"))
 237                 .setNodeId(sourceDevice).setTopologyId(destinationDevice).build();
 238         deviceVtepsList.add(deviceVteps);
 239         stringList.add(sourceDevice);
 240         dpId1List.add(dpId1);
 241         stringList.add("def");
 242         idOutputOptional1 = RpcResultBuilder.success(expectedId1).buildFuture();
 243         getIdInput1 = new AllocateIdInputBuilder()
 244                 .setPoolName(ITMConstants.ITM_IDPOOL_NAME)
 245                 .setIdKey("1:phy0:100:192.168.56.101:192.168.56.101:VXLAN").build();
 246         doReturn(idOutputOptional1).when(idManagerService).allocateId(getIdInput1);
 247         when(idManagerService.releaseId(any(ReleaseIdInput.class))).thenReturn(Futures.immediateFuture(RpcResultBuilder
 248                 .<Void>success().build()));
 249         trunkInterfaceName = ItmUtils.getTrunkInterfaceName(idManagerService, tunnelInterfaceName, ipAddress1
 250                 .getIpv4Address().getValue(), ipAddress1.getIpv4Address().getValue(), tunnelType1.getName());
 251         interfaceIdentifier = ItmUtils.buildId(trunkInterfaceName);
 252         tunnelEndPointsVxlan = new TunnelEndPointsBuilder().setVLANID(vlanId).setPortname(portName1).setIpAddress
 253                 (ipAddress1).setGwIpAddress(gtwyIp1).setInterfaceName(tunnelInterfaceName).setTzMembership
 254                 (ItmUtils.createTransportZoneMembership(transportZone1)).setTunnelType(tunnelType1).setSubnetMask
 255                 (ipPrefixTest).setKey(new TunnelEndPointsKey
 256                 (ipAddress1,portName1,tunnelType1,vlanId)).build();
 257         tunnelEndPointsListVxlan.add(tunnelEndPointsVxlan);
 258         dpntePsInfoVxlan = new DPNTEPsInfoBuilder().setDPNID(dpId1).setKey(new DPNTEPsInfoKey(dpId1)).setUp(true)
 259                 .setTunnelEndPoints(tunnelEndPointsListVxlan).build();
 260         cfgdDpnListVxlan.add(dpntePsInfoVxlan);
 261         dpnEndpoints = new DpnEndpointsBuilder().setDPNTEPsInfo(cfgdDpnListVxlan).build();
 262         externalTunnel= new ExternalTunnelBuilder().setSourceDevice(sourceDevice).setDestinationDevice
 263                 (destinationDevice).setTransportType(tunnelType1).setTunnelInterfaceName(tunnelInterfaceName).setKey(new
 264                 ExternalTunnelKey(destinationDevice,sourceDevice,tunnelType1)).build();
 265         internalTunnel = new InternalTunnelBuilder()
 266                 .setTunnelInterfaceNames(Collections.singletonList(tunnelInterfaceName)).setDestinationDPN(dpId2)
 267                 .setSourceDPN(dpId1).setTransportType(tunnelType1)
 268                 .setKey(new InternalTunnelKey(dpId2, dpId1, tunnelType1)).build();
 269         getInternalOrExternalInterfaceNameInput = new GetInternalOrExternalInterfaceNameInputBuilder()
 270                 .setDestinationIp(ipAddress1).setSourceDpid(dpId1).setTunnelType(tunnelType1).build();
 271         addExternalTunnelEndpointInput = new AddExternalTunnelEndpointInputBuilder().setTunnelType(tunnelType1)
 272                 .setDestinationIp(ipAddress1).build();
 273         addL2GwDeviceInput = new AddL2GwDeviceInputBuilder().setIpAddress(ipAddress1).setNodeId(sourceDevice)
 274                 .setTopologyId(destinationDevice).build();
 275         deleteL2GwDeviceInput = new DeleteL2GwDeviceInputBuilder().setIpAddress(ipAddress1).setNodeId(sourceDevice)
 276                 .setTopologyId(destinationDevice).build();
 277         addL2GwMlagDeviceInput = new AddL2GwMlagDeviceInputBuilder().setIpAddress(ipAddress1).setNodeId(stringList)
 278                 .setTopologyId(destinationDevice).build();
 279         deleteL2GwMlagDeviceInput = new DeleteL2GwMlagDeviceInputBuilder().setIpAddress(ipAddress1).setNodeId
 280                 (stringList).setTopologyId(destinationDevice).build();
 281         buildExternalTunnelFromDpnsInput = new BuildExternalTunnelFromDpnsInputBuilder().setTunnelType(tunnelType1)
 282                 .setDestinationIp(ipAddress1).setDpnId(dpId1List).build();
 283         removeExternalTunnelFromDpnsInput = new RemoveExternalTunnelFromDpnsInputBuilder().setTunnelType(tunnelType1)
 284                 .setDestinationIp(ipAddress1).setDpnId(dpId1List).build();
 285         removeExternalTunnelEndpointInput = new RemoveExternalTunnelEndpointInputBuilder().setTunnelType(tunnelType1)
 286                 .setDestinationIp(ipAddress1).build();
 287         removeTerminatingServiceActionsInput = new RemoveTerminatingServiceActionsInputBuilder().setServiceId(vlanId)
 288                 .setDpnId(dpId1).build();
 289         getTunnelInterfaceNameInput = new GetTunnelInterfaceNameInputBuilder().setTunnelType(tunnelType1)
 290                 .setSourceDpid(dpId1).setDestinationDpid(dpId2).build();
 291         getExternalTunnelInterfaceNameInput = new GetExternalTunnelInterfaceNameInputBuilder().setTunnelType
 292                 (tunnelType1).setDestinationNode(destinationDevice).setSourceNode(sourceDevice).build();
 293         iface = ItmUtils.buildTunnelInterface(dpId1,trunkInterfaceName, String.format("%s %s",
 294                 ItmUtils.convertTunnelTypetoString(tunnelType1), "Trunk Interface"),true,tunnelType1,tunnelEndPointsVxlan.getIpAddress()
 295                 ,ipAddress1,gtwyIp1,tunnelEndPointsVxlan.getVLANID(),false,false,monitorProtocol,null, false);
 296         subnetsTest = new SubnetsBuilder().setGatewayIp(gtwyIp1).setVlanId(vlanId).setKey(new SubnetsKey(ipPrefixTest))
 297                 .setDeviceVteps(deviceVtepsList).build();
 298         subnetsList.add(subnetsTest);
 299         transportZone = new TransportZoneBuilder().setZoneName(transportZone1).setTunnelType(tunnelType1).setKey(new
 300                 TransportZoneKey(transportZone1)).setSubnets(subnetsList).build();
 301         transportZoneList.add(transportZone);
 302         transportZones = new TransportZonesBuilder().setTransportZone(transportZoneList).build();
 303         doReturn(mockReadTx).when(dataBroker).newReadOnlyTransaction();
 304         doReturn(mockWriteTx).when(dataBroker).newWriteOnlyTransaction();
 305         doReturn(Futures.immediateCheckedFuture(null)).when(mockWriteTx).submit();
 306     }
 307
 308     @Test
 309     public void testGetInternalOrExternalInterfaceNameExtTunnelPresent(){
 310
 311         itmManagerRpcService.getInternalOrExternalInterfaceName(getInternalOrExternalInterfaceNameInput);
 312
 313         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifier);
 314
 315     }
 316
 317     @Test
 318     public void testGetInternalOrExternalInterfaceNameExtTunnelAbsent(){
 319
 320         doReturn(Futures.immediateCheckedFuture(Optional.absent())).when(mockReadTx).read(LogicalDatastoreType
 321                 .CONFIGURATION,externalTunnelIdentifier);
 322
 323         itmManagerRpcService.getInternalOrExternalInterfaceName(getInternalOrExternalInterfaceNameInput);
 324
 325         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifier);
 326         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,internalTunnelIdentifierNew);
 327
 328     }
 329
 330     @Test
 331     public void testAddExternalTunnelEndpoint(){
 332
 333         externalTunnelNew = ItmUtils.buildExternalTunnel( dpId1.toString(), ipAddress1.toString(),
 334                 tunnelType1, trunkInterfaceName);
 335
 336         itmManagerRpcService.addExternalTunnelEndpoint(addExternalTunnelEndpointInput);
 337
 338         verify(mockWriteTx).merge(LogicalDatastoreType.CONFIGURATION,interfaceIdentifier,iface,true);
 339         verify(mockWriteTx).merge(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifierNew,externalTunnelNew,true);
 340
 341     }
 342
 343     @Test
 344     public void testAddL2GwDevice(){
 345
 346         itmManagerRpcService.addL2GwDevice(addL2GwDeviceInput);
 347
 348         verify(mockWriteTx).put(LogicalDatastoreType.CONFIGURATION, deviceVtepsIdentifier, deviceVteps, true);
 349
 350     }
 351
 352     @Test
 353     public void testAddL2GwMlagDevice(){
 354
 355         itmManagerRpcService.addL2GwMlagDevice(addL2GwMlagDeviceInput);
 356
 357         verify(mockWriteTx).put(LogicalDatastoreType.CONFIGURATION, deviceVtepsIdentifier, deviceVteps, true);
 358
 359     }
 360
 361     @Test
 362     public void testDeleteL2GwDevice(){
 363
 364         itmManagerRpcService.deleteL2GwDevice(deleteL2GwDeviceInput);
 365
 366         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,deviceVtepsIdentifier);
 367
 368     }
 369
 370     @Test
 371     public void testDeleteL2GwMlagDevice(){
 372
 373         itmManagerRpcService.deleteL2GwMlagDevice(deleteL2GwMlagDeviceInput);
 374
 375         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,deviceVtepsIdentifier);
 376
 377     }
 378
 379     @Test
 380     public void testBuildExternalTunnelFromDpns(){
 381
 382         externalTunnelNew = ItmUtils.buildExternalTunnel( dpId1.toString(), ipAddress1.toString(),
 383                 tunnelType1, trunkInterfaceName);
 384
 385         itmManagerRpcService.buildExternalTunnelFromDpns(buildExternalTunnelFromDpnsInput);
 386
 387         verify(mockWriteTx).merge(LogicalDatastoreType.CONFIGURATION,interfaceIdentifier,iface,true);
 388         verify(mockWriteTx).merge(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifierNew,externalTunnelNew,true);
 389
 390     }
 391
 392     @Test
 393     public void testRemoveExternalTunnelFromDpns(){
 394
 395         itmManagerRpcService.removeExternalTunnelFromDpns(removeExternalTunnelFromDpnsInput);
 396
 397         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,interfaceIdentifier);
 398         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifierNew);
 399
 400     }
 401
 402     @Test
 403     public void testRemoveExternalTunnelEndpoint(){
 404
 405         itmManagerRpcService.removeExternalTunnelEndpoint(removeExternalTunnelEndpointInput);
 406
 407         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,interfaceIdentifier);
 408         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifierNew);
 409
 410     }
 411
 412     @Test
 413     public void testGetTunnelInterfaceName(){
 414
 415         itmManagerRpcService.getTunnelInterfaceName(getTunnelInterfaceNameInput);
 416
 417         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,internalTunnelIdentifier);
 418
 419     }
 420
 421     @Test
 422     public void testGetExternalTunnelInterfaceName(){
 423
 424         itmManagerRpcService.getExternalTunnelInterfaceName(getExternalTunnelInterfaceNameInput);
 425
 426         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifier1);
 427
 428     }
 429
 430 }