将测试bean注入主方法

Question

我正在使用JavaConfig来管理Spring bean并将其连接到我的Java应用程序中。Java 应用程序是一个主要方法 - 基本上作为批处理作业运行，通过 bash 文件调用。有没有办法在我的主方法中使用不同的（测试）配置？

public static void main(String[] args) {
final ApplicationContext context = new AnnotationConfigApplicationContext(ApplicationConfig.class);
// do Stuff
}

我之前在我的测试类中成功地使用了以下注释：@RunWith（SpringJUnit4ClassRunner.class） @ContextConfiguration（classes = { TestConfig.class }）

，但这不适用于“主要”应用程序。除了在Spring上下文中传递作为参数之外，我不确定我可以在这里做什么。谢谢

Answer 1

您也应该能够在实际的配置类中使用配置文件来做您想要的事情。通过设置所需的配置文件，您可以“注入”您想要的不同bean。

您的应用程序配置可能如下所示：

@Configuration
@Import({
    JndiDataConfig.class,
    TestDataConfig.class,
)
public class ApplicationConfig {
   ...

其中TestDataConfig看起来(部分)像:

@Configuration
@Profile("test")
public class TestDataConfig {

    @Bean
    public DataSource dataSource() {
        return new EmbeddedDatabaseBuilder()
        .setType(EmbeddedDatabaseType.HSQL)
        .addScript("classpath:com/bank/config/sql/schema.sql")
        .addScript("classpath:com/bank/config/sql/test-data.sql")
        .build();
    }

}

其中JndiDataConfig看起来像：

@Configuration
@Profile("production")
public class JndiDataConfig {

    @Bean
    public DataSource dataSource() throws Exception {
        Context ctx = new InitialContext();
        return (DataSource) ctx.lookup("java:comp/env/jdbc/datasource");
    }

}