datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/beans/NetworkRouteTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2019 Nordix Foundation.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.dmaap.datarouter.provisioning.beans;
  22
  23 import java.sql.SQLException;
  24 import javax.persistence.EntityManager;
  25 import javax.persistence.EntityManagerFactory;
  26 import javax.persistence.Persistence;
  27 import org.junit.AfterClass;
  28 import org.junit.Assert;
  29 import org.junit.Before;
  30 import org.junit.BeforeClass;
  31 import org.junit.Test;
  32 import org.junit.runner.RunWith;
  33 import org.onap.dmaap.datarouter.provisioning.utils.DB;
  34 import org.powermock.modules.junit4.PowerMockRunner;
  35
  36 @RunWith(PowerMockRunner.class)
  37 public class NetworkRouteTest {
  38
  39     private NetworkRoute networkRoute;
  40
  41     private static EntityManagerFactory emf;
  42     private static EntityManager em;
  43     private DB db;
  44
  45     @BeforeClass
  46     public static void init() {
  47         emf = Persistence.createEntityManagerFactory("dr-unit-tests");
  48         em = emf.createEntityManager();
  49         System.setProperty(
  50                 "org.onap.dmaap.datarouter.provserver.properties",
  51                 "src/test/resources/h2Database.properties");
  52     }
  53
  54     @AfterClass
  55     public static void tearDownClass() {
  56         em.clear();
  57         em.close();
  58         emf.close();
  59     }
  60     @Before
  61     public void setUp() throws Exception {
  62         db = new DB();
  63         networkRoute = new NetworkRoute("node01.","node03.","node02.");
  64     }
  65
  66     @Test
  67     public void Verify_NetworkRoute_Is_Added_Successfully() throws SQLException {
  68         Assert.assertEquals(1, NetworkRoute.getAllNetworkRoutes().size());
  69         networkRoute.doInsert(db.getConnection());
  70         Assert.assertEquals(2, NetworkRoute.getAllNetworkRoutes().size());
  71         networkRoute.doDelete(db.getConnection());
  72     }
  73
  74     @Test
  75     public void Verify_NetworkRoute_Is_Removed_Successfully() throws SQLException {
  76         Assert.assertEquals(1, NetworkRoute.getAllNetworkRoutes().size());
  77         NetworkRoute networkRoute = new NetworkRoute("stub_from.", "stub_to.");
  78         networkRoute.doDelete(db.getConnection());
  79         Assert.assertEquals(0, NetworkRoute.getAllNetworkRoutes().size());
  80     }
  81
  82     @Test
  83     public void Verify_NetworkRoute_Is_Updated_Successfully() throws SQLException {
  84         NetworkRoute networkRoute = new NetworkRoute("stub_from.", "stub_to.", "node02.");
  85         networkRoute.doUpdate(db.getConnection());
  86         //Assert.assertTrue(NetworkRoute.getAllNetworkRoutes().contains(networkRoute));
  87         for (NetworkRoute net :
  88             NetworkRoute.getAllNetworkRoutes()) {
  89             Assert.assertEquals(5, net.getVianode());
  90         }
  91         NetworkRoute networkRoute1 = new NetworkRoute("stub_from.", "stub_to.", "node02.");
  92         Assert.assertEquals(networkRoute.hashCode(), networkRoute1.hashCode());
  93         Assert.assertEquals(networkRoute, networkRoute1);
  94         Assert.assertEquals(networkRoute.toString(), networkRoute1.toString());
  95     }
  96 }