java - Spring Boot : Preload Data and configure H2 for tests. 特殊配置问题

Question

我正在开发一个演示 spring boot 应用程序，其中可以通过应用程序的 application.properties 使用 DDBBconfig 类配置 3 个不同的数据库。类 i 如下:

@Configuration
public class DDBBConfig {

    //Flags Configuration

    @Value("${DDBB.postgresql}")
    private Boolean postgresql;

    @Value("${DDBB.mysql}")
    private Boolean mysql;

    @Value("${DDBB.h2}")
    private Boolean h2;


    //Datasource Configuration parameters

    @Value("${mysql.datasource.driver-class-name}")
    private String driverMysql;

    @Value("${mysql.datasource.url}")
    private String urlMysql;

    @Value("${mysql.datasource.username}")
    private String usernameMysql;

    @Value("${mysql.datasource.password}")
    private String passwordMysql;

    @Value("${postgresql.datasource.driver-class-name}")
    private String driverPostgresql;

    @Value("${postgresql.datasource.url}")
    private String urlPostgresql;

    @Value("${postgresql.datasource.username}")
    private String usernamePostgresql;

    @Value("${postgresql.datasource.password}")
    private String passwordPostgresql;

    @Value("${spring.datasource.driverClassName}")
    private String driverH2;

    @Value("${spring.datasource.url}")
    private String urlH2;

    @Value("${spring.datasource.username}")
    private String usernameH2;

    @Value("${spring.datasource.password}")
    private String passwordH2;

    @Bean
    public DataSource getDataSource() {

        DataSourceBuilder dataSourceBuilder = DataSourceBuilder.create();

        validateOnlyOneFlag( postgresql, mysql, h2 );

        if (Boolean.TRUE.equals( mysql )) {
            dataSourceBuilder.driverClassName( driverMysql );
            dataSourceBuilder.url( urlMysql );
            dataSourceBuilder.username( usernameMysql );
            dataSourceBuilder.password( passwordMysql );
        } else if (Boolean.TRUE.equals( postgresql )) {
            dataSourceBuilder.driverClassName( driverPostgresql );
            dataSourceBuilder.url( urlPostgresql );
            dataSourceBuilder.username( usernamePostgresql );
            dataSourceBuilder.password( passwordPostgresql );
        } else if (Boolean.TRUE.equals( h2 )) {
            dataSourceBuilder.driverClassName( driverH2 );
            dataSourceBuilder.url( urlH2 );
            dataSourceBuilder.username( usernameH2 );
            dataSourceBuilder.password( passwordH2 );

        }

        return dataSourceBuilder.build();
    }

    public void validateOnlyOneFlag(Boolean postgress, Boolean mySql, Boolean h2) {

        Integer flagsTrue = 0;

        if (postgress) {
            flagsTrue++;
        }
        if (mySql) {
            flagsTrue++;
        }
        if (h2) {
            flagsTrue++;
        }

        if (flagsTrue > 1) {
            throw new IllegalArgumentException( "There is more than One database Flags to True." + "\n\tDDBB.postgresql=" + postgress +
                                 "\n\tDDBB.mysql=" + mySql + "\n\tDDBB.h2=" + h2 );
        } else if (flagsTrue == 0) {
            throw new IllegalArgumentException( "\n\n\tDatabase flags are all false, please set at least one flag to true" );
        }
    }

}

然后在 application.properties 中我有 somo 标志和配置参数。

#BBDD:
DDBB.postgresql=true
DDBB.mysql=false
DDBB.h2=false

#JPA:
spring.jpa.show-sql=true
spring.jpa.hibernate.ddl-auto=create-drop
spring.jpa.properties.hibernate.jdbc.lob.non_contextual_creation=true

#Properties mysql
mysql.datasource.url=jdbc:mysql://localhost:3306/myapp?serverTimezone=UTC
mysql.datasource.username=root
mysql.datasource.password=AdminPass123
mysql.datasource.driver-class-name=com.mysql.jdbc.Driver

