如何用PowerMockito测试getValue1时Mock getValue2方法

实现步骤 #

1. 引入依赖 #

如果使用Maven，需在pom.xml中添加PowerMockito与Mockito的相关依赖（版本可按需调整）：

<dependencies>
    <!-- PowerMockito核心依赖 -->
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-module-junit4</artifactId>
        <version>2.0.9</version>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-api-mockito2</artifactId>
        <version>2.0.9</version>
        <scope>test</scope>
    </dependency>
    <!-- JUnit依赖 -->
    <dependency>
        <groupId>junit</groupId>
        <artifactId>junit</artifactId>
        <version>4.13.2</version>
        <scope>test</scope>
    </dependency>
</dependencies>

2. 修改测试类代码 #

要实现保留getValue1真实逻辑，同时MockgetValue2切断后续调用链，需使用PowerMockito的spy（部分Mock）功能，具体代码如下：

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.junit.Assert.assertEquals;

@RunWith(PowerMockRunner.class) // 指定PowerMock运行器，启用Mock能力
@PrepareForTest(MyClass.class)  // 告知PowerMock需要对MyClass做字节码处理
public class MyClassTest {

    @Test
    public void testGetValue1() {
        // 创建MyClass的Spy实例：保留类的真实方法逻辑，仅对指定方法Mock
        MyClass obj = PowerMockito.spy(new MyClass());

        // Mock getValue2方法，指定返回值，直接切断getValue2->getValue3的调用链
        PowerMockito.when(obj.getValue2()).thenReturn("mock_value2");

        // 执行待测试方法
        String result = obj.getValue1();

        // 验证结果是否符合预期
        assertEquals("mock_value2", result);
    }
}

关键说明 #

• @RunWith(PowerMockRunner.class)：必须指定该运行器，否则PowerMockito的Mock功能无法生效。
• @PrepareForTest(MyClass.class)：PowerMock需要修改目标类的字节码来实现内部方法Mock，此注解用于声明需要处理的类。
• PowerMockito.spy(new MyClass())：Spy对象区别于普通Mock对象，它会保留类的真实方法实现，仅对显式指定的方法进行Mock。
• PowerMockito.when(obj.getValue2()).thenReturn("mock_value2")：预设getValue2的返回值，当getValue1调用getValue2时，直接返回Mock值，不会触发后续getValue3、getValue4、getValue5的执行。

如果MyClass需要依赖注入，可结合@InjectMocks使用：

import org.mockito.InjectMocks;
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.junit.Assert.assertEquals;

@RunWith(PowerMockRunner.class)
@PrepareForTest(MyClass.class)
public class MyClassTest {

    @InjectMocks
    MyClass obj;

    @Test
    public void testGetValue1() {
        // 对注入后的实例创建Spy对象
        obj = PowerMockito.spy(obj);
        PowerMockito.when(obj.getValue2()).thenReturn("mock_value2");
        
        String result = obj.getValue1();
        assertEquals("mock_value2", result);
    }
}

内容的提问来源于stack exchange，提问作者Tom