networkmodel/src/test/java/org/opendaylight/transportpce/networkmodel/util/WaveLengthServiceUtils.java

   1 /*
   2  * Copyright © 2018 Orange Systems, Inc. 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.transportpce.networkmodel.util;
  10
  11 import java.util.concurrent.ExecutionException;
  12 import org.opendaylight.mdsal.binding.api.WriteTransaction;
  13 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  14 import org.opendaylight.transportpce.common.NetworkUtils;
  15 import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
  16 import org.opendaylight.transportpce.test.AbstractTest;
  17 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.Node1;
  18 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev200529.TerminationPoint1;
  19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
  20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.Networks;
  21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NodeId;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.Network;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.NetworkKey;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.NodeKey;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.TpId;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.node.TerminationPointKey;
  27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  28 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
  29
  30 public final class WaveLengthServiceUtils extends AbstractTest {
  31
  32     private WaveLengthServiceUtils() {
  33
  34     }
  35
  36     private static InstanceIdentifierBuilder<TerminationPoint1> createTerminationPoint1IIDBuilder(
  37             String nodeId, String tpId) {
  38         return InstanceIdentifier
  39             .builder(Networks.class).child(Network.class,
  40                 new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
  41             .child(
  42                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network
  43                 .Node.class, new NodeKey(new NodeId(nodeId)))
  44             .augmentation(
  45                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Node1.class)
  46             .child(
  47                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks
  48                 .network.node.TerminationPoint.class, new TerminationPointKey(new TpId(tpId)))
  49             .augmentation(TerminationPoint1.class);
  50     }
  51
  52     private static InstanceIdentifierBuilder<org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529
  53             .TerminationPoint1> createTerminationPoint2IIDBuilder(
  54                 String nodeId, String tpId) {
  55         return InstanceIdentifier
  56             .builder(Networks.class).child(Network.class,
  57                 new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
  58             .child(
  59                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network
  60                 .Node.class, new NodeKey(new NodeId(nodeId)))
  61             .augmentation(
  62                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Node1.class)
  63             .child(
  64                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks
  65                 .network.node.TerminationPoint.class, new TerminationPointKey(new TpId(tpId)))
  66             .augmentation(org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529
  67                 .TerminationPoint1.class);
  68     }
  69
  70     public static void putTerminationPoint1ToDatastore(String nodeId, String tpId, TerminationPoint1 terminationPoint1,
  71             DeviceTransactionManager deviceTransactionManager) throws ExecutionException, InterruptedException {
  72         WriteTransaction writeTransaction = getDataBroker().newWriteOnlyTransaction();
  73         writeTransaction.put(LogicalDatastoreType.CONFIGURATION,
  74                 createTerminationPoint1IIDBuilder(nodeId, tpId).build(), terminationPoint1);
  75         writeTransaction.commit();
  76     }
  77
  78     public static void putTerminationPoint2ToDatastore(String nodeId, String tpId,
  79             org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.TerminationPoint1
  80                 terminationPoint1)
  81             throws ExecutionException, InterruptedException {
  82         WriteTransaction writeTransaction = getDataBroker().newWriteOnlyTransaction();
  83         writeTransaction.put(LogicalDatastoreType.CONFIGURATION,
  84                 createTerminationPoint2IIDBuilder(nodeId, tpId).build(), terminationPoint1);
  85         writeTransaction.commit();
  86     }
  87
  88     public static TerminationPoint1 getTerminationPoint1FromDatastore(String nodeId, String tpId,
  89             DeviceTransactionManager deviceTransactionManager)
  90             throws ExecutionException, InterruptedException {
  91         InstanceIdentifier<TerminationPoint1> tpIID = createTerminationPoint1IIDBuilder(nodeId, tpId).build();
  92         return getDataBroker().newReadOnlyTransaction().read(LogicalDatastoreType.CONFIGURATION, tpIID).get().get();
  93     }
  94
  95     public static org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529
  96             .TerminationPoint1 getTerminationPoint2FromDatastore(String nodeId, String tpId,
  97             DeviceTransactionManager deviceTransactionManager)
  98             throws ExecutionException, InterruptedException {
  99         InstanceIdentifier<org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.TerminationPoint1>
 100             tpIID = createTerminationPoint2IIDBuilder(nodeId, tpId).build();
 101         return getDataBroker().newReadOnlyTransaction().read(LogicalDatastoreType.CONFIGURATION, tpIID).get().get();
 102     }
 103
 104     private static InstanceIdentifier<Node1> createNode1IID(String nodeId) {
 105         return InstanceIdentifier
 106             .builder(Networks.class)
 107             .child(Network.class, new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
 108             .child(
 109                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network
 110                 .Node.class, new NodeKey(new NodeId(nodeId)))
 111             .augmentation(Node1.class)
 112             .build();
 113     }
 114
 115     private static InstanceIdentifier<org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529
 116         .Node1> createNode2IID(String nodeId) {
 117         return InstanceIdentifier
 118             .builder(Networks.class)
 119             .child(Network.class, new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
 120             .child(
 121                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network
 122                 .Node.class, new NodeKey(new NodeId(nodeId)))
 123             .augmentation(org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.Node1.class)
 124             .build();
 125     }
 126
 127     public static void putNode1ToDatastore(String nodeId, Node1 node1,
 128         DeviceTransactionManager deviceTransactionManager)
 129         throws ExecutionException, InterruptedException {
 130         InstanceIdentifier<Node1> nodeIID = createNode1IID(nodeId);
 131         WriteTransaction writeTransaction = getDataBroker().newWriteOnlyTransaction();
 132         writeTransaction.put(LogicalDatastoreType.CONFIGURATION,nodeIID, node1);
 133         writeTransaction.commit();
 134     }
 135
 136     public static void putNode2ToDatastore(String nodeId,
 137             org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.Node1 node1)
 138            throws ExecutionException, InterruptedException {
 139         InstanceIdentifier<org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529.Node1> nodeIID =
 140             createNode2IID(nodeId);
 141         WriteTransaction writeTransaction = getDataBroker().newWriteOnlyTransaction();
 142         writeTransaction.put(LogicalDatastoreType.CONFIGURATION,
 143                 nodeIID, node1);
 144         writeTransaction.commit();
 145     }
 146
 147     public static Node1 getNode1FromDatastore(String nodeId, DeviceTransactionManager deviceTransactionManager)
 148             throws ExecutionException, InterruptedException {
 149         InstanceIdentifier<Node1> nodeIID = createNode1IID(nodeId);
 150         return getDataBroker().newReadOnlyTransaction().read(LogicalDatastoreType.CONFIGURATION, nodeIID).get().get();
 151     }
 152
 153     public static org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529
 154             .Node1 getNode2FromDatastore(String nodeId, DeviceTransactionManager deviceTransactionManager)
 155             throws ExecutionException, InterruptedException {
 156         InstanceIdentifier<org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev200529
 157             .Node1> nodeIID = createNode2IID(nodeId);
 158         return getDataBroker().newReadOnlyTransaction().read(LogicalDatastoreType.CONFIGURATION, nodeIID).get().get();
 159     }
 160 }