如何在单元测试中正确模拟向迭代器遍历过程中的 List 添加对象

冬伟小哥_4403

冬伟小哥_4403

2026-05-14

563人浏览

原创

如何在单元测试中正确模拟向迭代器遍历过程中的 List 添加对象

本文详解如何在 Mockito 单元测试中正确模拟 List.add() 行为(尤其在 Iterator 遍历循环内),避免因错误地 mock ArrayList 实例导致断言失败,并指出关键误区:不应 mock 列表本身,而应确保真实列表被填充,或精准 stub add() 方法匹配任意目标类型。

本文详解如何在 mockito 单元测试中正确模拟 `list.add()` 行为(尤其在 `iterator` 遍历循环内),避免因错误地 mock 列表本身导致断言失败,并指出关键误区:不应 mock 列表本身,而应确保真实列表被填充,或精准 stub `add()` 方法匹配任意目标类型。

在测试类似 getUsers() 这类依赖外部服务(如 DynamoDB)并内部构建 List 的方法时,一个常见误区是:对 ArrayList 实例进行 mock(如 mock(ArrayList.class)),期望它能“记住”添加的元素用于后续断言。但这是反模式——mock 对象不会执行真实逻辑,add() 调用默认返回 false(除非显式 stub),且 get(0) 等操作将抛出 UnsupportedOperationException 或返回 null,导致 assertSame 断言必然失败。

✅ 正确做法是:使用真实列表(new ArrayList())作为被测方法内部容器,仅 mock 外部依赖(DynamoDB 相关组件),并确保 Util.jsonToObject(...) 在测试中能返回可控对象(可通过 @MockitoSettings(strictness = Strictness.LENIENT) 或直接 spy/替换工具类实现)。若必须 stub add()(例如验证调用行为),应使用 doReturn(true).when(realList).add(any(Users.class)) —— 但注意:这仅验证调用,不改变列表状态;真正需要断言内容时,必须让 jsonToObject 返回有效实例。

以下是推荐的、可运行的测试结构(基于 JUnit 5 + Mockito 4+):

@ExtendWith(MockitoExtension.class)
class DynamoDBServiceTest {

    @Mock
    private DynamoDB dynamoDB;

    @Mock
    private Table table;

    @Mock
    private Index index;

    @Mock
    private ItemCollection<queryoutcome> itemCollection;

    @Mock
    private Iterator<item> iterator; // 使用标准 Iterator,非 IteratorSupport

    @Mock
    private Item item;

    private DynamoDBService dynamoDBService;

    @BeforeEach
    void setUp() {
        // 真实初始化被测服务(注入 mock 的 dynamoDB)
        dynamoDBService = new DynamoDBService(dynamoDB);

        // 构建 mock 链
        when(dynamoDB.getTable("users")).thenReturn(table);
        when(table.getIndex("userIndex")).thenReturn(index);
        when(index.query(any(QuerySpec.class))).thenReturn(itemCollection);
        when(itemCollection.iterator()).thenReturn(iterator);

        // 模拟迭代器行为:一次 hasNext → true,第二次 → false
        when(iterator.hasNext()).thenReturn(true, false);
        when(iterator.next()).thenReturn(item);
        when(item.toJSONPretty()).thenReturn("{\"id\":\"1\",\"name\":\"Alice\"}");
    }

    @Test
    void getUsers_returnsPopulatedList() {
        // Arrange: 准备期望的 Users 对象(需确保 Util.jsonToObject 能解析)
        Users expectedUser = new Users("1", "Alice", null, "");

        // ⚠️ 关键:不要 mock ArrayList!让方法内部创建真实列表
        // 我们只需确保 jsonToObject 返回预期对象(此处假设 Util 是静态工具类)
        // 若 Util 不可测,建议将其抽象为可注入的依赖(如 JsonConverter 接口)

        // Act
        List<users> result = dynamoDBService.getUsers("test-id");

        // Assert
        assertThat(result).hasSize(1);
        assertThat(result.get(0)).usingRecursiveComparison()
            .isEqualTo(expectedUser);
    }
}</users></item></queryoutcome>

? 重要注意事项:

  • 永远避免 mock(ArrayList.class):它破坏了 List 的契约,使 get()、size() 等方法不可用,违背测试真实性。
  • doReturn().when(list).add(...) 仅用于验证调用,不替代数据构造:若想验证“是否调用了 add”,可用 verify(list).add(any(Users.class));但若要断言返回列表内容,请确保 jsonToObject 返回真实对象。
  • Util.jsonToObject 是测试盲点:当前代码中它未被 mock 或控制,会导致测试不稳定。最佳实践是将其封装进接口(如 JsonConverter),并在测试中注入 mock(JsonConverter.class) 并 stub convert(..., Users.class)。
  • 优先使用 @Mock 和 @InjectMocks 注解:提升可读性与维护性,减少手动 mock() 和 when() 嵌套。
  • doReturn(...).when(...) vs when(...).thenReturn(...):当 mock 对象方法有副作用(如 void 方法或 final 方法)时,doReturn 是唯一选择;对普通方法二者等效,但 doReturn 更安全(避免提前触发 real method)。

总结:模拟的核心是「隔离外部依赖」,而非「模拟内部数据结构」。让业务逻辑在真实集合上运行,只对 I/O、网络、第三方工具类等不可控部分进行 mock —— 这才是可靠、可维护的单元测试之道。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

10337

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

7462

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

6552

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1124

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

928

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1356

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

2729

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

20031

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1215

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.4万人学习