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