Load CQL scripts in tests. about embedded-cassandra-spring-boot-starter HOT 5 CLOSED

public class EmbeddedCassandraInitializer implements InitializingBean {

	private final CqlSession session;

	public EmbeddedCassandraInitializer(CqlSession session) {
		this.session = session;
	}

	@Override
	public void afterPropertiesSet() throws Exception {
		// initialize cassandra
		CqlScript.ofClassPath("schema.cql").forEachStatement(this.session::execute);
	}

}

public class EmbeddedCassandraInitializerPostProcessor implements BeanPostProcessor, Ordered, BeanFactoryAware {

	private BeanFactory beanFactory;

	@Override
	public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
		this.beanFactory = beanFactory;
	}

	@Override
	public int getOrder() {
		return Ordered.HIGHEST_PRECEDENCE + 1;
	}

	@Override
	public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
		return bean;
	}

	@Override
	public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
		if (bean instanceof Cassandra) {
			this.beanFactory.getBean(EmbeddedCassandraInitializer.class);
		}
		return bean;
	}

}

@Configuration(proxyBeanMethods = false)
@Import(EmbeddedCassandraInitializer.class)
public class CassandraConfiguration {

	@Bean
	public static EmbeddedCassandraInitializerPostProcessor embeddedCassandraInitializerPostProcessor() {
		return new EmbeddedCassandraInitializerPostProcessor();
	}

}

@Configuration(proxyBeanMethods = false)
static class CqlSessionFactoryBeanConfiguration {

@Bean
CqlSessionFactoryBean cqlSessionFactoryBean() {
	CqlSessionFactoryBean cqlSessionFactoryBean = new CqlSessionFactoryBean();
	cqlSessionFactoryBean.setLocalDatacenter("datacenter1");
	cqlSessionFactoryBean.setKeyspaceSpecifications();
	cqlSessionFactoryBean.setKeyspaceStartupScripts();
	cqlSessionFactoryBean.setKeyspaceShutdownScripts();
	return cqlSessionFactoryBean;
}

}