X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;ds=sidebyside;f=pce%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Ftransportpce%2Fpce%2Fconstraints%2FPceConstraintsCalcTest.java;h=543996d77f2773a440947e4aed47efe018f2c6ea;hb=1e2f9a502de80450411761fd2f636e2b7ee32301;hp=f3d815266d5f41267cff337018c3f57cbf51e590;hpb=e631441244bd5437554d1f85ff4b591ed4a54e7a;p=transportpce.git

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 f3d815266..543996d77 100644
--- 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,37 +10,68 @@ 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.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 DataBroker dataBroker = getDataBroker();
 
+    //TODO: review this test class. May be miss few assert.
     @Before
-    public void setup() {
-        networkTransactionService = Mockito.mock(NetworkTransactionService.class);
-
+    public void setup() throws Exception {
+        // networkTransactionService = Mockito.mock(NetworkTransactionService.class);
+        PceTestUtils.writeNetworkIntoDataStore(dataBroker, getDataStoreContextUtil(),
+                TransactionUtils.getNetworkForSpanLoss());
+        networkTransactionService = new NetworkTransactionImpl(new RequestProcessor(dataBroker));
     }
 
     @Test
     public void testNoHardOrSoftConstrainsExists() {
-        pceConstraintsCalc = new PceConstraintsCalc(PceTestData
-                .getEmptyPCERequest(), networkTransactionService);
+        PceTestData.getPCE_test2_request_54().getSoftConstraints();
+        new PceConstraintsCalc(PceTestData.getEmptyPCERequest(), networkTransactionService);
     }
 
-    @Test(expected = NullPointerException.class)
+    @Test()
     public void testHardConstrainsExists() {
-        pceConstraintsCalc = new PceConstraintsCalc(PceTestData
-                .getPCE_test2_request_54(), networkTransactionService);
+        new PceConstraintsCalc(
+            PceTestData.getPCE_simpletopology_test1_requestSetHardAndSoftConstrains(),
+            networkTransactionService);
+    }
+
+    @Test()
+    public void testHardConstrainsExists1() {
+        new PceConstraintsCalc(
+            PceTestData.getPathComputationRequestInputWithCoRoutingOrGeneral(),
+            networkTransactionService);
     }
 
     @Test
     public void testSoftConstrainsExists() {
-        pceConstraintsCalc = new PceConstraintsCalc(PceTestData
-                .getPCERequest(), networkTransactionService);
+        new PceConstraintsCalc(PceTestData.getPCERequest(), networkTransactionService);
+    }
+
+    @Test(expected = Exception.class)
+    public void testHardConstrainsExists2() {
+        new PceConstraintsCalc(
+            PceTestData.build_diversity_from_request(PceTestData.getPCERequest()),
+            networkTransactionService);
+    }
+
+    @Test()
+    public void testHardConstrainsExists3() {
+        new PceConstraintsCalc(PceTestData.getEmptyPCERequestServiceNameWithRequestId(), networkTransactionService);
+    }
+
+    @Test()
+    public void testHardConstrainsExists4() {
+        new PceConstraintsCalc(PceTestData.getPCE_test2_request_54(), networkTransactionService);
     }
 
 }