fbeefe7c1aa1cf3efc084af541709a45a0a25fc2
[transportpce.git] / renderer / src / test / java / org / opendaylight / transportpce / renderer / utils / 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.renderer.utils;
10
11 import java.util.concurrent.ExecutionException;
12 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
13 import org.opendaylight.transportpce.common.NetworkUtils;
14 import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
15 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130.Node1;
16 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev181130.TerminationPoint1;
17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.Networks;
19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NodeId;
20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.Network;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.NetworkKey;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.NodeKey;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.TpId;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.node.TerminationPointKey;
25 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
26 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
27
28 public final class WaveLengthServiceUtils {
29
30     private WaveLengthServiceUtils() {
31
32     }
33
34     private static InstanceIdentifierBuilder<TerminationPoint1> createTerminationPoint1IIDBuilder(String nodeId,
35         String tpId) {
36         return InstanceIdentifier
37             .builder(Networks.class).child(Network.class,
38                 new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
39             .child(
40                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network
41                 .Node.class, new NodeKey(new NodeId(nodeId)))
42             .augmentation(
43                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Node1.class)
44             .child(
45                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks
46                 .network.node.TerminationPoint.class, new TerminationPointKey(new TpId(tpId)))
47             .augmentation(TerminationPoint1.class);
48     }
49
50     public static void putTerminationPoint1ToDatastore(String nodeId, String tpId, TerminationPoint1 terminationPoint1,
51         DeviceTransactionManager deviceTransactionManager)
52         throws ExecutionException, InterruptedException {
53         TransactionUtils
54             .writeTransaction(deviceTransactionManager, nodeId, LogicalDatastoreType.CONFIGURATION,
55                 createTerminationPoint1IIDBuilder(nodeId, tpId).build(), terminationPoint1);
56     }
57
58     public static TerminationPoint1 getTerminationPoint1FromDatastore(String nodeId, String tpId,
59         DeviceTransactionManager deviceTransactionManager)
60         throws ExecutionException, InterruptedException {
61         InstanceIdentifier<TerminationPoint1> tpIID = createTerminationPoint1IIDBuilder(nodeId, tpId).build();
62         return (TerminationPoint1) TransactionUtils
63             .readTransaction(deviceTransactionManager, nodeId, LogicalDatastoreType.CONFIGURATION, tpIID);
64     }
65
66     private static InstanceIdentifier<Node1> createNode1IID(String nodeId) {
67         return InstanceIdentifier
68             .builder(Networks.class)
69             .child(Network.class, new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
70             .child(
71                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network
72                 .Node.class, new NodeKey(new NodeId(nodeId)))
73             .augmentation(Node1.class)
74             .build();
75     }
76
77     public static void putNode1ToDatastore(String nodeId, Node1 node1,
78         DeviceTransactionManager deviceTransactionManager)
79         throws ExecutionException, InterruptedException {
80         InstanceIdentifier<Node1> nodeIID = createNode1IID(nodeId);
81         TransactionUtils
82             .writeTransaction(deviceTransactionManager, nodeId,
83                 LogicalDatastoreType.CONFIGURATION, nodeIID, node1);
84     }
85
86     public static Node1 getNode1FromDatastore(String nodeId, DeviceTransactionManager deviceTransactionManager)
87         throws ExecutionException, InterruptedException {
88         InstanceIdentifier<Node1> nodeIID = createNode1IID(nodeId);
89         return (Node1) TransactionUtils
90             .readTransaction(deviceTransactionManager, nodeId, LogicalDatastoreType.CONFIGURATION, nodeIID);
91     }
92 }