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