Go语言测试策略：如何优雅地模拟ioutil.ReadFile

package main

import (
    "bytes"
    "fmt"
    "io"
    "io/ioutil" // 在Go 1.16+中，ioutil.ReadAll 推荐使用 io.ReadAll
)

// ObtainTranslationStringsFileChoice1 修改了函数签名，接受一个io.Reader
// 这样在测试时可以传入一个模拟的io.Reader
func ObtainTranslationStringsFileChoice1(rdr io.Reader) ([]string, error) {
    if contents, err := ioutil.ReadAll(rdr); err == nil {
        // 实际应用中，这里可能会对contents进行进一步处理
        // 示例中简化为直接返回一个包含内容的切片
        return []string{string(contents)}, nil
    } else {
        return nil, err
    }
}

func main() {
    // 在测试中，可以使用bytes.Buffer模拟文件内容
    payload := "Hello, Mocking World with io.Reader!"
    mockReader := bytes.NewBufferString(payload)

    result, err := ObtainTranslationStringsFileChoice1(mockReader)
    if err != nil {
        fmt.Printf("Error: %v\n", err)
        return
    }
    fmt.Printf("方法一结果: %#v\n", result)
}

package main

import (
    "bytes"
    "fmt"
    "io/ioutil"
)

// ReadFile 类型定义，与ioutil.ReadFile签名一致
type ReadFile func(filename string) ([]byte, error)

// myReadFile 是一个包级别的函数变量，初始化为ioutil.ReadFile
// 在测试时，可以将其重新赋值为模拟函数
var myReadFile = ioutil.ReadFile

// FakeReadFiler 结构体，用于实现一个模拟的ReadFile方法
type FakeReadFiler struct {
    Str string
}

// ReadFile 方法与ioutil.ReadFile签名一致，用于模拟
func (f FakeReadFiler) ReadFile(filename string) ([]byte, error) {
    // 忽略filename参数，直接返回预设的字符串内容
    return []byte(f.Str), nil
}

// ObtainTranslationStringsFileChoice2 使用了可替换的myReadFile变量
func ObtainTranslationStringsFileChoice2(path string) ([]string, error) {
    if contents, err := myReadFile(path); err == nil {
        return []string{string(contents)}, nil
    } else {
        return nil, err
    }
}

func main() {
    // 保存原始的myReadFile，以便测试后恢复，避免影响其他测试
    originalReadFile := myReadFile
    defer func() {
        myReadFile = originalReadFile // 确保测试结束后恢复原始函数
    }()

    // 在测试中，重新赋值myReadFile为模拟函数
    payload := "Mocked content for method 2 using function variable!"
    fakeFiler := FakeReadFiler{Str: payload}
    myReadFile = fakeFiler.ReadFile // 将包变量指向模拟方法

    // 调用被测试函数，路径参数在这里不再重要，因为myReadFile已被替换
    result, err := ObtainTranslationStringsFileChoice2("/dev/null")
    if err != nil {
        fmt.Printf("Error: %v\n", err)
        return
    }
    fmt.Printf("方法二结果: %#v\n", result)
}

package main

import (
    "fmt"
    "io/ioutil"
    "os"
)

// FileSystem 定义了文件系统操作的接口
type FileSystem interface {
    ReadFile(filename string) ([]byte, error)
    WriteFile(filename string, data []byte, perm os.FileMode) error
    // ... 其他文件操作方法，如MkdirAll, Stat等
}

// RealFileSystem 实现了 FileSystem 接口，使用标准库进行真实文件操作
type RealFileSystem struct{}

func (r RealFileSystem) ReadFile(filename string) ([]byte, error) {
    return ioutil.ReadFile(filename) // 或 os.ReadFile(filename)
}

func (r RealFileSystem) WriteFile(filename string, data []byte, perm os.FileMode) error {
    return ioutil.WriteFile(filename, data, perm) // 或 os.WriteFile(filename, data, perm)
}

// MockFileSystem 实现了 FileSystem 接口，用于测试，将文件存储在内存中
type MockFileSystem struct {
    Files map[string][]byte
}

func (m MockFileSystem) ReadFile(filename string) ([]byte, error) {
    if content, ok := m.Files[filename]; ok {
        return content, nil
    }
    return nil, os.ErrNotExist // 模拟文件不存在错误
}

func (m MockFileSystem) WriteFile(filename string, data []byte, perm os.FileMode) error {
    m.Files[filename] = data
    return nil
}

// Service 结构体依赖于 FileSystem 接口
type MyService struct {
    FS FileSystem
}

func (s *MyService) ProcessFile(path string) ([]string, error) {
    contents, err := s.FS.ReadFile(path)
    if err != nil {
        return nil, err
    }
    return []string{string(contents)}, nil
}

func main() {
    // 使用真实文件系统
    realService := &MyService{FS: RealFileSystem{}}
    // result, err := realService.ProcessFile("path/to/real/file.txt")

    // 使用模拟文件系统进行测试
    mockFS := MockFileSystem{
        Files: map[string][]byte{
            "/app/config.json": []byte(`{"key": "value"}`),
            "/app/data.txt":    []byte("some data"),
        },
    }
    mockService := &MyService{FS: mockFS}

    result, err := mockService.ProcessFile("/app/config.json")