PowerMock私有方法的简单尝试

最新推荐文章于 2025-06-24 10:03:21 发布

原创最新推荐文章于 2025-06-24 10:03:21 发布 · 4.9k 阅读

2 ·

CC 4.0 BY-SA版权

爬爬爬

Spring 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用PowerMock框架来测试包含私有方法的Java类的有效方法。通过示例展示了如何模拟私有方法的行为，从而提高代码测试覆盖率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

公司的代码都要设计到代码测试的覆盖率的问题，简单的测试工作还是可以的但是在处理私有方法的时候以前的都是直接测到方法内存，凭空多出了好多代码量。
这里找到了一中比较巧妙的方法感觉还不错。
代码如下：

package powermock;

import java.util.Random;

public class CodeWithPrivateMethod {
    public void meaningfulpublicAPI(){
        throw new RuntimeException("Duang duang");
    }

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

测试用例：

package powermock;

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.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.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_when_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();
    }

}