update PceLink test object and NetworkService

[transportpce.git] / pce / src / test / java / org / opendaylight / transportpce / pce / constraints / PceConstraintsCalcTest.java

diff --git a/pce/src/test/java/org/opendaylight/transportpce/pce/constraints/PceConstraintsCalcTest.java b/pce/src/test/java/org/opendaylight/transportpce/pce/constraints/PceConstraintsCalcTest.java
index f3d815266d5f41267cff337018c3f57cbf51e590..b6881c131165dbbe92deb623ec221fffd7dc74d6 100644 (file)
--- a/pce/src/test/java/org/opendaylight/transportpce/pce/constraints/PceConstraintsCalcTest.java
+++ b/pce/src/test/java/org/opendaylight/transportpce/pce/constraints/PceConstraintsCalcTest.java
@@ -10,18 +10,27 @@ package org.opendaylight.transportpce.pce.constraints;
 
 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.utils.PceTestData;
+import org.opendaylight.transportpce.pce.utils.PceTestUtils;
+import org.opendaylight.transportpce.pce.utils.TransactionUtils;
 import org.opendaylight.transportpce.test.AbstractTest;
 
 public class PceConstraintsCalcTest extends AbstractTest {
     private static PceConstraintsCalc pceConstraintsCalc = null;
     private static NetworkTransactionService networkTransactionService = null;
+    private DataStoreContext dataStoreContext = this.getDataStoreContextUtil();
+    private DataBroker dataBroker = this.getDataBroker();
 
     @Before
-    public void setup() {
-        networkTransactionService = Mockito.mock(NetworkTransactionService.class);
+    public void setup() throws Exception {
+        // networkTransactionService = Mockito.mock(NetworkTransactionService.class);
+        networkTransactionService = new NetworkTransactionImpl(new RequestProcessor(dataBroker));
+        PceTestUtils.writeNetworkIntoDataStore(dataBroker, dataStoreContext, TransactionUtils.getNetworkForSpanLoss());
 
     }