通过接口抽象文件操作并使用mock实现，可有效解耦IO依赖，提升Go单元测试的可靠性与速度。
在Go语言中进行单元测试时，如果遇到文件IO操作，直接读写真实文件会带来依赖问题，影响测试的可重复性和速度。最佳做法是通过接口抽象文件操作，并在测试中使用模拟（mock）或内存中的数据替代真实IO。下面是一个实用示例，展示如何对涉及文件读写的函数进行单元测试。
定义文件操作接口
为了便于测试，先将文件操作抽象成一个接口：
type FileReader interface {
ReadFile(filename string) ([]byte, error)
}
// 实现真实文件读取
type RealFileReader struct{}
func (r RealFileReader) ReadFile(filename string) ([]byte, error) {
return os.ReadFile(filename)
}
假设我们有一个函数，它依赖读取JSON配置文件并返回结构体：
type Config struct {
Host string `json:"host"`
Port int `json:"port"`
}
func LoadConfig(reader FileReader, filename string) (*Config, error) {
data, err := reader.ReadFile(filename)
if err != nil {
return nil, err
}
var config Config
if err := json.Unmarshal(data, &config); err != nil {
return nil, err
}
return &config, nil
}
编写模拟实现用于测试
在测试中，我们不希望真正读取磁盘文件，可以创建一个模拟的 FileReader：
立即学习“go语言免费学习笔记（深入）”；
type MockFileReader struct {
Data []byte
Err error
}
func (m MockFileReader) ReadFile(filename string) ([]byte, error) {
return m.Data, m.Err
}
编写单元测试
使用 mock 来测试 LoadConfig 函数的各种情况：
func TestLoadConfig_Success(t *testing.T) {
jsonData := `{"host": "localhost", "port": 8080}`
mockReader := MockFileReader{Data: []byte(jsonData)}
config, err := LoadConfig(mockReader, "config.json") // 文件名仅作占位
if err != nil {
t.Fatalf("Expected no error, got %v", err)
}
if config.Host != "localhost" || config.Port != 8080 {
t.Errorf("Expected localhost:8080, got %s:%d", config.Host, config.Port)
}
}
func TestLoadConfig_FileNotFound(t *testing.T) {
mockReader := MockFileReader{Err: os.ErrNotExist}
_, err := LoadConfig(mockReader, "missing.json")
if err == nil {
t.Fatal("Expected error, got nil")
}
if !errors.Is(err, os.ErrNotExist) {
t.Errorf("Expected os.ErrNotExist, got %v", err)
}
}
func TestLoadConfig_InvalidJSON(t *testing.T) {
mockReader := MockFileReader{Data: []byte("{invalid json}")}
_, err := LoadConfig(mockReader, "bad.json")
if err == nil {
t.Fatal("Expected unmarshal error")
}
}
这样就完全解耦了文件IO和业务逻辑，测试快速、可靠，无需准备真实文件或清理临时目录。
小结：关键点
- 用接口隔离文件IO，提升可测试性
- mock 返回值可覆盖成功、失败、格式错误等场景
- 避免在单元测试中使用 os.Create 或 ioutil.WriteFile 操作真实文件系统
- 若必须操作临时文件，可用 os.CreateTemp 并在测试结束时删除
基本上就这些。通过接口+mock的方式，既能保证逻辑正确，又能避免外部依赖带来的不确定性。
