如何使用PowerMock模拟测试的私有方法？

java testing junit mockito powermock

2022-08-31 13:36:50

我有一个类，我想用一个调用私有方法的公共方法来测试它。我想假设私有方法工作正常。例如，我想要类似 .我发现使用PowerMock有可能的解决方案，但这个解决方案对我不起作用。如何做到这一点？有人有这个问题吗？doReturn....when...

答案 1

我在这里没有看到问题。通过以下使用Mockito API的代码，我设法做到了这一点：

public class CodeWithPrivateMethod {

    public void meaningfulPublicApi() {
        if (doTheGamble("Whatever", 1 << 3)) {
            throw new RuntimeException("boom");
        }
    }

    private boolean doTheGamble(String whatever, int binary) {
        Random random = new Random(System.nanoTime());
        boolean gamble = random.nextBoolean();
        return gamble;
    }
}

这是JUnit测试：

import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import static org.mockito.Matchers.anyInt;
import static org.mockito.Matchers.anyString;
import static org.powermock.api.mockito.PowerMockito.when;
import static org.powermock.api.support.membermodification.MemberMatcher.method;

@RunWith(PowerMockRunner.class)
@PrepareForTest(CodeWithPrivateMethod.class)
public class CodeWithPrivateMethodTest {

    @Test(expected = RuntimeException.class)
    public void when_gambling_is_true_then_always_explode() throws Exception {
        CodeWithPrivateMethod spy = PowerMockito.spy(new CodeWithPrivateMethod());

        when(spy, method(CodeWithPrivateMethod.class, "doTheGamble", String.class, int.class))
                .withArguments(anyString(), anyInt())
                .thenReturn(true);

        spy.meaningfulPublicApi();
    }
}

答案 2

与任何测试框架（如果您的类是非-）一起使用的通用解决方案是手动创建自己的模拟。final

将专用方法更改为受保护方法。
在测试类中扩展类
重写以前的私有方法以返回所需的任何常量

这不使用任何框架，所以它不那么优雅，但它将永远工作：即使没有PowerMock。或者，如果您已经完成了步骤#1，则可以使用Mockito为您执行步骤#2和#3。

要直接模拟私有方法，您需要使用PowerMock，如其他答案所示。