如何在Dio单元测试中模拟拦截器添加操作
理解问题背景
在使用Dio进行HTTP请求时,拦截器(interceptors)是一个非常重要的功能,它允许我们在请求发出前或响应返回后对数据进行处理。在实际开发中,我们经常需要为Dio实例添加各种拦截器来实现诸如添加请求头、日志记录等功能。
然而,当我们在单元测试中尝试模拟(Mock)这种行为时,可能会遇到FakeUsedError错误,提示我们没有为interceptors属性设置存根(stub)。这是因为Mockito默认会为未存根的方法调用创建"fake"对象,而当这些fake对象被实际访问时就会抛出异常。
问题重现分析
假设我们有一个电影数据源的实现类,其中包含一个通过Dio获取电影列表的方法。在这个方法中,我们为Dio实例添加了一个自定义的请求头拦截器:
class MovieRemoteDataSourceImpl implements MovieDataSource {
final Dio dio;
MovieRemoteDataSourceImpl(this.dio);
@override
Future<Either<Exception, List<MovieResponse>>> fetchMovies() async {
const url = Endpoint.getTopRatedMovies;
try {
final addHeaderInterceptor = AddHeaderInterceptor();
dio.interceptors.add(addHeaderInterceptor); // 这里会在测试中出错
final response = await dio.get(url);
// ...处理响应数据
return Right(result);
} on Exception catch (e) {
return Left(e);
}
}
}
在对应的单元测试中,我们使用Mockito创建了Dio的模拟实例,并设置了get方法的存根,但没有处理interceptors属性:
@GenerateNiceMocks([MockSpec<Dio>()])
void main() {
group('Fetch list of movie.', () {
MockDio mockDioClient = MockDio();
final dataSource = MovieRemoteDataSourceImpl(mockDioClient);
test('Fetch success', () async {
when(mockDioClient.get(any)).thenAnswer((_) async => mockResponseSuccess);
final actual = await dataSource.fetchMovies();
expect(actual.isRight(), true);
});
});
}
运行测试时会抛出FakeUsedError,因为测试代码访问了未存根的interceptors属性。
解决方案
要解决这个问题,我们需要为Dio的interceptors属性设置存根。具体来说,我们需要:
- 创建一个真实的
Interceptors实例 - 使用Mockito的
whenAPI告诉模拟Dio对象返回这个实例
@GenerateNiceMocks([MockSpec<Dio>()])
void main() {
group('Fetch list of movie.', () {
MockDio mockDioClient = MockDio();
// 创建真实的拦截器列表
final fakeInterceptors = Interceptors();
// 设置interceptors属性的存根
when(mockDioClient.interceptors).thenReturn(fakeInterceptors);
final dataSource = MovieRemoteDataSourceImpl(mockDioClient);
test('Fetch success', () async {
when(mockDioClient.get(any)).thenAnswer((_) async => mockResponseSuccess);
final actual = await dataSource.fetchMovies();
expect(actual.isRight(), true);
});
});
}
深入理解
为什么需要这样做?因为Dio的interceptors属性实际上是一个Interceptors类型的对象,而当我们调用dio.interceptors.add()时,实际上是在这个Interceptors实例上调用add方法。
在单元测试中,我们使用Mockito创建的是Dio的模拟实例,默认情况下,所有未存根的方法和属性都会返回null或抛出异常。通过显式地为interceptors属性设置存根,我们确保测试能够正常执行拦截器的添加操作,而不会因为访问未存根的属性而失败。
最佳实践
-
隔离测试:确保每个测试只关注一个功能点,对于拦截器的测试应该与HTTP请求的测试分开
-
验证拦截器行为:如果需要验证拦截器是否被正确添加,可以使用Mockito的
verifyAPI:
test('should add header interceptor', () async {
when(mockDioClient.get(any)).thenAnswer((_) async => mockResponseSuccess);
await dataSource.fetchMovies();
verify(mockDioClient.interceptors.add(any)).called(1);
});
-
考虑使用真实实例:对于简单的拦截器,有时使用真实的
Interceptors实例比完全模拟更简单有效 -
清理拦截器:在测试之间,记得清理拦截器列表,避免测试间的相互影响
总结
在Dio的单元测试中模拟拦截器添加操作时,关键是要理解Dio内部如何管理拦截器。通过为interceptors属性设置适当的存根,我们可以确保测试能够顺利执行,同时保持测试的隔离性和可维护性。记住,良好的单元测试应该既验证代码的正确性,又保持简单和可读性。
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0193
cann-learning-hubCANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。Jupyter Notebook0121
MiMo-V2.5-Pro-FP4-DFlashMiMo-V2.5-Pro-FP4-DFlash 是驱动 MiMo-V2.5-Pro-UltraSpeed 的底层模型: FP4 量化骨干网络:对 MoE 专家采用 MXFP4 量化,同时保持模型其他部分的更高精度,在几乎无损质量的前提下,显著减小模型体积并降低内存带宽压力。 BF16 DFlash 草稿生成器:用于块扩散推测解码,每次前向传播可生成一整个块的 tokens,并让骨干网络一步完成验证。 两者协同作用,既降低了每参数的位宽,又减少了骨干网络前向传播的次数,而这两者正是万亿参数模型解码过程中的两大主要成本来源。Python00
JoyAI-EchoJoyAI-Echo,这是一个独立的、仅用于推理的版本,旨在实现分钟级多镜头音视频生成。它采用了经过蒸馏的DMD生成器、配对的跨模态记忆以及故事级别的一致性。其性能的核心在于,一个跨模态视听记忆库能够在长达五分钟的视频中保持角色外观和语音音色的一致性。同时,一个训练后处理流程将基于记忆的强化学习与分布匹配蒸馏相结合,实现了7.5倍的速度提升,显著增强了视觉质量和对齐效果。00
AstrBot✨ 易上手的多平台 LLM 聊天机器人及开发框架 ✨ 平台支持 QQ、QQ频道、Telegram、微信、企微、飞书 | OpenAI、DeepSeek、Gemini、硅基流动、月之暗面、Ollama、OneAPI、Dify 等。附带 WebUI。Python05
handy-ollama动手学Ollama,CPU玩转大模型部署,在线阅读地址:https://datawhalechina.github.io/handy-ollama/Jupyter Notebook05