itm/itm-impl/src/test/java/org/opendaylight/genius/itm/tests/ItmTepAutoConfigTestUtil.java

   1 /*
   2  * Copyright (c) 2017 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 package org.opendaylight.genius.itm.tests;
   9
  10 import com.google.common.util.concurrent.CheckedFuture;
  11 import com.google.common.util.concurrent.ListenableFuture;
  12 import java.math.BigInteger;
  13 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  14 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  15 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  16 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  17 import org.opendaylight.genius.infra.ManagedNewTransactionRunner;
  18 import org.opendaylight.genius.itm.confighelpers.OvsdbTepAddConfigHelper;
  19 import org.opendaylight.genius.itm.confighelpers.OvsdbTepRemoveConfigHelper;
  20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.config.rev160406.ItmConfig;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.NotHostedTransportZones;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.TransportZones;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.not.hosted.transport.zones.TepsInNotHostedTransportZone;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.not.hosted.transport.zones.TepsInNotHostedTransportZoneKey;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.TransportZone;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.TransportZoneKey;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.Subnets;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.SubnetsKey;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.subnets.Vteps;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.subnets.VtepsKey;
  32 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  33
  34 public final class ItmTepAutoConfigTestUtil {
  35
  36     private ItmTepAutoConfigTestUtil() { }
  37
  38     /* transaction methods */
  39     public static ListenableFuture<Void> addTep(String tepIp, String strDpnId, String tzName, boolean ofTunnel,
  40                                                 DataBroker dataBroker, ManagedNewTransactionRunner tx) {
  41         return
  42             OvsdbTepAddConfigHelper.addTepReceivedFromOvsdb(tepIp, strDpnId, tzName, ofTunnel, dataBroker, tx).get(0);
  43     }
  44
  45     public static ListenableFuture<Void> deleteTep(String tepIp, String strDpnId, String tzName,
  46                                                    DataBroker dataBroker, ManagedNewTransactionRunner tx) {
  47         return OvsdbTepRemoveConfigHelper.removeTepReceivedFromOvsdb(tepIp, strDpnId, tzName, dataBroker, tx).get(0);
  48     }
  49
  50     public static CheckedFuture<Void, TransactionCommitFailedException> writeItmConfig(
  51         InstanceIdentifier<ItmConfig> iid, ItmConfig itmConfig, DataBroker dataBroker) {
  52         WriteTransaction tx = dataBroker.newWriteOnlyTransaction();
  53         tx.put(LogicalDatastoreType.CONFIGURATION, iid, itmConfig);
  54         return tx.submit();
  55     }
  56
  57     /* utility methods */
  58     public static InstanceIdentifier<TransportZone> getTzIid(String tzName) {
  59         InstanceIdentifier<TransportZone> tzoneIid =
  60             InstanceIdentifier.builder(TransportZones.class)
  61                 .child(TransportZone.class, new TransportZoneKey(tzName))
  62                 .build();
  63
  64         return tzoneIid;
  65     }
  66
  67     public static InstanceIdentifier<Vteps> getTepIid(IpPrefix subnetMaskObj, String tzName,
  68         BigInteger dpnId, String portName) {
  69         SubnetsKey subnetsKey = new SubnetsKey(subnetMaskObj);
  70         VtepsKey vtepkey = new VtepsKey(dpnId, portName);
  71
  72         InstanceIdentifier<Vteps> vtepIid = InstanceIdentifier.builder(TransportZones.class)
  73             .child(TransportZone.class, new TransportZoneKey(tzName))
  74             .child(Subnets.class, subnetsKey).child(Vteps.class, vtepkey).build();
  75
  76         return vtepIid;
  77     }
  78
  79     public static InstanceIdentifier<TepsInNotHostedTransportZone> getTepNotHostedInTZIid(String tzName) {
  80         InstanceIdentifier<TepsInNotHostedTransportZone> tzonePath =
  81             InstanceIdentifier.builder(NotHostedTransportZones.class)
  82                 .child(TepsInNotHostedTransportZone.class,
  83                     new TepsInNotHostedTransportZoneKey(tzName)).build();
  84
  85         return tzonePath;
  86     }
  87 }