使用PowerMockito模拟Java中其他类的静态函数

时间:2018-02-17 13:37:11

标签: java unit-testing junit mockito powermock

我编写了用于测试类Main中的函数的测试用例,称为functionMain()。我见过有人使用PowerMockito来测试正在测试的类Main中的静态函数。

但就我而言,functionMain()正在使用名为Branch的名为staticBranchFunction()的另一个类的静态函数。

我想在staticBranchFunction()类的测试中模拟Main

这个main函数实际上调用了来自不同类Branch1Branch2等的静态函数。

请帮忙。

1 个答案:

答案 0 :(得分:1)

import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.when;
import static org.mockito.Mockito.times;
import static org.powermock.api.mockito.PowerMockito.mockStatic;
import static org.powermock.api.mockito.PowerMockito.verifyStatic;

@RunWith(PowerMockRunner.class)
@PrepareForTest({Boom.class})
public class DocTest {

    public String boomWrapper() {
        return Boom.detonate();
    }

    @Test
    public void testBoom() {
        mockStatic(Boom.class);
        when(Boom.detonate()).thenReturn("defused");
        String actual = boomWrapper();
        verifyStatic(Boom.class, times(1));
        Boom.detonate();
        assertEquals("defused", actual);
    }    
}

class Boom {
    private static final String BOOM = "Boom!";  
    public static String detonate() {
        return BOOM;
    }
}

依赖关系:

junit:junit:4.12  
org.mockito:mockito-core:2.13.0  
org.powermock:powermock-module-junit4:2.0.0-beta.5  
org.powermock:powermock-api-mockito2:2.0.0-beta.5  

说明

如需更多支持的版本,请阅读:Mockito + PowerMockother supported frameworks 要求:

  • 以逗号分隔@PrepareForTest({Boom.class})中的所有静态类。
  • 用逗号分隔PowerMockito.mockStatic(Boom.class)所有静态类。
  • 使用常规模拟方法来设置您的期望,例如Mockito.when(Boom.detonate()).thenReturn("defused")
  • PowerMockito.verifyStatic(Boom.class, Mockito.times(1)); Boom.detonate();验证行为重要提示:您需要按方法验证调用PowerMockito.verifyStatic(Boom.class)

有关PowerMock wiki的更多详情。