Mockito关于同一个方法多次执行根据参数不同来mock不同的返回值

 


前言

今天写项目UT的时候遇到需要根据同一个方法多次执行根据参数不同来mock不同的返回值,找到解决方案,于此记录。

 

一、场景

相同的方法根据参数不同来mock不同的返回值
Date maxDate = summaryMapper.getMaxDATE(condition, "MONTH");
Date maxDate = summaryMapper.getMaxDATE(condition, "WEEK");
Date maxDate = summaryMapper.getMaxDATE(condition, "DAY");

这里原代码是根据三个相同的方法不同的日期type找出最大的日期,写UT的话则需要对此方法mock三个相应的返回值。

二、代码示例

这里我是用的是Mockito中的Answer,根据参数列表里面传的值来确定返回值,参数列表下标从0开始

when(summaryMapper.getMaxDATE(anyString(), anyString())).thenAnswer(mock -> {
    if ("MONTH".equals(mock.getArgument(1))) {
        return this.addDate(new Date(), -2);
    }
    if ("WEEK".equals(mock.getArgument(1))) {
        return this.addDate(new Date(), -3);
    }
    if ("DAY".equals(mock.getArgument(1))) {
        return this.addDate(new Date(), -3);
    }
    return null;
});

 


总结

问题看似简单,但是找到解决方案的过程是费时费力的,有用的话点个赞再走。

 

### Java 中使用 Mock 框架模拟静态方法的行为 在 Java 开发中,当需要对静态方法进行单元测试时,通常会借助一些强大的 Mock 框架,比如 Mockito 和 PowerMock。以下是两种主流方式的具体实现。 #### 方法一:使用 PowerMockito 模拟静态方法 PowerMock 是一个扩展工具库,它允许开发者修改字节码以支持对静态方法、私有方法以及构造函数的 Mock。下面是一个完整的示例: ```java import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.Mockito; import org.powermock.api.mockito.PowerMockito; import org.powermock.core.classloader.annotations.PrepareForTest; import org.powermock.modules.junit4.PowerMockRunner; @RunWith(PowerMockRunner.class) @PrepareForTest(StaticClass.class) // 声明要被 Mock 的类 public class StaticMethodTest { @Test public void testStaticMethod() throws Exception { // 配置静态方法的模拟行为 PowerMockito.mockStatic(StaticClass.class); Mockito.when(StaticClass.someStaticMethod()).thenReturn("mockedValue"); // 调用依赖于静态方法的代码并验证其行为 MyClass myClass = new MyClass(); String result = myClass.methodUnderTest(); // 断言结果是否符合预期 org.junit.Assert.assertEquals("mockedValue", result); // 验证静态方法是否被正确调用 PowerMockito.verifyStatic(StaticClass.class); // 确认静态方法确实被执行过 StaticClass.someStaticMethod(); } } ``` 此代码片段展示了如何通过 `PowerMockito` 来模拟 `StaticClass` 类中的静态方法,并对其进行断言和验证[^1]。 --- #### 方法二:使用 Mockito (3.4.0 及以上版本) 模拟静态方法Mockito 3.4.0 版本起,原生支持对静态方法Mock 功能,无需额外引入其他库。下面是具体的实现方式: ```java import static org.mockito.Mockito.*; import org.junit.jupiter.api.AfterEach; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; public class MathUtilsTest { @BeforeEach public void setUp() { mockStatic(MathUtils.class); // 初始化静态方法Mock } @AfterEach public void tearDown() { close(); // 关闭静态 Mock,释放资源 } @Test public void testAddWithParameterMatching() { // 使用参数匹配器配置静态方法的行为 when(MathUtils.add(anyInt(), anyInt())).thenAnswer(invocation -> { int a = invocation.getArgument(0); int b = invocation.getArgument(1); return a * b; // 返回乘积而非实际加法的结果 }); // 测试不同输入下的返回值 org.junit.jupiter.api.Assertions.assertEquals(50, MathUtils.add(10, 5)); org.junit.jupiter.api.Assertions.assertEquals(18, MathUtils.add(2, 9)); // 验证静态方法是否按预期次数调用 verifyStatic(MathUtils.class, times(2)); // 确保 add 方法被调用了两次 MathUtils.add(anyInt(), anyInt()); } } ``` 在此例子中,`MathUtils` 类的静态方法 `add(int, int)` 被成功模拟,并且可以通过参数匹配器动态调整其行为[^2]。 --- ### 注意事项 - **版本要求**:如果选择使用 Mockito,则需确保其版本不低于 3.4.0。 - **线程安全性**:虽然 `mockStatic` 是线程安全的,但它的作用范围仅限于当前线程。因此,在多线程环境下务必小心处理。 - **资源管理**:每次完成测试后应显式调用 `.close()` 方法来清除静态 Mock 所占用的资源,防止潜在的内存泄漏问题。 - **验证机制**:利用 `verifyStatic` 或类似的 API 对静态方法的实际执行情况进行确认,从而提高测试覆盖度。 --- ### 总结 无论是采用功能全面的 PowerMock 还是轻量级的 Mockito,都可以有效地解决 Java 单元测试中关于静态方法难以直接操作的问题。具体选用哪种方案取决于项目需求和个人偏好。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值