本文介绍在 Mockito 测试中，如何根据 TypeReference 的实际泛型类型（如 Map 或 Map）精准匹配并返回不同的模拟结果，解决因类型擦除导致的 mock 冲突问题。
本文介绍在 mockito 测试中，如何根据 `typereference
在 Java 单元测试中，当被测方法接收 TypeReference
根本解法是放弃静态类型匹配，转而利用 thenAnswer 在运行时检查真实泛型信息。TypeReference 是一个抽象类，其子类通过匿名内部类保留了完整的泛型签名（通过 getType() 获取 java.lang.reflect.Type）。我们可据此做精确判断：
@Test
void testGetConfigurationWithDifferentTypeReferences() {
ObjectMapper mapper = mock(ObjectMapper.class);
ConfigService service = new ConfigService(mapper); // 假设被测服务持有 mapper
when(mapper.readValue(eq(SDK_BYTES), any(TypeReference.class)))
.thenAnswer(invocation -> {
TypeReference> typeRef = invocation.getArgument(1);
Type type = typeRef.getType();
// 关键：通过 getType().getTypeName() 或更健壮的 instanceof + getActualTypeArguments 判断
if (type instanceof ParameterizedType paramType) {
Class> rawType = (Class>) paramType.getRawType();
if (rawType == Map.class && paramType.getActualTypeArguments().length == 2) {
Type keyType = paramType.getActualTypeArguments()[0];
Type valueType = paramType.getActualTypeArguments()[1];
// 匹配 Map<string list>>
if (keyType.equals(String.class)
&& valueType instanceof ParameterizedType vt
&& ((Class>) vt.getRawType()) == List.class
&& vt.getActualTypeArguments()[0].equals(String.class)) {
return Map.of("feature1", List.of("a", "b"));
}
// 匹配 Map<string featurelaunchstatus>
else if (keyType.equals(String.class)
&& valueType == FeatureLaunchStatus.class) {
return Map.of("feature2", new FeatureLaunchStatus(true));
}
}
}
throw new IllegalArgumentException("Unsupported TypeReference: " + type.getTypeName());
});
// 测试场景1：期望返回 Map<string list>>
TypeReference<map list>>> listMapRef = new TypeReference() {};
Map<string list>> result1 = service.getConfiguration("config", listMapRef);
assertThat(result1).containsEntry("feature1", List.of("a", "b"));
// 测试场景2：期望返回 Map<string featurelaunchstatus>
TypeReference<map featurelaunchstatus>> statusMapRef = new TypeReference() {};
Map<string featurelaunchstatus> result2 = service.getConfiguration("config", statusMapRef);
assertThat(result2).containsEntry("feature2", new FeatureLaunchStatus(true));
}</string></map></string></string></map></string></string></string>
⚠️ 注意事项：
-
避免过度依赖 getTypeName() 字符串匹配（如 "java.util.Map
>"），它易受包名变更、类型别名等影响，应优先使用 instanceof ParameterizedType + getRawType()/getActualTypeArguments() 进行结构化校验； - 确保 TypeReference 子类为匿名类（即 new TypeReference<...>() {}），否则泛型信息会在编译期丢失；
- 若逻辑复杂度持续上升（如需支持十余种类型），建议改用轻量 Fake 实现替代 Mockito 模拟，例如自定义 FakeObjectMapper，既提升可读性，又规避反射边界风险；
- 此方案适用于 Jackson 的 TypeReference，其他库（如 Gson 的 TypeToken）需调整类型检查逻辑，但 thenAnswer 思路完全通用。
总结：面对泛型类型驱动的行为分支，thenAnswer 提供了运行时参数感知能力，是突破类型擦除限制的可靠方案。关键在于善用 TypeReference.getType() 返回的完整类型描述，并以结构化方式解析，而非字符串匹配。
