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