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