73

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

4

6 回答 6

103

我认为这里没有问题。通过使用 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();
    }
}
于 2011-10-18T08:31:20.983 回答
30

适用于任何测试框架（如果您的类不是）的通用解决方案final是手动创建自己的模拟。

  1. 将您的私有方法更改为受保护。
  2. 在您的测试类中扩展类
  3. 覆盖以前的私有方法以返回您想要的任何常量

这不使用任何框架，因此它不那么优雅，但它总是可以工作：即使没有 PowerMock。或者，如果您已经完成了第 1 步，您可以使用 Mockito 为您执行第 2 步和第 3 步。

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

于 2011-10-19T19:03:00.553 回答
4

出于某种原因，布莱斯的回答对我不起作用。我能够对其进行一些操作以使其正常工作。可能只是因为我有更新版本的 PowerMock。我正在使用 1.6.5。

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();
    }
}
于 2018-04-06T16:26:05.803 回答
1

我知道一种方法，你可以称你为私有函数在 mockito 中测试

@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);
    }
于 2017-08-24T08:21:08.043 回答
1

没有论据：

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");
于 2019-09-18T11:24:44.300 回答
0

需要考虑的事情

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

于 2021-05-03T07:16:02.183 回答