#Properties postgresql
postgresql.datasource.url=jdbc:postgresql://localhost/petapp
postgresql.datasource.username=postgres
postgresql.datasource.password=1234
postgresql.datasource.driver-class-name=org.postgresql.Driver

#Properties H2
h2.datasource.url=jdbc:h2:mem:petappDB
h2.datasource.username=sa
h2.datasource.password=password
h2.datasource.driverClassName=org.h2.Driver
spring.h2.console.enabled=${DDBB.h2}

它工作完美，所以我可以配置 3 个数据库。

然后我还有一个数据库初始化类，它添加了一些初始数据用于初始化和前端开发。它初始化角色，为登录创建默认管理员用户等。

@Component
public class DDBBInitializer {

    private static final Logger log = LoggerFactory.getLogger( DDBBInitializer.class );

    @Autowired
    private RoleRepository roleRepository;

    @Autowired
    private UserRepository userRepository;

    @Autowired
    private OwnerRepository ownerRepository;

    @Autowired
    private PetRepository petRepository;

    @Autowired
    private PasswordEncoder passwordEncoder;

    @Value("${pettapp.config.default.username}")
    private String defaultUsername;

    @Value("${pettapp.config.default.email}")
    private String defaultEmail;

    @Value("${pettapp.config.default.password}")
    private String defaultPassword;

    @PostConstruct
    private void init() {

        log.info( "Initializacion of the DDBB" );

        for (Roles role : Roles.values()) {
            Role roleEntity = new Role( role );
            roleRepository.save( roleEntity );
            log.info( "Role: " + roleEntity.getRoleName() + " stored on DDBB" );
        }

        User defaultAdminUser = new User();
        defaultAdminUser.setUserId( new BigDecimal( 1 ) );
        defaultAdminUser.addOneRole( Roles.ADMIN );
        defaultAdminUser.setUsername( defaultUsername );
        defaultAdminUser.setPassword( passwordEncoder.encode( defaultPassword ) );
        defaultAdminUser.setActive( true );
        defaultAdminUser.setEmail( defaultEmail );

        log.info( "Default AdminUser Created: " + defaultAdminUser.getUsername() + "/" + defaultPassword );

        Owner adminOwnerProfile = new Owner();
        adminOwnerProfile.setAddress( "Calle de jacinto Nº6 Bajo B" );
        adminOwnerProfile.setName( "Manolo" );
        adminOwnerProfile.setSurname( "Amelgas" );
        adminOwnerProfile.setDefaulter( false );
        adminOwnerProfile.setTelephoneNumber( "678987656 " );
        adminOwnerProfile.setUser( defaultAdminUser );

        defaultAdminUser.setOwner( adminOwnerProfile );

        log.info( "Default Owner Created: " + adminOwnerProfile.getName() + " " + adminOwnerProfile.getSurname() );

        Pet testPet = new Pet();
        testPet.setAlive( true );
        testPet.setBitrh( new Date() );
        testPet.setBreed( "Carlino" );
        testPet.setFur( "White" );
        testPet.setName( "Lucky" );
        testPet.setOwner( adminOwnerProfile );

        adminOwnerProfile.addPet( testPet );

        log.info( "Default Pet Created: " + testPet.getName() );

        userRepository.save( defaultAdminUser );

    }
}

现在我想用 Junit 为我的应用程序中的所有方法创建所有测试(目前我有 10 个 Controller 、15 个服务和 30 个不同的测试方法)我想将测试配置为始终使用 H2 数据库并加载一些初始数据，但我找不到关于它的好文档。我应该如何加载初始数据，就像我为应用程序所做的那样，但仅用于测试？

