Question

代码库是典型的基于Spring的企业代码库，具有大约150万行代码。我们有很多弹簧上下文文件。测试基础是一个问题。

对于测试用例，我创建了另一组测试弹簧文件（主要是导入相关的项目弹簧上下文），而少数bean包含用于外部服务的模拟bean。所有测试类都使用相同的上下文配置文件集，并且90％的时间都是如此。

但在某些情况下，我会想要嘲笑一个豆子。但是我不想编辑spring-text.xml（因为它会干扰所有类），我也不希望每个测试类都有单独的xml集。一个非常简单的说法就是：

@Autowired
@Qualifier("fundManager")
FundManager fundManager;

@Test
public void testSomething(){
    TransactionManager tx = mock(TransactionManager.class);
    fundManager.setTransactionManager(tx);
    //now all is well.
}

这在某些情况下有效。但有时候，希望在整个代码库中使用tx bean的地方设置这个新的临时bean TransactionManager。

代理类恕我直言不是一个很好的解决方案，因为我必须用包装器包装所有bean。这就是我理想的选择：

@Test
public void testSomething(){
    TransactionManager tx = mock(TransactionManager.class);
    replace("transactionManagerBean",tx);
    //For bean with id:transactionManagerBean should be replace with `tx`
}

BeanPostProcessor看起来像是另一种建议，但我遇到了一些小问题。

Answer 1

想象一下，你把豆A注入豆B：

public static class A {
}

public static class B {
    @Autowired
    private A a;

    @Override
    public String toString() {
        return "B [a=" + a + ']';
    }
}

初始化应用程序的spring上下文：

<?xml version="1.0" encoding="UTF-8"?>
<beans ...>
    <context:annotation-config/>
    <bean id="a" class="test.Test$A"/>
    <bean id="b" class="test.Test$B"/>
</beans>

然后，以下代码段将替换上下文中的所有bean：

public static void main(String[] args) throws Exception {
    ApplicationContext ctx = new ClassPathXmlApplicationContext("test.xml");

    System.out.println(ctx.getBean("b"));

    final A replacement = new A();
    for (String name : ctx.getBeanDefinitionNames()) {
        final Object bean = ctx.getBean(name);
        ReflectionUtils.doWithFields(bean.getClass(),
            field -> {
                field.setAccessible(true);
                field.set(bean, replacement);
            },
            // Here you can provide your filtering.
            field -> field.getType().equals(A.class)
        );
    }

    System.out.println(ctx.getBean("b"));
}

这个例子是用Java 8 + Spring 4.1制作的。但是，修改Java和Spring的旧版本的代码会很简单。

在运行时替换bean

1 个答案: