使用 PowerMockito 模拟 new A() 行为

使用 PowerMockito 模拟 new A() 行为

在 Java 测试中,当你想模拟 new A() 实例化的对象行为时,可以使用 PowerMockito。下面是一个完整的示例:

完整实现方案

复制代码
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.Mockito;
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;

// 1. 使用PowerMockRunner
@RunWith(PowerMockRunner.class)
// 2. 准备要修改字节码的类
@PrepareForTest(YourClass.class) // 包含对new A()调用的类
public class YourClassTest {

    @Mock
    private A mockA; // Mock对象

    @Test
    public void testYourMethod() throws Exception {
        // 3. 设置mock对象的返回值
        Mockito.when(mockA.count()).thenReturn(15); // 模拟返回15
        
        // 4. 模拟new A()操作
        PowerMockito.whenNew(A.class).withNoArguments().thenReturn(mockA);
        
        // 5. 创建被测类实例
        YourClass yourClass = new YourClass();
        
        // 6. 执行测试
        int result = yourClass.yourMethod();
        
        // 7. 验证结果
        assertEquals(15, result);
        
        // 可选:验证方法是否被调用
        Mockito.verify(mockA).count();
    }
}

class A {
    public int count() {
        return 0; // 实际实现
    }
}

class YourClass {
    public int yourMethod() {
        A a = new A(); // 需要被模拟的实例化
        return a.count();
    }
}

Maven 依赖配置

复制代码
<dependencies>
    <!-- JUnit -->
    <dependency>
        <groupId>junit</groupId>
        <artifactId>junit</artifactId>
        <version>4.13.2</version>
        <scope>test</scope>
    </dependency>
    
    <!-- Mockito Core -->
    <dependency>
        <groupId>org.mockito</groupId>
        <artifactId>mockito-core</artifactId>
        <version>5.3.1</version>
        <scope>test</scope>
    </dependency>
    
    <!-- PowerMock -->
    <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>
</dependencies>

关键步骤解析

  1. 测试类配置​:

    • @RunWith(PowerMockRunner.class):使用PowerMock运行器
    • @PrepareForTest(YourClass.class):指定需要修改字节码的类
  2. 模拟构造函数​:

    复制代码
    PowerMockito.whenNew(A.class)   // 拦截对A类的实例化
        .withNoArguments()           // 使用无参构造函数
        .thenReturn(mockA);          // 返回模拟对象
  3. 设置方法返回值​:

    复制代码
    Mockito.when(mockA.count())  // 当调用count()方法时
           .thenReturn(15);      // 返回指定值

实际效果说明

yourMethod() 中的 new A() 执行时:

  1. PowerMock 会拦截构造函数调用
  2. 返回预先创建的 mockA 实例
  3. 调用 count() 方法时返回预设值 (15)

这样既不需要修改被测试类的代码,也能完全控制新创建对象的行为。

注意事项

  1. 测试效率:PowerMock 修改字节码,会使测试执行变慢
  2. 兼容性:确保 PowerMock 版本与 Mockito/JUnit 版本兼容
  3. 替代方案:优先考虑重构代码使用依赖注入
  4. 范围限制 :只能模拟 new 表达式直接出现的类

对于简单场景,推荐优先重构代码使用依赖注入。但当处理遗留代码或第三方库时,PowerMock 提供的这种能力非常有用。

相关推荐
老朋友此林15 小时前
MongoDB GEO 项目场景 ms-scope 实战
java·数据库·spring boot·mongodb
豆沙沙包?16 小时前
2025年--Lc165--H637.二叉树的层平均值(二叉树的层序遍历)--Java版
java·开发语言
迎風吹頭髮16 小时前
UNIX下C语言编程与实践40-UNIX 全局跳转:setjmp 与 longjmp 函数的使用与注意事项
java·c语言·unix
春风霓裳16 小时前
maven-setting配置
java·maven
小蒜学长16 小时前
springboot二手儿童绘本交易系统设计与实现(代码+数据库+LW)
java·开发语言·spring boot·后端
wangwangmoon_light18 小时前
0.0 编码基础模板
java·数据结构·算法
Terio_my18 小时前
Spring Boot 热部署配置与自定义排除项
java·spring boot·后端
JAVA学习通19 小时前
微服务项目->在线oj系统(Java-Spring)--C端用户(超详细)
java·开发语言·spring
计算机毕业设计小帅19 小时前
【2026计算机毕业设计】基于jsp的毕业论文管理系统
java·开发语言·毕业设计·课程设计