如何使用Mockito为Elasticsearch聚合映射逻辑编写可维护的单元测试

Mock存根失效的核心原因 #

你之前的测试代码存在调用链路断连问题，并不是Mockito语法本身有问题：

• 你独立创建了Aggregations类型的Mock对象，但从未将该对象和SearchResponse实例做绑定：也就是缺失了when(searchResponse.getAggregations()).thenReturn(aggregations)这一步关键存根
• 你写的链式存根when(searchResponse.getAggregations().get("facet")).thenReturn(global)，实际执行逻辑是：每次调用searchResponse.getAggregations()时，Mockito都会为这个临时调用生成一个默认空Mock对象（所有方法返回值为null的代理），你之前手动创建的global、aggregations实例完全没有被纳入调用链路，自然没法让存根生效。

正确的多层聚合Mock实现方案 #

按照你实际的三层聚合读取逻辑，需要逐层完成Mock对象的依赖绑定，不能跳步：

基础绑定写法 #

// 初始化全链路需要的所有Mock实例
SearchResponse searchResponse = mock(SearchResponse.class);
Aggregations topLevelAggregations = mock(Aggregations.class);
Global globalFacetAgg = mock(Global.class);
Aggregations globalSubAggregations = mock(Aggregations.class);
Filter generalFilterAgg = mock(Filter.class);
Aggregations filterSubAggregations = mock(Aggregations.class);
Nested targetNestedAgg = mock(Nested.class);

// 逐层绑定调用链路，从最外层到最内层依次存根
when(searchResponse.getAggregations()).thenReturn(topLevelAggregations);
when(topLevelAggregations.get(FACETS)).thenReturn(globalFacetAgg);
when(globalFacetAgg.getAggregations()).thenReturn(globalSubAggregations);
when(globalSubAggregations.get(GENERAL_AGGREGATION)).thenReturn(generalFilterAgg);
when(generalFilterAgg.getAggregations()).thenReturn(filterSubAggregations);
when(filterSubAggregations.get(facetAttribute)).thenReturn(targetNestedAgg);

简化写法 #

如果不需要对中间层的聚合对象做单独的行为校验，可以用Mockito的深度存根模式简化代码，自动处理链式调用的中间Mock实例：

// 开启深度存根，自动生成链式调用路径上的中间Mock对象
SearchResponse searchResponse = mock(SearchResponse.class, RETURNS_DEEP_STUBS);
// 直接按照业务调用路径写最终存根，不需要手动绑定中间的Aggregations实例
when(searchResponse.getAggregations().get(FACETS)
        .getAggregations().get(GENERAL_AGGREGATION)
        .getAggregations().get(facetAttribute))
    .thenReturn(targetNestedAgg);

注意：深度存根模式仅适合快速写用例的场景，如果后续需要校验某层聚合的调用次数、入参，必须回到手动创建实例绑定的方式，避免深度存根导致的校验失效。

比纯Mock更低维护成本的替代方案 #

如果纯Mock写起来还是繁琐，可以不用依赖Stub JSON，直接用Elasticsearch提供的聚合类构造真实实例：

• 直接实例化Global、Filter、Nested对应的内部实现类，把各层聚合的子结果组装成InternalAggregations对象，直接传入SearchResponse
• 这种方式不需要维护冗长的JSON Stub，调整测试用例时只需要修改Java代码的构造参数，比解析JSON的维护成本低很多，同时比纯Mock更贴近真实运行的聚合返回结构。

内容的提问来源于stack exchange，提问作者J.N