001package com.unitils.boot.autoconfigure;
002
003import com.unitils.boot.SpringBootModule;
004import com.unitils.boot.util.UnitilsBootBlockJUnit4ClassRunner;
005import org.springframework.beans.factory.InitializingBean;
006import org.springframework.beans.factory.annotation.Autowired;
007import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
008import org.springframework.context.ConfigurableApplicationContext;
009import org.springframework.context.annotation.Configuration;
010
011/**
012 * @Author: yangjianzhou
013 * @Description:
014 * @Date:Created in 2018-07-08
015 */
016@Configuration
017@ConditionalOnClass(UnitilsBootBlockJUnit4ClassRunner.class)
018public class ConfigurableApplicationContextAware implements InitializingBean {
019
020    @Autowired
021    private ConfigurableApplicationContext configurableApplicationContext;
022
023    @Override
024    public void afterPropertiesSet() throws Exception {
025        SpringBootModule.setApplicationContext(configurableApplicationContext);
026    }
027}