2 * ============LICENSE_START=======================================================
3 * ONAP : ccsdk features
4 * ================================================================================
5 * Copyright (C) 2021 highstreet technologies GmbH Intellectual Property.
7 * ================================================================================
8 * Update Copyright (C) 2021 Samsung Electronics Intellectual Property. All rights reserved.
9 * ================================================================================================
10 * Licensed under the Apache License, Version 2.0 (the "License");
11 * you may not use this file except in compliance with the License.
12 * You may obtain a copy of the License at
14 * http://www.apache.org/licenses/LICENSE-2.0
16 * Unless required by applicable law or agreed to in writing, software
17 * distributed under the License is distributed on an "AS IS" BASIS,
18 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
19 * See the License for the specific language governing permissions and
20 * limitations under the License.
21 * ============LICENSE_END=========================================================
24 package org.onap.ccsdk.features.sdnr.wt.dataprovider.setup;
26 import static org.junit.Assert.fail;
28 import java.sql.SQLException;
29 import java.util.HashMap;
31 import java.util.Random;
32 import org.onap.ccsdk.features.sdnr.wt.common.configuration.ConfigurationFileRepresentation;
33 import org.onap.ccsdk.features.sdnr.wt.common.configuration.subtypes.Section;
34 import org.onap.ccsdk.features.sdnr.wt.common.configuration.subtypes.Section.EnvGetter;
35 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.SqlDBClient;
36 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.SqlDBConfig;
37 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.data.SqlDBDataProvider;
38 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.database.SqlDBMapper;
39 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.database.SqlDBMapper.UnableToMapClassException;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.CmlogEntity;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.ConnectionlogEntity;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Entity;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.EventlogEntity;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.FaultcurrentEntity;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.FaultlogEntity;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.GuicutthroughEntity;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.InventoryEntity;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.MaintenanceEntity;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.MediatorServerEntity;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.NetworkElementConnectionEntity;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Pmdata15mEntity;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Pmdata24hEntity;
53 import ch.vorburger.exec.ManagedProcessException;
54 import ch.vorburger.mariadb4j.DB;
55 import ch.vorburger.mariadb4j.DBConfigurationBuilder;
57 public class MariaDBTestBase {
59 private final SqlDBDataProvider dbProvider;
61 private static final Map<String, String> envDefaultValues = initEnvDefaultValues();
62 private static final String SDNRDBDATABASETEST="test";
63 private static final String TESTPROPFILE = "test.properties";
65 public MariaDBTestBase() throws ManagedProcessException {
66 this(new Random().nextInt(1000) + 50000);
68 private static String dbUrl(String host, int port, String dbName) {
69 return String.format("jdbc:mysql://%s:%d/%s", host,port,dbName);
71 private static Map<String, String> initEnvDefaultValues() {
72 Map<String, String> defaults = new HashMap<>();
73 defaults.put("SDNRDBURL", dbUrl("localhost",3306,SDNRDBDATABASETEST));
74 defaults.put("SDNRDBDATABASE", "test");
78 public MariaDBTestBase(String host, int port) {
79 this(host, port, SDNRDBDATABASETEST );
81 public MariaDBTestBase(String host, int port, String dbName) {
82 EnvGetter env = Section.getEnvGetter();
83 envDefaultValues.put("SDNRDBURL", dbUrl(host,port,dbName));
84 envDefaultValues.put("SDNRDBUSERNAME", "sdnrdb");
85 envDefaultValues.put("SDNRDBPASSWORD", "sdnrdb");
86 Section.setEnvGetter((envname) -> {
87 return envDefaultValues.getOrDefault(envname, env.getenv(envname));
89 SqlDBConfig config = new SqlDBConfig(new ConfigurationFileRepresentation(TESTPROPFILE));
90 config.setControllerId("test123");
92 this.dbProvider = new SqlDBDataProvider(config, false);
93 //testCreateTable(this.dbProvider.getDBService());
96 public MariaDBTestBase(int port) throws ManagedProcessException {
97 EnvGetter env = Section.getEnvGetter();
98 envDefaultValues.put("SDNRDBURL",dbUrl("localhost",port,SDNRDBDATABASETEST));
99 Section.setEnvGetter((envname) -> {
100 return envDefaultValues.getOrDefault(envname, env.getenv(envname));
102 SqlDBConfig config = new SqlDBConfig(new ConfigurationFileRepresentation(TESTPROPFILE));
104 this.db = startDatabase(port);
105 //create db with name sdnrdb
106 this.dbProvider = new SqlDBDataProvider(config, false);
107 //testCreateTable(this.dbProvider.getDBService());
110 public void close() throws ManagedProcessException {
114 File f = new File(TESTPROPFILE);
120 public SqlDBDataProvider getDbProvider() {
128 private static DB startDatabase(int port) throws ManagedProcessException {
129 // Start MariaDB4j database
130 DBConfigurationBuilder dbconfig = DBConfigurationBuilder.newBuilder();
131 dbconfig.setPort(port); // 0 => autom. detect free port
132 DB db = DB.newEmbeddedDB(dbconfig.build());
136 public static void testCreateTable(SqlDBClient dbService) {
137 createTableOdl(dbService);
138 createTable(dbService, ConnectionlogEntity.class, Entity.Connectionlog, true);
139 createTable(dbService, EventlogEntity.class, Entity.Eventlog, true);
140 createTable(dbService, FaultcurrentEntity.class, Entity.Faultcurrent, false);
141 createTable(dbService, FaultlogEntity.class, Entity.Faultlog, true);
142 createTable(dbService, CmlogEntity.class, Entity.Cmlog, true);
143 createTable(dbService, GuicutthroughEntity.class, Entity.Guicutthrough, true);
144 createTable(dbService, Pmdata15mEntity.class, Entity.Historicalperformance15min, false);
145 createTable(dbService, Pmdata24hEntity.class, Entity.Historicalperformance24h, false);
146 createTable(dbService, InventoryEntity.class, Entity.Inventoryequipment, false);
147 createTable(dbService, MaintenanceEntity.class, Entity.Maintenancemode, false);
148 createTable(dbService, MediatorServerEntity.class, Entity.MediatorServer, true);
149 createTable(dbService, NetworkElementConnectionEntity.class, Entity.NetworkelementConnection, false);
151 public static boolean createTableOdl(SqlDBClient dbService) {
152 String createStatement = null;
153 createStatement = SqlDBMapper.createTableOdl();
154 System.out.println(createStatement);
156 return dbService.write(createStatement);
157 } catch (SQLException e) {
158 // TODO Auto-generated catch block
165 public static boolean createTable(SqlDBClient dbService, Class<?> cls, Entity entity, boolean autoIndex) {
166 return createTable(dbService, cls, entity, autoIndex, true);
168 public static boolean createTable(SqlDBClient dbService, Class<?> cls, Entity entity, boolean autoIndex,
169 boolean withControllerId) {
170 String createStatement = null;
172 createStatement = SqlDBMapper.createTable(cls, entity, "", autoIndex,withControllerId);
173 } catch (UnableToMapClassException e) {
174 fail(e.getMessage());
176 System.out.println(createStatement);
177 return dbService.createTable(createStatement);
179 public String getDBUrl() {
180 return envDefaultValues.get("SDNRDBURL");
182 public String getDBUsername() {
183 return envDefaultValues.getOrDefault("SDNRDBUSERNAME","");
185 public String getDBPassword() {
186 return envDefaultValues.getOrDefault("SDNRDBPASSWORD","");