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