v1/src/test/java/com/bami/dfm/config/SqlTestContainersSpringContextCustomizerFactory.java - main - developer.bami/DFM

v1/src/test/java/com/bami/dfm/config/SqlTestContainersSpringContextCustomizerFactory.java
54 строки3 KB
Michael
Initial version of dfm generated by generator-jhipster@8.0.0-beta.1
28 янв 2024, 15:02
28 янв 2024, 15:02a03b62a
package com.bami.dfm.config;

import java.util.Arrays;
import java.util.List;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.boot.test.util.TestPropertyValues;
import org.springframework.core.annotation.AnnotatedElementUtils;
import org.springframework.test.context.ContextConfigurationAttributes;
import org.springframework.test.context.ContextCustomizer;
import org.springframework.test.context.ContextCustomizerFactory;
import tech.jhipster.config.JHipsterConstants;

public class SqlTestContainersSpringContextCustomizerFactory implements ContextCustomizerFactory {

    private Logger log = LoggerFactory.getLogger(SqlTestContainersSpringContextCustomizerFactory.class);

    private static SqlTestContainer prodTestContainer;

    @Override
    public ContextCustomizer createContextCustomizer(Class<?> testClass, List<ContextConfigurationAttributes> configAttributes) {
        return (context, mergedConfig) -> {
            ConfigurableListableBeanFactory beanFactory = context.getBeanFactory();
            TestPropertyValues testValues = TestPropertyValues.empty();
            EmbeddedSQL sqlAnnotation = AnnotatedElementUtils.findMergedAnnotation(testClass, EmbeddedSQL.class);
            boolean usingTestProdProfile = Arrays
                .asList(context.getEnvironment().getActiveProfiles())
                .contains("test" + JHipsterConstants.SPRING_PROFILE_PRODUCTION);
            if (null != sqlAnnotation && usingTestProdProfile) {
                log.debug("detected the EmbeddedSQL annotation on class {}", testClass.getName());
                log.info("Warming up the sql database");
                if (null == prodTestContainer) {
                    try {
                        Class<? extends SqlTestContainer> containerClass = (Class<? extends SqlTestContainer>) Class.forName(
                            this.getClass().getPackageName() + ".PostgreSqlTestContainer"
                        );
                        prodTestContainer = beanFactory.createBean(containerClass);
                        beanFactory.registerSingleton(containerClass.getName(), prodTestContainer);
                        // ((DefaultListableBeanFactory)beanFactory).registerDisposableBean(containerClass.getName(), prodTestContainer);
                    } catch (ClassNotFoundException e) {
                        throw new RuntimeException(e);
                    }
                }
                testValues =
                    testValues.and("spring.r2dbc.url=" + prodTestContainer.getTestContainer().getJdbcUrl().replace("jdbc", "r2dbc") + "");
                testValues = testValues.and("spring.r2dbc.username=" + prodTestContainer.getTestContainer().getUsername());
                testValues = testValues.and("spring.r2dbc.password=" + prodTestContainer.getTestContainer().getPassword());
                testValues = testValues.and("spring.liquibase.url=" + prodTestContainer.getTestContainer().getJdbcUrl() + "");
            }
            testValues.applyTo(context);
        };
    }
}