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.FluentFuture;
  11 import com.google.common.util.concurrent.ListenableFuture;
  12 import org.eclipse.jdt.annotation.NonNull;
  13 import org.opendaylight.genius.infra.ManagedNewTransactionRunner;
  14 import org.opendaylight.genius.itm.confighelpers.OvsdbTepAddConfigHelper;
  15 import org.opendaylight.genius.itm.confighelpers.OvsdbTepRemoveConfigHelper;
  16 import org.opendaylight.mdsal.binding.api.DataBroker;
  17 import org.opendaylight.mdsal.binding.api.WriteTransaction;
  18 import org.opendaylight.mdsal.common.api.CommitInfo;
  19 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  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 import org.opendaylight.yangtools.yang.common.Uint64;
  32
  33 public final class ItmTepAutoConfigTestUtil {
  34     private ItmTepAutoConfigTestUtil() {
  35
  36     }
  37
  38     /* transaction methods */
  39     public static ListenableFuture<Void> addTep(String tepIp, String strDpnId, String tzName, boolean ofTunnel,
  40                                                 DataBroker dataBroker, ManagedNewTransactionRunner tx)
  41                                                 throws Exception {
  42         return
  43             OvsdbTepAddConfigHelper.addTepReceivedFromOvsdb(tepIp, strDpnId, tzName, ofTunnel, dataBroker, tx).get(0);
  44     }
  45
  46     public static ListenableFuture<Void> deleteTep(String tepIp, String strDpnId, String tzName,
  47                                                    DataBroker dataBroker, ManagedNewTransactionRunner tx) {
  48         return OvsdbTepRemoveConfigHelper.removeTepReceivedFromOvsdb(tepIp, strDpnId, tzName, dataBroker, tx).get(0);
  49     }
  50
  51     public static FluentFuture<? extends @NonNull CommitInfo> writeItmConfig(
  52         InstanceIdentifier<ItmConfig> iid, ItmConfig itmConfig, DataBroker dataBroker) {
  53         WriteTransaction tx = dataBroker.newWriteOnlyTransaction();
  54         tx.put(LogicalDatastoreType.CONFIGURATION, iid, itmConfig);
  55         return tx.commit();
  56     }
  57
  58     /* utility methods */
  59     public static InstanceIdentifier<TransportZone> getTzIid(String tzName) {
  60         InstanceIdentifier<TransportZone> tzoneIid =
  61                 InstanceIdentifier.builder(TransportZones.class)
  62                         .child(TransportZone.class, new TransportZoneKey(tzName))
  63                         .build();
  64
  65         return tzoneIid;
  66     }
  67
  68     public static InstanceIdentifier<Vteps> getTepIid(IpPrefix subnetMaskObj, String tzName,
  69             Uint64 dpnId, String portName) {
  70         VtepsKey vtepkey = new VtepsKey(dpnId);
  71
  72         InstanceIdentifier<Vteps> vtepIid = InstanceIdentifier.builder(TransportZones.class)
  73             .child(TransportZone.class, new TransportZoneKey(tzName))
  74             .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 }