add unit test for graph package.
[transportpce.git] / pce / src / test / java / org / opendaylight / transportpce / pce / graph / PceGraphTest.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 java.util.Map;
12 import org.junit.Assert;
13 import org.junit.Before;
14 import org.junit.Test;
15 import org.opendaylight.transportpce.pce.constraints.PceConstraints;
16 import org.opendaylight.transportpce.pce.networkanalyzer.PceLink;
17 import org.opendaylight.transportpce.pce.networkanalyzer.PceNode;
18 import org.opendaylight.transportpce.pce.networkanalyzer.PceOpticalNode;
19 import org.opendaylight.transportpce.pce.networkanalyzer.PceResult;
20 import org.opendaylight.transportpce.pce.utils.NodeUtils;
21 import org.opendaylight.transportpce.pce.utils.PceTestData;
22 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.pce.rev200128.PathComputationRequestInput;
23 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev181130.OpenroadmNodeType;
24 import org.opendaylight.yang.gen.v1.http.org.openroadm.service.format.rev190531.ServiceFormat;
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.topology.rev180226.networks.network.Link;
28
29
30 public class PceGraphTest {
31
32     private Link link = null;
33     private Node node = null;
34     private PceLink pceLink = null;
35     private PceGraph pceGraph = null;
36     private PathComputationRequestInput requestInput = PceTestData.getPCE_test2_request_54();
37     private PceConstraints pceHardConstraints = null;
38     private PceResult rc = null;
39     private PceOpticalNode pceOpticalNode = null;
40     private PceOpticalNode pceOpticalNode2 = null;
41     private Map<NodeId, PceNode> allPceNodes = null;
42
43     @Before
44     public void setUp() {
45         // Build Link
46         link = NodeUtils.createRoadmToRoadm("OpenROADM-3-2-DEG1",
47                 "OpenROADM-3-1-DEG1",
48                 "DEG1-TTP-TX", "DEG1-TTP-RX").build();
49
50         node =  NodeUtils.getNodeBuilder(NodeUtils.geSupportingNodes()).build();
51
52         pceOpticalNode = new PceOpticalNode(node,
53                 OpenroadmNodeType.SRG, new NodeId("OpenROADM-3-2-DEG1"), ServiceFormat.Ethernet,
54                 "DEGREE");
55
56         pceOpticalNode2 = new PceOpticalNode(node,
57                 OpenroadmNodeType.SRG, new NodeId("OpenROADM-3-1-DEG1"), ServiceFormat.Ethernet,
58                 "DEGREE");
59
60         pceLink = new PceLink(link, pceOpticalNode, pceOpticalNode2);
61         pceLink.setClient("XPONDER-CLIENT");
62
63         pceLink.getDestId();
64         pceOpticalNode.addOutgoingLink(pceLink);
65
66         // init PceHardContraints
67         pceHardConstraints = new PceConstraints();
68         // pceHardConstraints.setp
69         allPceNodes = Map.of(new NodeId("OpenROADM-3-2-DEG1"), pceOpticalNode,
70                 new NodeId("OpenROADM-3-1-DEG1"), pceOpticalNode2);
71         rc = new PceResult();
72         pceGraph = new PceGraph(pceOpticalNode, pceOpticalNode2, allPceNodes,
73                 pceHardConstraints,
74                 null, rc,
75                 "ODU4");
76     }
77
78     @Test
79     public void clacPath() {
80
81         Assert.assertEquals(pceGraph.calcPath(), true);
82     }
83
84     @Test
85     public void clacPathFalse() {
86         pceGraph = new PceGraph(pceOpticalNode, pceOpticalNode2, allPceNodes,
87                 pceHardConstraints,
88                 null, rc,
89                 "100GE");
90
91         Assert.assertEquals(pceGraph.calcPath(), false);
92     }
93
94     @Test(expected = Exception.class)
95     public void clacPath1GE() {
96         pceGraph = new PceGraph(pceOpticalNode, pceOpticalNode2, allPceNodes,
97                 pceHardConstraints,
98                 null, rc,
99                 "10GE");
100
101         Assert.assertEquals(pceGraph.calcPath(), false);
102     }
103 }
104