update PceLink test object and NetworkService

[transportpce.git] / pce / src / test / java / org / opendaylight / transportpce / pce / service / PathComputationServiceImplTest.java

diff --git a/pce/src/test/java/org/opendaylight/transportpce/pce/service/PathComputationServiceImplTest.java b/pce/src/test/java/org/opendaylight/transportpce/pce/service/PathComputationServiceImplTest.java
index 71d57e075c300af6cb6dba39fe8aba2c4d765dcf..62de9fb3fe171542a1833021a9ba114934e93a83 100644 (file)
--- a/pce/src/test/java/org/opendaylight/transportpce/pce/service/PathComputationServiceImplTest.java
+++ b/pce/src/test/java/org/opendaylight/transportpce/pce/service/PathComputationServiceImplTest.java
@@ -5,32 +5,65 @@
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
  * and is available at http://www.eclipse.org/legal/epl-v10.html
  */
-package org.opendaylight.transportpce.pce.utils.service;
+package org.opendaylight.transportpce.pce.service;
 
+import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
+import org.mockito.Mockito;
+import org.opendaylight.mdsal.binding.api.DataBroker;
+import org.opendaylight.transportpce.common.DataStoreContext;
 import org.opendaylight.transportpce.common.network.NetworkTransactionImpl;
+import org.opendaylight.transportpce.common.network.NetworkTransactionService;
 import org.opendaylight.transportpce.common.network.RequestProcessor;
-import org.opendaylight.transportpce.pce.service.PathComputationServiceImpl;
+import org.opendaylight.transportpce.pce.gnpy.GnpyResult;
+import org.opendaylight.transportpce.pce.gnpy.GnpyTopoImpl;
 import org.opendaylight.transportpce.pce.utils.PceTestData;
+import org.opendaylight.transportpce.pce.utils.PceTestUtils;
+import org.opendaylight.transportpce.pce.utils.TransactionUtils;
 import org.opendaylight.transportpce.test.AbstractTest;
 
 public class PathComputationServiceImplTest extends AbstractTest {
 
     private PathComputationServiceImpl pathComputationServiceImpl;
+    private static NetworkTransactionService networkTransactionService = null;
+    private static GnpyTopoImpl gnpyTopoImpl = null;
+    private static GnpyResult gnpyResult = null;
+    private DataStoreContext dataStoreContext = this.getDataStoreContextUtil();
+    private DataBroker dataBroker = this.getDataBroker();
 
     @Before
     public void setUp() {
+        networkTransactionService = Mockito.mock(NetworkTransactionService.class);
+        gnpyTopoImpl = Mockito.mock(GnpyTopoImpl.class);
+        gnpyResult = Mockito.mock(GnpyResult.class);
         pathComputationServiceImpl = new PathComputationServiceImpl(
-                new NetworkTransactionImpl(new RequestProcessor(this.getDataBroker())),
+                networkTransactionService,
                 this.getNotificationPublishService());
         pathComputationServiceImpl.init();
     }
 
     @Test
     public void pathComputationRequestTest() {
-        Assert.assertNotNull(pathComputationServiceImpl.pathComputationRequest(PceTestData.getPCERequest()));
+        Assert.assertNotNull(
+                pathComputationServiceImpl.pathComputationRequest(PceTestData.getPCE_simpletopology_test1_request()));
+
+    }
+
+    @Test(expected = Exception.class)
+    public void generateGnpyResponse() throws Exception {
+        PceTestUtils.writeNetworkIntoDataStore(dataBroker, dataStoreContext, TransactionUtils.getNetworkForSpanLoss());
+        GnpyResult gnpyResult2 =
+                new GnpyResult("A-to-Z",
+                        new GnpyTopoImpl(new NetworkTransactionImpl(
+                                new RequestProcessor(dataBroker))));
+        pathComputationServiceImpl.generateGnpyResponse(gnpyResult2.getResponse(), "A-to-Z");
+    }
+
+
+    @After
+    public void destroy() {
         pathComputationServiceImpl.close();
     }
 }