Junit tests for network model
[transportpce.git] / networkmodel / src / test / java / org / opendaylight / transportpce / networkmodel / util / OpenRoadmNetworkTest.java
1 /*
2  * Copyright © 2020 Orange Labs, 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 package org.opendaylight.transportpce.networkmodel.util;
9
10
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertNull;
13
14 import org.junit.Ignore;
15 import org.junit.Test;
16 import org.opendaylight.transportpce.common.NetworkUtils;
17 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev200429.network.nodes.NodeInfo;
18 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev200429.network.nodes.NodeInfo.OpenroadmVersion;
19 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev200429.network.nodes.NodeInfoBuilder;
20 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev181130.Node1;
21 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.types.rev181019.NodeTypes;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NodeId;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.Node;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.node.SupportingNode;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.node.SupportingNodeBuilder;
29
30 public class OpenRoadmNetworkTest {
31
32     @Test
33     public void createXpdrNodeTest() {
34         NodeInfo nodeInfo = computeNodeInfo(NodeTypes.Xpdr, "nodeA");
35         Node createdNode = OpenRoadmNetwork.createNode("XPDRA01", nodeInfo);
36         assertEquals("XPDRA01", createdNode.getNodeId().getValue());
37         assertEquals("XPONDER", createdNode.augmentation(Node1.class).getNodeType().getName());
38
39         supportingNodeTest(nodeInfo.getNodeClli(), createdNode);
40     }
41
42     @Test
43     public void createRdmNodeTest() {
44         NodeInfo nodeInfo = computeNodeInfo(NodeTypes.Rdm, "nodeA");
45         Node createdNode = OpenRoadmNetwork.createNode("XPDRA01", nodeInfo);
46         assertEquals("XPDRA01", createdNode.getNodeId().getValue());
47         assertEquals("ROADM", createdNode.augmentation(Node1.class).getNodeType().getName());
48
49         supportingNodeTest(nodeInfo.getNodeClli(), createdNode);
50     }
51
52     @Test
53     public void createNodeWithBadNodeTypeTest() {
54         NodeInfo nodeInfo = computeNodeInfo(NodeTypes.Ila, "nodeA");
55         Node createdNode = OpenRoadmNetwork.createNode("XPDRA01", nodeInfo);
56         assertEquals("XPDRA01", createdNode.getNodeId().getValue());
57         assertNull("NodeType should be ROADM or XPONDER", createdNode.augmentation(Node1.class).getNodeType());
58
59         supportingNodeTest(nodeInfo.getNodeClli(), createdNode);
60     }
61
62     @Ignore
63     @Test
64     public void createNodeWithoutClliTest() {
65         NodeInfo nodeInfo = computeNodeInfo(NodeTypes.Xpdr, null);
66         Node createdNode = OpenRoadmNetwork.createNode("XPDRA01", nodeInfo);
67         assertEquals("XPDRA01", createdNode.getNodeId().getValue());
68         assertEquals("XPONDER", createdNode.augmentation(Node1.class).getNodeType().getName());
69         assertEquals(0, createdNode.getSupportingNode().size());
70     }
71
72     private NodeInfo computeNodeInfo(NodeTypes nodeType, String clli) {
73         return new NodeInfoBuilder()
74             .setNodeModel("NodeModel")
75             .setNodeIpAddress(new IpAddress(new Ipv4Address("127.0.0.1")))
76             .setNodeType(nodeType)
77             .setNodeClli(clli)
78             .setNodeVendor("VendorA")
79             .setOpenroadmVersion(OpenroadmVersion._121)
80             .build();
81     }
82
83     private SupportingNode computeSupportingNode(String clli) {
84         return new SupportingNodeBuilder()
85             .setNetworkRef(new NetworkId(NetworkUtils.CLLI_NETWORK_ID))
86             .setNodeRef(new NodeId(clli))
87             .build();
88     }
89
90     private void supportingNodeTest(String clli, Node createdNode) {
91         SupportingNode supportingNode = computeSupportingNode(clli);
92         assertEquals(1, createdNode.getSupportingNode().size());
93         assertEquals(supportingNode, createdNode.getSupportingNode().get(0));
94     }
95 }