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

我有一个类，我想用一个调用私有方法的公共方法来测试它。我想假设私有方法正确工作。例如，我喜欢像 doReturn....when...这样的东西。我发现有 使用 PowerMock 的可能解决方案，但这个解决方案不适合我。 怎么做? 有人有这个问题吗？

198831 次浏览
小开
最佳答案

我看不出有什么问题。通过下面使用 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();
}
}
小开

使用任何测试框架(如果您的类是非 final的)的通用解决方案是手动创建您自己的模拟。

  1. 将您的私有方法更改为 protected。
  2. 在测试类中扩展该类
  3. 重写之前的私有方法以返回您想要的任何常量

这并没有使用任何框架，因此它并不优雅，但它总是能够工作: 即使没有 PowerMock。或者，如果您已经完成了步骤 # 1，那么您可以使用 Mockito 为您完成步骤 # 2 & # 3。

要直接模拟私有方法，需要使用 PowerMock，如 另一个答案所示。

小开

我知道一个方法，你可以叫你的私人功能，测试在模仿

@Test
public  void  commandEndHandlerTest() throws  Exception
{
Method retryClientDetail_privateMethod =yourclass.class.getDeclaredMethod("Your_function_name",null);
retryClientDetail_privateMethod.setAccessible(true);
retryClientDetail_privateMethod.invoke(yourclass.class, null);
}
小开

不知道为什么 Brice 的回答对我不起作用。我可以稍微操纵一下让它工作起来。可能是因为我有了新版的 PowerMock。我用的是1.6。

import java.util.Random;


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;
}
}

测试类看起来如下:

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.doReturn;


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


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


doReturn(true).when(classToTest, "doTheGamble", anyString(), anyInt());


classToTest.meaningfulPublicApi();
}
}
小开

毫无疑问:

ourObject = PowerMockito.spy(new OurClass());
when(ourObject , "ourPrivateMethodName").thenReturn("mocked result");

以 String为例:

ourObject = PowerMockito.spy(new OurClass());
when(ourObject, method(OurClass.class, "ourPrivateMethodName", String.class))
.withArguments(anyString()).thenReturn("mocked result");
小开

一些需要考虑的事情

确保 private 函数正在调用另一个公共函数，并且只能继续模拟公共函数。


提交答案