mso-api-handlers/mso-api-handler-infra/src/test/java/org/onap/so/apihandlerinfra/EmbeddedCatalogDbConfig.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   4  * ================================================================================
   5  * Copyright (C) 2017 - 2018 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.so.apihandlerinfra;
  22 import ch.vorburger.exec.ManagedProcessException;
  23 import ch.vorburger.mariadb4j.DBConfigurationBuilder;
  24 import ch.vorburger.mariadb4j.springframework.MariaDB4jSpringService;
  25
  26 import org.springframework.beans.factory.annotation.Qualifier;
  27 import org.springframework.beans.factory.annotation.Value;
  28 import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
  29 import org.springframework.boot.orm.jpa.EntityManagerFactoryBuilder;
  30 import org.springframework.context.annotation.Bean;
  31 import org.springframework.context.annotation.Configuration;
  32 import org.springframework.context.annotation.Primary;
  33 import org.springframework.context.annotation.Profile;
  34 import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
  35 import org.springframework.orm.jpa.JpaTransactionManager;
  36 import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
  37 import org.springframework.transaction.PlatformTransactionManager;
  38 import org.springframework.transaction.annotation.EnableTransactionManagement;
  39
  40 import javax.persistence.EntityManagerFactory;
  41 import javax.sql.DataSource;
  42
  43 @Configuration
  44 @Profile({"test"})
  45 @EnableTransactionManagement
  46 @EnableJpaRepositories(
  47                 entityManagerFactoryRef = "entityManagerFactory",
  48                 basePackages = {"org.onap.so.db.catalog.data.repository"}
  49                 )
  50 public class EmbeddedCatalogDbConfig {
  51
  52     @Bean
  53     MariaDB4jSpringService mariaDB4jSpringService() {
  54         return new MariaDB4jSpringService();
  55     }
  56
  57         @Primary
  58         @Bean(name = "dataSource")
  59     DataSource dataSource(MariaDB4jSpringService mariaDB4jSpringService,
  60                           @Value("${mariaDB4j.databaseName}") String databaseName,
  61                           @Value("${spring.datasource.username}") String datasourceUsername,
  62                           @Value("${spring.datasource.password}") String datasourcePassword,
  63                           @Value("${spring.datasource.driver-class-name}") String datasourceDriver) throws ManagedProcessException {
  64         //Create our database with default root user and no password
  65         mariaDB4jSpringService.getDB().createDB(databaseName);
  66
  67         DBConfigurationBuilder config = mariaDB4jSpringService.getConfiguration();
  68
  69         return DataSourceBuilder
  70                 .create()
  71                 .username(datasourceUsername)
  72                 .password(datasourcePassword)
  73                 .url(config.getURL(databaseName))
  74                 .driverClassName(datasourceDriver)
  75                 .build();
  76     }
  77
  78
  79         @Primary
  80         @Bean(name = "entityManagerFactory")
  81         public LocalContainerEntityManagerFactoryBean
  82         entityManagerFactory(
  83                         EntityManagerFactoryBuilder builder,
  84                         @Qualifier("dataSource") DataSource dataSource
  85                         ) {
  86                 return builder
  87                                 .dataSource(dataSource)
  88                                 .packages("org.onap.so.db.catalog.beans")
  89                                 .persistenceUnit("catalogDB")
  90                                 .build();
  91         }
  92
  93         @Primary
  94         @Bean(name = "transactionManager")
  95         public PlatformTransactionManager transactionManager(
  96                         @Qualifier("entityManagerFactory") EntityManagerFactory
  97                         entityManagerFactory
  98                         ) {
  99                 return new JpaTransactionManager(entityManagerFactory);
 100         }
 101
 102 }