我创建了一个/test/resources/application-test.properties，其参数与/java/resources/application.properties 中的参数相同，但配置了 H2(这意味着 H"标志设置为 true)。

DDBB.postgresql=false
DDBB.mysql=false
DDBB.h2=true

在编译应用程序时，它仍在使用 application.properties 文件进行测试，这可以在日志中看到:

[INFO] -------------------------------------------------------
[INFO]  T E S T S
[INFO] -------------------------------------------------------
[INFO] Running com.mashosoft.backEndTest.BackEndTestApplicationTests
[INFO] -------------------------------------------------------
[INFO]  T E S T S
[INFO] -------------------------------------------------------
[INFO] Running com.mashosoft.backEndTest.BackEndTestApplicationTests
12:01:02.759 [main] DEBUG org.springframework.test.context.junit4.SpringJUnit4ClassRunner - SpringJUnit4ClassRunner constructor called with [class com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:02.768 [main] DEBUG org.springframework.test.context.BootstrapUtils - Instantiating CacheAwareContextLoaderDelegate from class [org.springframework.test.context.cache.DefaultCacheAwareContextLoaderDelegate]
12:01:02.783 [main] DEBUG org.springframework.test.context.BootstrapUtils - Instantiating BootstrapContext using constructor [public org.springframework.test.context.support.DefaultBootstrapContext(java.lang.Class,org.springframework.test.context.CacheAwareContextLoaderDelegate)]
12:01:02.824 [main] DEBUG org.springframework.test.context.BootstrapUtils - Instantiating TestContextBootstrapper for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests] from class [org.springframework.boot.test.context.SpringBootTestContextBootstrapper]
12:01:02.864 [main] INFO org.springframework.boot.test.context.SpringBootTestContextBootstrapper - Neither @ContextConfiguration nor @ContextHierarchy found for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests], using SpringBootContextLoader
12:01:02.869 [main] DEBUG org.springframework.test.context.support.AbstractContextLoader - Did not detect default resource location for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests]: class path resource [com/mashosoft/backEndTest/BackEndTestApplicationTests-context.xml] does not exist
12:01:02.870 [main] DEBUG org.springframework.test.context.support.AbstractContextLoader - Did not detect default resource location for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests]: class path resource [com/mashosoft/backEndTest/BackEndTestApplicationTestsContext.groovy] does not exist
12:01:02.870 [main] INFO org.springframework.test.context.support.AbstractContextLoader - Could not detect default resource locations for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests]: no resource found for suffixes {-context.xml, Context.groovy}.
12:01:02.872 [main] INFO org.springframework.test.context.support.AnnotationConfigContextLoaderUtils - Could not detect default configuration classes for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests]: BackEndTestApplicationTests does not declare any static, non-private, non-final, nested classes annotated with @Configuration.
12:01:02.961 [main] DEBUG org.springframework.test.context.support.ActiveProfilesUtils - Could not find an 'annotation declaring class' for annotation type [org.springframework.test.context.ActiveProfiles] and class [com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:03.068 [main] DEBUG org.springframework.context.annotation.ClassPathScanningCandidateComponentProvider - Identified candidate component class: file [D:\Proyectos\PetApp\Back\target\classes\com\mashosoft\backEndTest\BackEndTestApplication.class]
12:01:03.084 [main] INFO org.springframework.boot.test.context.SpringBootTestContextBootstrapper - Found @SpringBootConfiguration com.mashosoft.backEndTest.BackEndTestApplication for test class com.mashosoft.backEndTest.BackEndTestApplicationTests
12:01:03.203 [main] DEBUG org.springframework.boot.test.context.SpringBootTestContextBootstrapper - @TestExecutionListeners is not present for class [com.mashosoft.backEndTest.BackEndTestApplicationTests]: using defaults.
12:01:03.203 [main] INFO org.springframework.boot.test.context.SpringBootTestContextBootstrapper - Loaded default TestExecutionListener class names from location [META-INF/spring.factories]: [org.springframework.boot.test.mock.mockito.MockitoTestExecutionListener, org.springframework.boot.test.mock.mockito.ResetMocksTestExecutionListener, org.springframework.boot.test.autoconfigure.restdocs.RestDocsTestExecutionListener, org.springframework.boot.test.autoconfigure.web.client.MockRestServiceServerResetTestExecutionListener, org.springframework.boot.test.autoconfigure.web.servlet.MockMvcPrintOnlyOnFailureTestExecutionListener, org.springframework.boot.test.autoconfigure.web.servlet.WebDriverTestExecutionListener, org.springframework.security.test.context.support.WithSecurityContextTestExecutionListener, org.springframework.security.test.context.support.ReactorContextTestExecutionListener, org.springframework.test.context.web.ServletTestExecutionListener, org.springframework.test.context.support.DirtiesContextBeforeModesTestExecutionListener, org.springframework.test.context.support.DependencyInjectionTestExecutionListener, org.springframework.test.context.support.DirtiesContextTestExecutionListener, org.springframework.test.context.transaction.TransactionalTestExecutionListener, org.springframework.test.context.jdbc.SqlScriptsTestExecutionListener]
12:01:03.223 [main] INFO org.springframework.boot.test.context.SpringBootTestContextBootstrapper - Using TestExecutionListeners: [org.springframework.test.context.web.ServletTestExecutionListener@7d20d0b, org.springframework.test.context.support.DirtiesContextBeforeModesTestExecutionListener@77f1baf5, org.springframework.boot.test.mock.mockito.MockitoTestExecutionListener@41a2befb, org.springframework.boot.test.autoconfigure.SpringBootDependencyInjectionTestExecutionListener@6c40365c, org.springframework.test.context.support.DirtiesContextTestExecutionListener@7bedc48a, org.springframework.test.context.transaction.TransactionalTestExecutionListener@131ef10, org.springframework.test.context.jdbc.SqlScriptsTestExecutionListener@55b0dcab, org.springframework.security.test.context.support.WithSecurityContextTestExecutionListener@38afe297, org.springframework.security.test.context.support.ReactorContextTestExecutionListener@2df3b89c, org.springframework.boot.test.mock.mockito.ResetMocksTestExecutionListener@23348b5d, org.springframework.boot.test.autoconfigure.restdocs.RestDocsTestExecutionListener@70325e14, org.springframework.boot.test.autoconfigure.web.client.MockRestServiceServerResetTestExecutionListener@37ceb1df, org.springframework.boot.test.autoconfigure.web.servlet.MockMvcPrintOnlyOnFailureTestExecutionListener@7c9d8e2, org.springframework.boot.test.autoconfigure.web.servlet.WebDriverTestExecutionListener@20d525]
12:01:03.223 [main] DEBUG org.springframework.test.annotation.ProfileValueUtils - Retrieved @ProfileValueSourceConfiguration [null] for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:03.223 [main] DEBUG org.springframework.test.annotation.ProfileValueUtils - Retrieved ProfileValueSource type [class org.springframework.test.annotation.SystemProfileValueSource] for class [com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:03.223 [main] DEBUG org.springframework.test.annotation.ProfileValueUtils - Retrieved @ProfileValueSourceConfiguration [null] for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:03.223 [main] DEBUG org.springframework.test.annotation.ProfileValueUtils - Retrieved ProfileValueSource type [class org.springframework.test.annotation.SystemProfileValueSource] for class [com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:03.223 [main] DEBUG org.springframework.test.annotation.ProfileValueUtils - Retrieved @ProfileValueSourceConfiguration [null] for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:03.223 [main] DEBUG org.springframework.test.annotation.ProfileValueUtils - Retrieved ProfileValueSource type [class org.springframework.test.annotation.SystemProfileValueSource] for class [com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:03.239 [main] DEBUG org.springframework.test.context.support.AbstractDirtiesContextTestExecutionListener - Before test class: context [DefaultTestContext@75db5df9 testClass = BackEndTestApplicationTests, testInstance = [null], testMethod = [null], testException = [null], mergedContextConfiguration = [WebMergedContextConfiguration@707194ba testClass = BackEndTestApplicationTests, locations = '{}', classes = '{class com.mashosoft.backEndTest.BackEndTestApplication}', contextInitializerClasses = '[]', activeProfiles = '{}', propertySourceLocations = '{}', propertySourceProperties = '{org.springframework.boot.test.context.SpringBootTestContextBootstrapper=true}', contextCustomizers = set[org.springframework.boot.test.context.filter.ExcludeFilterContextCustomizer@b9afc07, org.springframework.boot.test.json.DuplicateJsonObjectContextCustomizerFactory$DuplicateJsonObjectContextCustomizer@80169cf, org.springframework.boot.test.mock.mockito.MockitoContextCustomizer@0, org.springframework.boot.test.web.client.TestRestTemplateContextCustomizer@6c9f5c0d, org.springframework.boot.test.autoconfigure.properties.PropertyMappingContextCustomizer@0, org.springframework.boot.test.autoconfigure.web.servlet.WebDriverContextCustomizerFactory$Customizer@1ee807c6], resourceBasePath = 'src/main/webapp', contextLoader = 'org.springframework.boot.test.context.SpringBootContextLoader', parent = [null]], attributes = map['org.springframework.test.context.web.ServletTestExecutionListener.activateListener' -> true]], class annotated with @DirtiesContext [false] with mode [null].
12:01:03.239 [main] DEBUG org.springframework.test.annotation.ProfileValueUtils - Retrieved @ProfileValueSourceConfiguration [null] for test class [com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:03.239 [main] DEBUG org.springframework.test.annotation.ProfileValueUtils - Retrieved ProfileValueSource type [class org.springframework.test.annotation.SystemProfileValueSource] for class [com.mashosoft.backEndTest.BackEndTestApplicationTests]
12:01:03.270 [main] DEBUG org.springframework.test.context.support.TestPropertySourceUtils - Adding inlined properties to environment: {spring.jmx.enabled=false, org.springframework.boot.test.context.SpringBootTestContextBootstrapper=true, server.port=-1}
###########################################################
MyAPP BANNER
###########################################################

2019-10-16 12:01:04,040 INFO  [main] org.springframework.boot.StartupInfoLogger: Starting BackEndTestApplicationTests on gggarrido10 with PID 5252 (started by gggarrido in D:\Proyectos\PetApp\Back)
2019-10-16 12:01:04,040 DEBUG [main] org.springframework.boot.StartupInfoLogger: Running with Spring Boot v2.1.6.RELEASE, Spring v5.1.8.RELEASE
2019-10-16 12:01:04,040 INFO  [main] org.springframework.boot.SpringApplication: No active profile set, falling back to default profiles: default
2019-10-16 12:01:07,437 INFO  [main] com.mashosoft.backEndTest.config.DDBB.DDBBConfig: POSTGRESQL DDBB selected

可以看出，for test依然是配置postgresql而不是H2我现在知道如何进行了

如能提供帮助，我将不胜感激。

Answer 1

如果我正确理解你的问题，你是在问你的 DDBBConfig 类在测试中使用时(作为 bean 或初始化为局部变量)是否会使用 application-test.properties 或 application.properties。特性。答案是它将使用 application-test.properties。重要的是要知道 application-test.properties 将完全取代 application.properties。您不能在 application-test.properties 中只定义修改后的增量 - 您必须复制原始文件并进行修改。

您的测试应如下所示:

package your.package;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import what.ever.you.need;

@RunWith(SpringRunner.class)
@SpringBootTest
public class YourServiceTestClassName {

    @Autowired
    private DDBBConfig config;

    @Autowired
    private DDBBInitializer initializer;

    @Test
    public void testWhatYouNeed() {
        DateSource ds;
        ds = config.getDateSource();
        // do whatever you need

    }