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, mdsalApiManager);
 234     }
 235
 236     @After
 237     public void cleanUp() {
 238     }
 239
 240     private void setupMocks(){
 241
 242         deviceVteps = new DeviceVtepsBuilder().setIpAddress(ipAddress1).setKey(new DeviceVtepsKey(ipAddress1,"abc"))
 243                 .setNodeId(sourceDevice).setTopologyId(destinationDevice).build();
 244         deviceVtepsList.add(deviceVteps);
 245         stringList.add(sourceDevice);
 246         dpId1List.add(dpId1);
 247         stringList.add("def");
 248         idOutputOptional1 = RpcResultBuilder.success(expectedId1).buildFuture();
 249         getIdInput1 = new AllocateIdInputBuilder()
 250                 .setPoolName(ITMConstants.ITM_IDPOOL_NAME)
 251                 .setIdKey("1:phy0:100:192.168.56.101:192.168.56.101:VXLAN").build();
 252         doReturn(idOutputOptional1).when(idManagerService).allocateId(getIdInput1);
 253         when(idManagerService.releaseId(any(ReleaseIdInput.class))).thenReturn(Futures.immediateFuture(RpcResultBuilder
 254                 .<Void>success().build()));
 255         trunkInterfaceName = ItmUtils.getTrunkInterfaceName(idManagerService, tunnelInterfaceName, ipAddress1
 256                 .getIpv4Address().getValue(), ipAddress1.getIpv4Address().getValue(), tunnelType1.getName());
 257         interfaceIdentifier = ItmUtils.buildId(trunkInterfaceName);
 258         tunnelEndPointsVxlan = new TunnelEndPointsBuilder().setVLANID(vlanId).setPortname(portName1).setIpAddress
 259                 (ipAddress1).setGwIpAddress(gtwyIp1).setInterfaceName(tunnelInterfaceName).setTzMembership
 260                 (ItmUtils.createTransportZoneMembership(transportZone1)).setTunnelType(tunnelType1).setSubnetMask
 261                 (ipPrefixTest).setKey(new TunnelEndPointsKey
 262                 (ipAddress1,portName1,tunnelType1,vlanId)).build();
 263         tunnelEndPointsListVxlan.add(tunnelEndPointsVxlan);
 264         dpntePsInfoVxlan = new DPNTEPsInfoBuilder().setDPNID(dpId1).setKey(new DPNTEPsInfoKey(dpId1)).setUp(true)
 265                 .setTunnelEndPoints(tunnelEndPointsListVxlan).build();
 266         cfgdDpnListVxlan.add(dpntePsInfoVxlan);
 267         dpnEndpoints = new DpnEndpointsBuilder().setDPNTEPsInfo(cfgdDpnListVxlan).build();
 268         externalTunnel= new ExternalTunnelBuilder().setSourceDevice(sourceDevice).setDestinationDevice
 269                 (destinationDevice).setTransportType(tunnelType1).setTunnelInterfaceName(tunnelInterfaceName).setKey(new
 270                 ExternalTunnelKey(destinationDevice,sourceDevice,tunnelType1)).build();
 271         internalTunnel = new InternalTunnelBuilder().setTunnelInterfaceName(tunnelInterfaceName).setDestinationDPN
 272                 (dpId2).setSourceDPN(dpId1).setTransportType(tunnelType1).setKey(new InternalTunnelKey(dpId2,dpId1,
 273                 tunnelType1)).build();
 274         getInternalOrExternalInterfaceNameInput = new GetInternalOrExternalInterfaceNameInputBuilder()
 275                 .setDestinationIp(ipAddress1).setSourceDpid(dpId1).setTunnelType(tunnelType1).build();
 276         addExternalTunnelEndpointInput = new AddExternalTunnelEndpointInputBuilder().setTunnelType(tunnelType1)
 277                 .setDestinationIp(ipAddress1).build();
 278         addL2GwDeviceInput = new AddL2GwDeviceInputBuilder().setIpAddress(ipAddress1).setNodeId(sourceDevice)
 279                 .setTopologyId(destinationDevice).build();
 280         deleteL2GwDeviceInput = new DeleteL2GwDeviceInputBuilder().setIpAddress(ipAddress1).setNodeId(sourceDevice)
 281                 .setTopologyId(destinationDevice).build();
 282         addL2GwMlagDeviceInput = new AddL2GwMlagDeviceInputBuilder().setIpAddress(ipAddress1).setNodeId(stringList)
 283                 .setTopologyId(destinationDevice).build();
 284         deleteL2GwMlagDeviceInput = new DeleteL2GwMlagDeviceInputBuilder().setIpAddress(ipAddress1).setNodeId
 285                 (stringList).setTopologyId(destinationDevice).build();
 286         buildExternalTunnelFromDpnsInput = new BuildExternalTunnelFromDpnsInputBuilder().setTunnelType(tunnelType1)
 287                 .setDestinationIp(ipAddress1).setDpnId(dpId1List).build();
 288         removeExternalTunnelFromDpnsInput = new RemoveExternalTunnelFromDpnsInputBuilder().setTunnelType(tunnelType1)
 289                 .setDestinationIp(ipAddress1).setDpnId(dpId1List).build();
 290         removeExternalTunnelEndpointInput = new RemoveExternalTunnelEndpointInputBuilder().setTunnelType(tunnelType1)
 291                 .setDestinationIp(ipAddress1).build();
 292         removeTerminatingServiceActionsInput = new RemoveTerminatingServiceActionsInputBuilder().setServiceId(vlanId)
 293                 .setDpnId(dpId1).build();
 294         getTunnelInterfaceNameInput = new GetTunnelInterfaceNameInputBuilder().setTunnelType(tunnelType1)
 295                 .setSourceDpid(dpId1).setDestinationDpid(dpId2).build();
 296         getExternalTunnelInterfaceNameInput = new GetExternalTunnelInterfaceNameInputBuilder().setTunnelType
 297                 (tunnelType1).setDestinationNode(destinationDevice).setSourceNode(sourceDevice).build();
 298         iface = ItmUtils.buildTunnelInterface(dpId1,trunkInterfaceName, String.format("%s %s",
 299                 ItmUtils.convertTunnelTypetoString(tunnelType1), "Trunk Interface"),true,tunnelType1,tunnelEndPointsVxlan.getIpAddress()
 300                 ,ipAddress1,gtwyIp1,tunnelEndPointsVxlan.getVLANID(),false,false,monitorProtocol,null, false);
 301         subnetsTest = new SubnetsBuilder().setGatewayIp(gtwyIp1).setVlanId(vlanId).setKey(new SubnetsKey(ipPrefixTest))
 302                 .setDeviceVteps(deviceVtepsList).build();
 303         subnetsList.add(subnetsTest);
 304         transportZone = new TransportZoneBuilder().setZoneName(transportZone1).setTunnelType(tunnelType1).setKey(new
 305                 TransportZoneKey(transportZone1)).setSubnets(subnetsList).build();
 306         transportZoneList.add(transportZone);
 307         transportZones = new TransportZonesBuilder().setTransportZone(transportZoneList).build();
 308         doReturn(mockReadTx).when(dataBroker).newReadOnlyTransaction();
 309         doReturn(mockWriteTx).when(dataBroker).newWriteOnlyTransaction();
 310         doReturn(Futures.immediateCheckedFuture(null)).when(mockWriteTx).submit();
 311     }
 312
 313     @Test
 314     public void testGetInternalOrExternalInterfaceNameExtTunnelPresent(){
 315
 316         itmManagerRpcService.getInternalOrExternalInterfaceName(getInternalOrExternalInterfaceNameInput);
 317
 318         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifier);
 319
 320     }
 321
 322     @Test
 323     public void testGetInternalOrExternalInterfaceNameExtTunnelAbsent(){
 324
 325         doReturn(Futures.immediateCheckedFuture(Optional.absent())).when(mockReadTx).read(LogicalDatastoreType
 326                 .CONFIGURATION,externalTunnelIdentifier);
 327
 328         itmManagerRpcService.getInternalOrExternalInterfaceName(getInternalOrExternalInterfaceNameInput);
 329
 330         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifier);
 331         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,internalTunnelIdentifierNew);
 332
 333     }
 334
 335     @Test
 336     public void testAddExternalTunnelEndpoint(){
 337
 338         externalTunnelNew = ItmUtils.buildExternalTunnel( dpId1.toString(), ipAddress1.toString(),
 339                 tunnelType1, trunkInterfaceName);
 340
 341         itmManagerRpcService.addExternalTunnelEndpoint(addExternalTunnelEndpointInput);
 342
 343         verify(mockWriteTx).merge(LogicalDatastoreType.CONFIGURATION,interfaceIdentifier,iface,true);
 344         verify(mockWriteTx).merge(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifierNew,externalTunnelNew,true);
 345
 346     }
 347
 348     @Test
 349     public void testAddL2GwDevice(){
 350
 351         itmManagerRpcService.addL2GwDevice(addL2GwDeviceInput);
 352
 353         verify(mockWriteTx).put(LogicalDatastoreType.CONFIGURATION, deviceVtepsIdentifier, deviceVteps, true);
 354
 355     }
 356
 357     @Test
 358     public void testAddL2GwMlagDevice(){
 359
 360         itmManagerRpcService.addL2GwMlagDevice(addL2GwMlagDeviceInput);
 361
 362         verify(mockWriteTx).put(LogicalDatastoreType.CONFIGURATION, deviceVtepsIdentifier, deviceVteps, true);
 363
 364     }
 365
 366     @Test
 367     public void testDeleteL2GwDevice(){
 368
 369         itmManagerRpcService.deleteL2GwDevice(deleteL2GwDeviceInput);
 370
 371         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,deviceVtepsIdentifier);
 372
 373     }
 374
 375     @Test
 376     public void testDeleteL2GwMlagDevice(){
 377
 378         itmManagerRpcService.deleteL2GwMlagDevice(deleteL2GwMlagDeviceInput);
 379
 380         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,deviceVtepsIdentifier);
 381
 382     }
 383
 384     @Test
 385     public void testBuildExternalTunnelFromDpns(){
 386
 387         externalTunnelNew = ItmUtils.buildExternalTunnel( dpId1.toString(), ipAddress1.toString(),
 388                 tunnelType1, trunkInterfaceName);
 389
 390         itmManagerRpcService.buildExternalTunnelFromDpns(buildExternalTunnelFromDpnsInput);
 391
 392         verify(mockWriteTx).merge(LogicalDatastoreType.CONFIGURATION,interfaceIdentifier,iface,true);
 393         verify(mockWriteTx).merge(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifierNew,externalTunnelNew,true);
 394
 395     }
 396
 397     @Test
 398     public void testRemoveExternalTunnelFromDpns(){
 399
 400         itmManagerRpcService.removeExternalTunnelFromDpns(removeExternalTunnelFromDpnsInput);
 401
 402         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,interfaceIdentifier);
 403         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifierNew);
 404
 405     }
 406
 407     @Test
 408     public void testRemoveExternalTunnelEndpoint(){
 409
 410         itmManagerRpcService.removeExternalTunnelEndpoint(removeExternalTunnelEndpointInput);
 411
 412         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,interfaceIdentifier);
 413         verify(mockWriteTx).delete(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifierNew);
 414
 415     }
 416
 417     @Test
 418     public void testGetTunnelInterfaceName(){
 419
 420         itmManagerRpcService.getTunnelInterfaceName(getTunnelInterfaceNameInput);
 421
 422         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,internalTunnelIdentifier);
 423
 424     }
 425
 426     @Test
 427     public void testGetExternalTunnelInterfaceName(){
 428
 429         itmManagerRpcService.getExternalTunnelInterfaceName(getExternalTunnelInterfaceNameInput);
 430
 431         verify(mockReadTx).read(LogicalDatastoreType.CONFIGURATION,externalTunnelIdentifier1);
 432
 433     }
 434
 435 }