com.github.fridujo.automocker.base.MockJdbc Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-automocker Show documentation
The newest version!
package com.github.fridujo.automocker.base;

import com.github.fridujo.automocker.api.AfterBeanRegistration;
import com.github.fridujo.automocker.api.AfterBeanRegistrationExecutable;
import com.github.fridujo.automocker.api.ExtendedBeanDefinitionRegistry;
import com.github.fridujo.automocker.api.jdbc.DataSourceResetter;
import com.github.fridujo.automocker.utils.Classes;
import com.github.fridujo.automocker.utils.PropertiesBuilder;
import org.springframework.util.PropertyPlaceholderHelper;

import javax.sql.DataSource;
import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * Alter bean definitions of {@link DataSource DataSources} by setting the URL of an embedded memory one.
 */
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented

@AfterBeanRegistration(MockJdbc.MockJdbcExecutable.class)
public @interface MockJdbc {

    String url() default "jdbc:h2:mem:${beanName};DB_CLOSE_DELAY=-1";

    class MockJdbcExecutable implements AfterBeanRegistrationExecutable {
        private static final String H2_DATASOURCE_CLASS = "org.h2.jdbcx.JdbcDataSource";
        private static final PropertyPlaceholderHelper PROPERTY_PLACEHOLDER_HELPER
            = new PropertyPlaceholderHelper("${", "}");

        @Override
        public void execute(MockJdbc annotation, ExtendedBeanDefinitionRegistry extendedBeanDefinitionRegistry) {
            Set dataSourceBeans = extendedBeanDefinitionRegistry
                .getBeanDefinitionsForClass(DataSource.class);

            if (!dataSourceBeans.isEmpty()) {
                if (Classes.isPresent(H2_DATASOURCE_CLASS)) {
                    dataSourceBeans.forEach(meta -> {
                        meta.beanDefinitionModifier()
                            .copyFactoryQualifiersAsDetached()
                            .reset()
                            .setBeanClass(Classes.forName(H2_DATASOURCE_CLASS))
                            .addPropertyValue("url",
                                PROPERTY_PLACEHOLDER_HELPER.replacePlaceholders(annotation.url(), PropertiesBuilder.of("beanName", meta.name())));
                    });
                } else {
                    throw new IllegalStateException("\nAutomocker is missing class [" + H2_DATASOURCE_CLASS + "] to mock " + dataSourceBeans.size() + " bean(s) of type [" + DataSource.class.getName() + "]: " +
                        dataSourceBeans.stream().map(ExtendedBeanDefinitionRegistry.BeanDefinitionMetadata::name).collect(Collectors.joining(", ")) +
                        "\nMake sure h2.jar is in the test classpath");
                }
                extendedBeanDefinitionRegistry.registerBeanDefinition(DataSourceResetter.class);
            }
        }
    }
}