6a40ff16f468d8d4f2389cd2b345b2c36eecaf9f
[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.rev200529.OpenroadmNodeType;
18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.Node;
19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.Link;
20
21 public class PceGraphEdgeTest {
22
23     private Link link = null;
24     private Node node = null;
25     private PceLink pceLink = null;
26     private PceGraphEdge pceGraphEdge = null;
27
28
29     @Before
30     public void setUp() {
31         // Build Link
32         link = NodeUtils.createRoadmToRoadm("OpenROADM-3-2-DEG1",
33                 "OpenROADM-3-1-DEG1",
34                 "DEG1-TTP-TX", "DEG1-TTP-RX").build();
35
36         //  Link link=genereateLinkBuilder();
37         node =  NodeUtils.getNodeBuilder(NodeUtils.geSupportingNodes()).build();
38
39         PceOpticalNode pceOpticalNode = new PceOpticalNode(node, OpenroadmNodeType.SRG);
40         PceOpticalNode pceOpticalNode2 = new PceOpticalNode(node, OpenroadmNodeType.SRG);
41
42         pceLink = new PceLink(link, pceOpticalNode, pceOpticalNode2);
43         pceLink.setClient("XPONDER-CLIENT");
44
45         pceGraphEdge = new PceGraphEdge(pceLink);
46     }
47
48     @Test
49     public void getLink() {
50         Assert.assertEquals(pceGraphEdge.link().getLinkId(),
51                 pceLink.getLinkId());
52     }
53 }