测试文件读写逻辑的关键是用内存模拟而非真实磁盘操作:用 bytes.Buffer 替代 io.Writer、strings.Reader 模拟 io.Reader、afero.MemMapFs 构建虚拟文件系统,仅在必要时用 TempDir 测试真实 I/O。

测试文件读写逻辑的关键,不是真去碰磁盘,而是用内存模拟文件系统或注入可控制的输入输出流。
用 bytes.Buffer 替代文件写入
当函数接收 io.Writer(比如 *os.File)时,可直接传入 bytes.Buffer。它实现了 io.Writer 接口,所有写入内容都存在内存里,方便断言。
- 不用创建临时文件,不依赖路径权限,测试快且隔离
- 写完后调用
buf.String()或buf.Bytes()检查内容是否符合预期 - 示例:若函数是
WriteConfig(w io.Writer, cfg Config),测试时传&bytes.Buffer{}
用 strings.Reader 模拟文件读取
当函数需要从 io.Reader 读数据(如解析配置),可用 strings.Reader 包装测试用的字符串。
- 避免读真实文件,也不需提前生成测试文件
- 支持多次读、带错误模拟(配合
io.MultiReader或自定义 reader) - 示例:传
strings.NewReader(`{"name":"test"}`)给 JSON 解析函数
用 testify/fs 或 afero 做虚拟文件系统
当代码强依赖 os.Open、os.Stat、os.WriteFile 等具体文件操作,且难以改造成接口时,推荐引入虚拟文件系统。
立即学习“go语言免费学习笔记(深入)”;
-
afero提供afero.MemMapFs—— 全内存文件系统,支持目录、权限、修改时间等模拟 - 把原函数依赖的
os操作封装成可注入的fs.Fs参数,测试时换为内存版 - 比手动 mock
os函数更稳定,也更容易覆盖路径不存在、权限拒绝等边界情况
慎用 TempDir,仅用于必须测真实 I/O 的场景
只有当测试目标本身就是“与磁盘交互的稳定性”(如大文件分块写、原子重命名)时,才用 os.MkdirTemp 创建临时目录。
- 务必在
defer os.RemoveAll(dir)清理,避免残留 - 注意跨平台路径分隔符(用
filepath.Join)、Windows 权限限制 - 这类测试应标记为
// +build integration或单独归类,不放入常规单元测试流程
基本上就这些。核心思路是:让被测函数面向接口(io.Reader/io.Writer/fs.Fs)而非具体类型,测试时用轻量、可控的实现替换,既准又快。










