如何使用PowerMockito模拟私有字段?

时间:2018-02-08 17:53:23

标签: java junit powermock powermockito

作为重复提议的解决方案不是PowerMockito解决方案,因此不回答这个问题。此外,这个问题在下面合法地回答。

如果这是重复的IDK,但我确定无法找到有问题的项目。我一直希望这很简单,因为反射非常简单,但我宁愿使用正确的工具。

澄清:遗留代码。没有吸气剂/安装者。

使用Whitebox是否正确?我认为它是“Off Limits”,即内部API的一部分? ......还是那个严格的Mockito?

1 个答案:

答案 0 :(得分:2)

请参阅Whitebox.setInternalState(...)

例如 - 给定的类A需要测试:

public class A {
    private B b;

    public A() {
        b = new B();
    }

    public void doSomething() {
        b.doSomething();
    }

}

,其私有实例为B

public class B {

    public void doSomething() {
        // some long running, resource intensive process...
        System.out.println("Real B.doSomething() was invoked.");
    }

}

然后Whitebox可用于设置A的私有状态,以便进行测试:

import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.verify;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import org.powermock.reflect.Whitebox;

@RunWith(MockitoJUnitRunner.class)
public class ATest {

    @Mock
    private B b;

    private A a;

    @Before
    public void prepareTest() {
        doNothing().when(b).doSomething();

        a = new A();
        Whitebox.setInternalState(a, B.class, b);
    }

    @Test
    public void doSomething() {
        a.doSomething();
        verify(b).doSomething();
    }

}