Java测试中如何Mock某类所有实例的指定方法？

当然可以做到！不过因为你的methodA方法里是直接通过new MyClass()创建实例，标准的Mockito没法直接拦截这种构造调用，这里有两种常见的解决方案：

方案一：使用PowerMock拦截构造函数调用 #

PowerMock可以配合Mockito一起，拦截类内部的构造函数调用，让我们返回预先定义好的Mock对象。

步骤1：引入PowerMock依赖 #

如果用Maven，在pom.xml中添加以下依赖（版本可根据实际情况调整）：

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

步骤2：编写测试代码 #

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.Mockito.when;
import static org.junit.Assert.assertEquals;

@RunWith(PowerMockRunner.class)
// 必须指定包含new语句的类（也就是A类），PowerMock需要修改它的字节码来拦截构造调用
@PrepareForTest(A.class)
public class ATest {

    @Test
    public void testMethodA() throws Exception {
        // 1. 创建MyClass的Mock对象
        MyClass mockMyClass = PowerMockito.mock(MyClass.class);
        
        // 2. 拦截MyClass的构造函数调用：无论传入什么int参数，都返回我们的Mock对象
        PowerMockito.whenNew(MyClass.class).withAnyArguments().thenReturn(mockMyClass);
        
        // 3. 定义Mock对象的method方法行为：调用时返回"abcd"
        when(mockMyClass.method()).thenReturn("abcd");
        
        // 4. 实例化被测试类并调用方法
        A testInstance = new A();
        String result = testInstance.methodA(1, 2);
        
        // 5. 断言结果（这里根据你的methodA实际返回值调整）
        assertEquals("result", result);
        
        // 可选：验证构造函数和method方法是否被正确调用
        PowerMockito.verifyNew(MyClass.class).withArguments(1, 2);
        PowerMockito.verify(mockMyClass).method();
    }
}

关键说明 #

• @PrepareForTest(A.class)：这是核心配置，告诉PowerMock要处理A类的字节码，因为A类内部有new MyClass()的调用。
• whenNew(MyClass.class)：PowerMock特有的API，用来拦截构造函数，返回我们的Mock对象。你也可以用withArguments(1,2)来匹配特定参数的构造调用。

方案二：重构代码，通过工厂方法解耦实例创建 #

如果不想引入PowerMock这类“重型”工具，更推荐的方式是重构代码，把实例创建的逻辑抽成独立的方法，这样用标准Mockito就能轻松Mock。

步骤1：重构A类 #

把new MyClass()的逻辑抽成一个protected方法，方便测试时Mock（顺便修正了你原代码里的变量名重复问题）：

public class A { 
    public String methodA(int a, int b){ 
        MyClass object = createMyClass(a, b); 
        String value = object.method(); 
        return "result"; 
    }
    
    // 抽成protected方法，供测试时Mock
    protected MyClass createMyClass(int a, int b) {
        return new MyClass(a, b);
    }
}

步骤2：编写测试代码 #

用Mockito的spy来Mock这个工厂方法：

import org.junit.Test;
import static org.mockito.Mockito.*;
import static org.junit.Assert.assertEquals;

public class ATest {

    @Test
    public void testMethodA() {
        // 1. 创建A类的Spy对象（保留原有逻辑，只Mock特定方法）
        A aSpy = spy(new A());
        
        // 2. 创建MyClass的Mock对象
        MyClass mockMyClass = mock(MyClass.class);
        
        // 3. 当调用createMyClass方法时，返回我们的Mock对象
        doReturn(mockMyClass).when(aSpy).createMyClass(anyInt(), anyInt());
        
        // 4. 定义Mock对象的method方法行为
        when(mockMyClass.method()).thenReturn("abcd");
        
        // 5. 调用被测试方法
        String result = aSpy.methodA(1, 2);
        
        // 6. 断言结果
        assertEquals("result", result);
        
        // 可选：验证方法调用
        verify(aSpy).createMyClass(1, 2);
        verify(mockMyClass).method();
    }
}

方案对比 #

• PowerMock方案：不需要修改原有代码，但引入了额外依赖，且因为修改字节码，可能和某些测试框架或工具存在兼容性问题。
• 重构方案：更符合“测试友好”的代码设计，不需要额外依赖，但需要调整原有代码结构。

内容的提问来源于stack exchange，提问作者lubo-pisk