add unit test for graph package.
[transportpce.git] / pce / src / test / java / org / opendaylight / transportpce / pce / graph / PceGraphEdgeTest.java
1 /*
2  * Copyright © 2020 Orange, 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.pce.graph;
10
11 import org.junit.Assert;
12 import org.junit.Before;
13 import org.junit.Test;
14 import org.opendaylight.transportpce.pce.networkanalyzer.PceLink;
15 import org.opendaylight.transportpce.pce.networkanalyzer.PceOpticalNode;
16 import org.opendaylight.transportpce.pce.utils.NodeUtils;
17 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev181130.OpenroadmNodeType;
18 import org.opendaylight.yang.gen.v1.http.org.openroadm.service.format.rev190531.ServiceFormat;
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.Node;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.Link;
22
23 public class PceGraphEdgeTest {
24
25     private Link link = null;
26     private Node node = null;
27     private PceLink pceLink = null;
28     private PceGraphEdge pceGraphEdge = null;
29
30
31     @Before
32     public void setUp() {
33         // Build Link
34         link = NodeUtils.createRoadmToRoadm("OpenROADM-3-2-DEG1",
35                 "OpenROADM-3-1-DEG1",
36                 "DEG1-TTP-TX", "DEG1-TTP-RX").build();
37
38         //  Link link=genereateLinkBuilder();
39         node =  NodeUtils.getNodeBuilder(NodeUtils.geSupportingNodes()).build();
40
41         PceOpticalNode pceOpticalNode = new PceOpticalNode(node,
42                 OpenroadmNodeType.SRG, new NodeId("OpenROADM-3-2-DEG1"), ServiceFormat.Ethernet,
43                 "DEGREE");
44         PceOpticalNode pceOpticalNode2 = new PceOpticalNode(node,
45                 OpenroadmNodeType.SRG, new NodeId("OpenROADM-3-1-DEG1"), ServiceFormat.Ethernet,
46                 "DEGREE");
47
48         pceLink = new PceLink(link, pceOpticalNode, pceOpticalNode2);
49         pceLink.setClient("XPONDER-CLIENT");
50
51         pceGraphEdge = new PceGraphEdge(pceLink);
52     }
53
54     @Test
55     public void getLink() {
56         Assert.assertEquals(pceGraphEdge.link().getLinkId(),
57                 pceLink.getLinkId());
58     }
59 }