PowerMock与Mockito无法及时Mock静态字段的技术求助

这种静态字段在类加载阶段就硬初始化的情况确实头疼，我之前处理过类似的老旧API测试问题，给你几个实用的解决方案：

方案1：用PowerMock抑制类A的静态初始化（无需修改原代码） #

PowerMock其实提供了专门的注解来拦截类的静态初始化流程，刚好能解决你说的“Mock时机太晚”的问题。你可以用@SuppressStaticInitializationFor跳过类A中静态字段的自动初始化，之后再手动注入Mock的B对象：

@RunWith(PowerMockRunner.class)
// 替换成你的类A的实际全类名
@SuppressStaticInitializationFor("com.yourpackage.A") 
public class ATest {

    @Mock
    private B mockB;

    @Before
    public void setUp() {
        MockitoAnnotations.openMocks(this);
        // 手动把Mock的B对象设置到A的静态字段b中
        Whitebox.setInternalState(A.class, "b", mockB);
    }

    @Test
    public void testFoo() {
        // 现在静态字段b已经是Mock对象，不会触发B构造的异常了
        A a = new A();
        assertEquals(666, a.foo());
    }
}

原理是@SuppressStaticInitializationFor会阻止类A在加载时执行静态字段的初始化逻辑（也就是不会调用new B()），完全避开了B构造带来的异常，之后再通过Whitebox工具类手动设置静态字段的值。

方案2：重构类A，引入依赖注入（推荐长期方案） #

如果项目允许修改原代码的话，这是最长期、最优雅的解决方案——把静态耦合改成实例级的依赖注入，彻底解决测试难的问题：

修改后的类A：

public class A { 
    private B b; 

    // 提供构造器注入的方式，方便测试时传入Mock对象
    public A(B b) {
        this.b = b;
    }

    // 保留无参构造器，兼容原有业务代码的调用逻辑
    public A() {
        this.b = new B();
    }

    // 也可以加Setter方法，灵活替换B对象
    public void setB(B b) {
        this.b = b;
    }

    public int foo(){return 666;} 
}

测试代码就简单多了，直接用Mockito注入Mock的B：

public class ATest {
    @Mock
    private B mockB;

    @Before
    public void setUp() {
        MockitoAnnotations.openMocks(this);
    }

    @Test
    public void testFoo() {
        A a = new A(mockB);
        assertEquals(666, a.foo());
    }
}

这种方式不仅解决了测试问题，还提升了代码的可维护性，后续如果要替换B的实现也会更方便。

方案3：用PowerMock拦截B的构造调用（备选方案） #

如果方案1的抑制初始化不适用，你也可以尝试用PowerMock的whenNew来拦截B的构造，提前返回Mock对象：

@RunWith(PowerMockRunner.class)
@PrepareForTest({A.class, B.class})
public class ATest {

    @Mock
    private B mockB;

    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.openMocks(this);
        // 拦截B的无参构造，返回Mock对象
        PowerMockito.whenNew(B.class).withNoArguments().thenReturn(mockB);
        // 保险起见，手动重置A的静态字段（避免类提前加载导致初始化完成）
        Whitebox.setInternalState(A.class, "b", mockB);
    }

    @Test
    public void testFoo() {
        A a = new A();
        assertEquals(666, a.foo());
    }
}

不过这个方案可能会因为类加载顺序的问题失效（比如类A在测试类加载前就已经被加载，导致静态字段已经初始化完成），所以优先级不如方案1。

总结一下：如果不能改原代码，优先用方案1；如果可以修改，强烈推荐方案2从根源解决问题。

内容的提问来源于stack exchange，提问作者czpona