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.Vteps;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rev160406.transport.zones.transport.zone.VtepsKey;
  30 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  31
  32 public final class ItmTepAutoConfigTestUtil {
  33     private ItmTepAutoConfigTestUtil() {
  34
  35     }
  36
  37     /* transaction methods */
  38     public static ListenableFuture<Void> addTep(String tepIp, String strDpnId, String tzName, boolean ofTunnel,
  39                                                 DataBroker dataBroker, ManagedNewTransactionRunner tx) {
  40         return
  41             OvsdbTepAddConfigHelper.addTepReceivedFromOvsdb(tepIp, strDpnId, tzName, ofTunnel, dataBroker, tx).get(0);
  42     }
  43
  44     public static ListenableFuture<Void> deleteTep(String tepIp, String strDpnId, String tzName,
  45                                                    DataBroker dataBroker, ManagedNewTransactionRunner tx) {
  46         return OvsdbTepRemoveConfigHelper.removeTepReceivedFromOvsdb(tepIp, strDpnId, tzName, dataBroker, tx).get(0);
  47     }
  48
  49     public static CheckedFuture<Void, TransactionCommitFailedException> writeItmConfig(
  50         InstanceIdentifier<ItmConfig> iid, ItmConfig itmConfig, DataBroker dataBroker) {
  51         WriteTransaction tx = dataBroker.newWriteOnlyTransaction();
  52         tx.put(LogicalDatastoreType.CONFIGURATION, iid, itmConfig);
  53         return tx.submit();
  54     }
  55
  56     /* utility methods */
  57     public static InstanceIdentifier<TransportZone> getTzIid(String tzName) {
  58         InstanceIdentifier<TransportZone> tzoneIid =
  59                 InstanceIdentifier.builder(TransportZones.class)
  60                         .child(TransportZone.class, new TransportZoneKey(tzName))
  61                         .build();
  62
  63         return tzoneIid;
  64     }
  65
  66     public static InstanceIdentifier<Vteps> getTepIid(IpPrefix subnetMaskObj, String tzName,
  67         BigInteger dpnId, String portName) {
  68         VtepsKey vtepkey = new VtepsKey(dpnId);
  69
  70         InstanceIdentifier<Vteps> vtepIid = InstanceIdentifier.builder(TransportZones.class)
  71             .child(TransportZone.class, new TransportZoneKey(tzName))
  72             .child(Vteps.class, vtepkey).build();
  73
  74         return vtepIid;
  75     }
  76
  77     public static InstanceIdentifier<TepsInNotHostedTransportZone> getTepNotHostedInTZIid(String tzName) {
  78         InstanceIdentifier<TepsInNotHostedTransportZone> tzonePath =
  79             InstanceIdentifier.builder(NotHostedTransportZones.class)
  80                 .child(TepsInNotHostedTransportZone.class,
  81                     new TepsInNotHostedTransportZoneKey(tzName)).build();
  82
  83         return tzonePath;
  84     }
  85 }