Refactor Prov DB handling
[dmaap/datarouter.git] / datarouter-prov / src / test / java / org / onap / dmaap / datarouter / provisioning / beans / EgressRouteTest.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 EgressRouteTest {
39
40     private EgressRoute egressRoute;
41
42     private static EntityManagerFactory emf;
43     private static EntityManager em;
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         egressRoute = new EgressRoute(2, 1);
63     }
64
65     @Test
66     public void Verify_EgressRoute_Is_Removed_Successfully() throws SQLException {
67         Assert.assertEquals(1, EgressRoute.getAllEgressRoutes().size());
68         EgressRoute egressRoute = new EgressRoute(1, 1);
69         try (Connection conn = ProvDbUtils.getInstance().getConnection()) {
70             egressRoute.doDelete(conn);
71         }
72         Assert.assertEquals(0, EgressRoute.getAllEgressRoutes().size());
73     }
74
75     @Test
76     public void Verify_EgressRoute_Is_Updated_Successfully() {
77         EgressRoute egressRoute = new EgressRoute(1, 1);
78         EgressRoute egressRoute1 = new EgressRoute(1, 1);
79         Assert.assertEquals(egressRoute.hashCode(), egressRoute1.hashCode());
80         Assert.assertEquals(egressRoute, egressRoute1);
81         Assert.assertEquals(egressRoute.toString(), egressRoute1.toString());
82     }
83 }