Refactor Prov DB handling
[dmaap/datarouter.git] / 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.Connection;
24 import java.sql.SQLException;
25 import javax.persistence.EntityManager;
26 import javax.persistence.EntityManagerFactory;
27 import javax.persistence.Persistence;
28 import org.junit.AfterClass;
29 import org.junit.Assert;
30 import org.junit.Before;
31 import org.junit.BeforeClass;
32 import org.junit.Test;
33 import org.junit.runner.RunWith;
34 import org.onap.dmaap.datarouter.provisioning.utils.ProvDbUtils;
35 import org.powermock.modules.junit4.PowerMockRunner;
36
37 @RunWith(PowerMockRunner.class)
38 public class NetworkRouteTest {
39
40     private NetworkRoute networkRoute = new NetworkRoute("node01.","node03.","node02.");
41     private ProvDbUtils provDbUtils = ProvDbUtils.getInstance();
42
43
44     private static EntityManagerFactory emf;
45     private static EntityManager em;
46
47     @BeforeClass
48     public static void init() {
49         emf = Persistence.createEntityManagerFactory("dr-unit-tests");
50         em = emf.createEntityManager();
51         System.setProperty(
52                 "org.onap.dmaap.datarouter.provserver.properties",
53                 "src/test/resources/h2Database.properties");
54     }
55
56     @AfterClass
57     public static void tearDownClass() {
58         em.clear();
59         em.close();
60         emf.close();
61     }
62     @Before
63     public void setUp() throws Exception {
64         try (Connection conn = provDbUtils.getConnection()) {
65             networkRoute.doInsert(conn);
66         }
67     }
68
69     @Test
70     public void Verify_NetworkRoute_Is_Removed_Successfully() throws SQLException {
71         Assert.assertEquals(2, NetworkRoute.getAllNetworkRoutes().size());
72         networkRoute.doDelete(provDbUtils.getConnection());
73         Assert.assertEquals(1, NetworkRoute.getAllNetworkRoutes().size());
74     }
75
76     @Test
77     public void Verify_NetworkRoute_Is_Updated_Successfully() throws SQLException {
78         NetworkRoute networkRoute = new NetworkRoute("stub_from.", "stub_to.", "node02.");
79         networkRoute.doUpdate(provDbUtils.getConnection());
80         for (NetworkRoute net :
81             NetworkRoute.getAllNetworkRoutes()) {
82             Assert.assertEquals(5, net.getVianode());
83         }
84         NetworkRoute networkRoute1 = new NetworkRoute("stub_from.", "stub_to.", "node02.");
85         Assert.assertEquals(networkRoute.hashCode(), networkRoute1.hashCode());
86         Assert.assertEquals(networkRoute, networkRoute1);
87         Assert.assertEquals(networkRoute.toString(), networkRoute1.toString());
88     }
89 }