improve test cases in PCE contraints

[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..49f241a34960542a8e473660051302ff05a73591 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,31 +10,49 @@ 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);
+        PceTestUtils.writeNetworkIntoDataStore(dataBroker, this.getDataStoreContextUtil(),
+                TransactionUtils.getNetworkForSpanLoss());
+        networkTransactionService = new NetworkTransactionImpl(new RequestProcessor(dataBroker));
+
 
     }
 
     @Test
     public void testNoHardOrSoftConstrainsExists() {
+        PceTestData.getPCE_test2_request_54().getSoftConstraints();
         pceConstraintsCalc = new PceConstraintsCalc(PceTestData
                 .getEmptyPCERequest(), networkTransactionService);
     }
 
-    @Test(expected = NullPointerException.class)
+    @Test()
     public void testHardConstrainsExists() {
         pceConstraintsCalc = new PceConstraintsCalc(PceTestData
-                .getPCE_test2_request_54(), networkTransactionService);
+                .getPCE_simpletopology_test1_requestSetHardAndSoftConstrains(), networkTransactionService);
+    }
+
+    @Test()
+    public void testHardConstrainsExists1() {
+        pceConstraintsCalc = new PceConstraintsCalc(PceTestData
+                .getPathComputationRequestInputWithCoRoutingOrGeneral(), networkTransactionService);
     }
 
     @Test
@@ -43,4 +61,22 @@ public class PceConstraintsCalcTest extends AbstractTest {
                 .getPCERequest(), networkTransactionService);
     }
 
+    @Test(expected = Exception.class)
+    public void testHardConstrainsExists2() {
+        pceConstraintsCalc = new PceConstraintsCalc(PceTestData
+                .build_diversity_from_request(PceTestData.getPCERequest()), networkTransactionService);
+    }
+
+    @Test()
+    public void testHardConstrainsExists3() {
+        pceConstraintsCalc = new PceConstraintsCalc(PceTestData
+                .getEmptyPCERequestServiceNameWithRequestId(), networkTransactionService);
+    }
+
+    @Test(expected = Exception.class)
+    public void testHardConstrainsExists4() {
+        pceConstraintsCalc = new PceConstraintsCalc(PceTestData
+                .getPCE_test2_request_54(), networkTransactionService);
+    }
+
 }