Migrate PCE module to Aluminium
[transportpce.git] / pce / src / test / java / org / opendaylight / transportpce / pce / gnpy / GnpyUtilitiesImplTest.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.pce.gnpy;
9
10 import static org.junit.Assert.assertNotNull;
11 import static org.junit.Assert.assertNull;
12 import static org.junit.Assert.fail;
13
14 import com.google.gson.stream.JsonReader;
15 import java.io.FileNotFoundException;
16 import java.io.FileReader;
17 import java.io.IOException;
18 import java.io.Reader;
19 import java.net.URI;
20 import java.util.Map;
21 import java.util.concurrent.ExecutionException;
22 import org.glassfish.grizzly.http.server.HttpServer;
23 import org.glassfish.jersey.grizzly2.httpserver.GrizzlyHttpServerFactory;
24 import org.glassfish.jersey.server.ResourceConfig;
25 import org.junit.AfterClass;
26 import org.junit.Assert;
27 import org.junit.BeforeClass;
28 import org.junit.Test;
29 import org.opendaylight.mdsal.binding.api.WriteTransaction;
30 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
31 import org.opendaylight.transportpce.common.NetworkUtils;
32 import org.opendaylight.transportpce.common.network.NetworkTransactionImpl;
33 import org.opendaylight.transportpce.common.network.RequestProcessor;
34 import org.opendaylight.transportpce.pce.constraints.PceConstraints;
35 import org.opendaylight.transportpce.pce.constraints.PceConstraintsCalc;
36 import org.opendaylight.transportpce.pce.utils.JsonUtil;
37 import org.opendaylight.transportpce.pce.utils.PceTestData;
38 import org.opendaylight.transportpce.test.AbstractTest;
39 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev200629.path.description.AToZDirectionBuilder;
40 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev200629.path.description.ZToADirectionBuilder;
41 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev200629.path.description.atoz.direction.AToZ;
42 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev200629.path.description.atoz.direction.AToZBuilder;
43 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev200629.path.description.atoz.direction.AToZKey;
44 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev200629.pce.resource.Resource;
45 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev200629.pce.resource.ResourceBuilder;
46 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev200629.pce.resource.resource.resource.TerminationPoint;
47 import org.opendaylight.yang.gen.v1.http.org.transportpce.b.c._interface.pathdescription.rev200629.pce.resource.resource.resource.TerminationPointBuilder;
48 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
49 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.Networks;
50 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.Network;
51 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.NetworkKey;
52 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
53 import org.opendaylight.yangtools.yang.common.QName;
54 import org.opendaylight.yangtools.yang.common.Uint32;
55 import org.slf4j.Logger;
56 import org.slf4j.LoggerFactory;
57
58
59 public class GnpyUtilitiesImplTest extends AbstractTest {
60
61     private static final Logger LOG = LoggerFactory.getLogger(GnpyUtilitiesImplTest.class);
62     private GnpyUtilitiesImpl gnpyUtilitiesImpl;
63     private NetworkTransactionImpl networkTransaction;
64     private static HttpServer httpServer;
65
66     public GnpyUtilitiesImplTest() {
67         networkTransaction = new NetworkTransactionImpl(new RequestProcessor(this.getDataBroker()));
68         JsonReader networkReader = null;
69         JsonReader topoReader = null;
70
71         try {
72             // load openroadm-network
73             Reader gnpyNetwork = new FileReader("src/test/resources/gnpy/gnpy_network.json");
74
75             networkReader = new JsonReader(gnpyNetwork);
76             Networks networks = (Networks) JsonUtil.getInstance().getDataObjectFromJson(networkReader,
77                     QName.create("urn:ietf:params:xml:ns:yang:ietf-network", "2018-02-26", "networks"));
78             saveOpenRoadmNetwork(networks.getNetwork().get(0), NetworkUtils.UNDERLAY_NETWORK_ID);
79             // load openroadm-topology
80             Reader gnpyTopo = new FileReader("src/test/resources/gnpy/gnpy_topology.json");
81             topoReader = new JsonReader(gnpyTopo);
82             networks = (Networks) JsonUtil.getInstance().getDataObjectFromJson(topoReader,
83                     QName.create("urn:ietf:params:xml:ns:yang:ietf-network", "2018-02-26", "networks"));
84             saveOpenRoadmNetwork(networks.getNetwork().get(0), NetworkUtils.OVERLAY_NETWORK_ID);
85         } catch (FileNotFoundException | InterruptedException | ExecutionException e) {
86             LOG.error("Cannot init test ", e);
87             fail("Cannot init test ");
88
89         } finally {
90             try {
91                 if (networkReader != null) {
92                     networkReader.close();
93                 }
94                 if (topoReader != null) {
95                     topoReader.close();
96                 }
97             } catch (IOException e) {
98                 LOG.warn("Cannot close reader ", e);
99             }
100         }
101
102     }
103
104     @BeforeClass
105     public static void init() {
106         // here we cannot use JerseyTest as we already extends AbstractTest
107         final ResourceConfig rc = new ResourceConfig(GnpyStub.class);
108         httpServer = GrizzlyHttpServerFactory.createHttpServer(URI.create("http://127.0.0.1:8008"), rc);
109     }
110
111     @AfterClass
112     public static void tearDown() {
113         httpServer.shutdownNow();
114     }
115
116     private void saveOpenRoadmNetwork(Network network, String networkId)
117             throws InterruptedException, ExecutionException {
118         InstanceIdentifier<Network> nwInstanceIdentifier = InstanceIdentifier.builder(Networks.class)
119                 .child(Network.class, new NetworkKey(new NetworkId(networkId))).build();
120         WriteTransaction dataWriteTransaction = this.getDataBroker().newWriteOnlyTransaction();
121         dataWriteTransaction.put(LogicalDatastoreType.CONFIGURATION, nwInstanceIdentifier, network);
122         dataWriteTransaction.commit().get();
123     }
124
125     @Test
126     public void askNewPathFromGnpyNullResultTest() throws Exception {
127         gnpyUtilitiesImpl = new GnpyUtilitiesImpl(networkTransaction,
128                 PceTestData.getGnpyPCERequest("XPONDER-1", "XPONDER-2"));
129         assertNull("No hard constraints should be available", gnpyUtilitiesImpl.askNewPathFromGnpy(null));
130
131     }
132
133     @Test
134     public void askNewPathFromGnpyTest() throws Exception {
135         gnpyUtilitiesImpl = new GnpyUtilitiesImpl(networkTransaction,
136                 PceTestData.getGnpyPCERequest("XPONDER-3", "XPONDER-4"));
137         PceConstraintsCalc constraints = new PceConstraintsCalc(PceTestData.getPCE_simpletopology_test1_request(),
138                 networkTransaction);
139         PceConstraints pceHardConstraints = constraints.getPceHardConstraints();
140         assertNotNull("Hard constraints should be available", gnpyUtilitiesImpl.askNewPathFromGnpy(pceHardConstraints));
141
142
143     }
144
145     @Test
146     public void verifyComputationByGnpyTest() throws Exception {
147         // build AtoZ
148         AToZDirectionBuilder atoZDirectionBldr = buildAtZ();
149         // build ZtoA
150         ZToADirectionBuilder ztoADirectionBldr = buildZtoA();
151
152         gnpyUtilitiesImpl = new GnpyUtilitiesImpl(networkTransaction,
153                 PceTestData.getGnpyPCERequest("XPONDER-1", "XPONDER-2"));
154         PceConstraintsCalc constraints = new PceConstraintsCalc(PceTestData.getPCE_simpletopology_test1_request(),
155                 networkTransaction);
156         PceConstraints pceHardConstraints = constraints.getPceHardConstraints();
157         boolean result = gnpyUtilitiesImpl.verifyComputationByGnpy(atoZDirectionBldr.build(),
158                 ztoADirectionBldr.build(),
159                 pceHardConstraints);
160         Assert.assertFalse("Gnpy Computation should be false",result);
161     }
162
163     private AToZDirectionBuilder buildAtZ() {
164         AToZKey clientKey = new AToZKey("key");
165         TerminationPoint stp = new TerminationPointBuilder()
166                 .setTpId("tpName").setTpNodeId("xname")
167                 .build();
168         Resource clientResource = new ResourceBuilder().setResource(stp).build();
169         AToZ firstResource = new AToZBuilder().setId("tpName").withKey(clientKey).setResource(clientResource).build();
170
171         return new AToZDirectionBuilder()
172                 .setRate(Uint32.valueOf(100))
173                 .setAToZ(Map.of(firstResource.key(),firstResource))
174                 .setAToZWavelengthNumber(Uint32.valueOf(0));
175
176     }
177
178     private ZToADirectionBuilder buildZtoA() {
179         return new ZToADirectionBuilder()
180                 .setRate(Uint32.valueOf(100))
181                 .setZToA(Map.of())
182                 .setZToAWavelengthNumber(Uint32.valueOf(0));
183
184     }
185 }