adapters/mso-requests-db-adapter/src/test/java/org/onap/so/adapters/requestsdb/application/EmbeddedMariaDbConfig.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.adapters.requestsdb.application;
  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.jdbc.DataSourceBuilder;
  29 import org.springframework.boot.context.properties.ConfigurationProperties;
  30 import org.springframework.boot.orm.jpa.EntityManagerFactoryBuilder;
  31 import org.springframework.context.annotation.Bean;
  32 import org.springframework.context.annotation.Configuration;
  33 import org.springframework.context.annotation.Primary;
  34 import org.springframework.context.annotation.Profile;
  35 import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
  36 import org.springframework.orm.jpa.JpaTransactionManager;
  37 import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
  38 import org.springframework.transaction.PlatformTransactionManager;
  39 import org.springframework.transaction.annotation.EnableTransactionManagement;
  40
  41 import javax.persistence.EntityManagerFactory;
  42 import javax.sql.DataSource;
  43
  44 @Configuration
  45 @Profile({"test"})
  46 @EnableTransactionManagement
  47 @EnableJpaRepositories(
  48                 entityManagerFactoryRef = "requestEntityManagerFactory",transactionManagerRef = "requestTransactionManager",
  49                 basePackages = { "org.onap.so.db.request.data.repository"}
  50                 )
  51 public class EmbeddedMariaDbConfig {
  52
  53     @Bean
  54     MariaDB4jSpringService mariaDB4jSpringService() {
  55         return new MariaDB4jSpringService();
  56     }
  57
  58     @Primary
  59         @Bean(name = "requestDataSource")
  60         @ConfigurationProperties(prefix = "spring.datasource")
  61     DataSource dataSource(MariaDB4jSpringService mariaDB4jSpringService,
  62                           @Value("${mariaDB4j.databaseName}") String databaseName,
  63                           @Value("${spring.datasource.username}") String datasourceUsername,
  64                           @Value("${spring.datasource.password}") String datasourcePassword,
  65                           @Value("${spring.datasource.driver-class-name}") String datasourceDriver) throws ManagedProcessException {
  66         //Create our database with default root user and no password
  67         mariaDB4jSpringService.getDB().createDB(databaseName);
  68
  69         DBConfigurationBuilder config = mariaDB4jSpringService.getConfiguration();
  70
  71         return DataSourceBuilder
  72                 .create()
  73                 .username(datasourceUsername)
  74                 .password(datasourcePassword)
  75                 .url(config.getURL(databaseName))
  76                 .driverClassName(datasourceDriver)
  77                 .build();
  78     }
  79
  80         @Primary
  81         @Bean(name = "requestEntityManagerFactory")
  82         public LocalContainerEntityManagerFactoryBean
  83         entityManagerFactory(
  84                         EntityManagerFactoryBuilder builder,
  85                         @Qualifier("requestDataSource") DataSource dataSource
  86                         ) {
  87                 return builder
  88                                 .dataSource(dataSource)
  89                                 .packages("org.onap.so.db.request.beans")
  90                                 .persistenceUnit("requestDB")
  91                                 .build();
  92         }
  93
  94         @Primary
  95         @Bean(name = "requestTransactionManager")
  96         public PlatformTransactionManager transactionManager(
  97                         @Qualifier("requestEntityManagerFactory") EntityManagerFactory
  98                         entityManagerFactory
  99                         ) {
 100                 return new JpaTransactionManager(entityManagerFactory);
 101         }
 102 }