Go语言中如何优雅地模拟 ioutil.ReadFile 进行单元测试

package main

import (
    "bytes"
    "fmt"
    "io"
    "io/ioutil" // 尽管 ReadFile 不直接使用，但 ReadAll 仍在此包中
)

// ObtainTranslationStringsFileChoice1 接受一个 io.Reader 接口，使得文件内容可模拟
func ObtainTranslationStringsFileChoice1(rdr io.Reader) ([]string, error) {
    // 从 io.Reader 中读取所有内容
    if contents, err := ioutil.ReadAll(rdr); err == nil {
        // 假设这里是对读取到的内容进行处理，简化为直接返回字符串切片
        return []string{string(contents)}, nil
    } else {
        return nil, err
    }
}

func main() {
    payload := "Hello, Go Mocking via Interface!"

    // 在测试环境中，我们可以使用 bytes.NewBufferString 来模拟文件内容
    mockReader := bytes.NewBufferString(payload)
    result1, err := ObtainTranslationStringsFileChoice1(mockReader)
    fmt.Printf("方法一结果: %#v, 错误: %#v\n", result1, err)

    // 在生产环境中，可以传入一个真实的 os.File 对象：
    // file, err := os.Open("path/to/real/file.txt")
    // if err != nil { /* handle error */ }
    // defer file.Close()
    // realResult, realErr := ObtainTranslationStringsFileChoice1(file)
}

package main

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

// ReadFileFunc 定义了 ioutil.ReadFile 的函数签名
type ReadFileFunc func(filename string) ([]byte, error)

// myReadFile 是一个包级变量，默认指向 ioutil.ReadFile
// 在测试中可以被重新赋值为模拟函数
var myReadFile ReadFileFunc = ioutil.ReadFile

// FakeReadFiler 结构体及其方法可以用于创建模拟的 ReadFile 实现
type FakeReadFiler struct {
    Content string
}

// ReadFile 方法匹配 ReadFileFunc 签名，用于模拟文件读取
func (f FakeReadFiler) ReadFile(filename string) ([]byte, error) {
    // 模拟读取指定内容，忽略 filename 参数
    // 实际应用中可以根据 filename 返回不同的内容或错误
    return []byte(f.Content), 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() {
    payload := "Mocked content via package variable!"
    path := "/path/to/mocked/file.txt"

    // 在测试前，将 myReadFile 替换为模拟实现
    // 注意：这种方式会影响整个包，因此在并行测试中需谨慎处理，
    // 并在测试结束后恢复原始值（通常在 TestXxx 函数的 defer 中完成）
    originalReadFile := myReadFile // 保存原始函数，以便测试结束后恢复
    defer func() {
        myReadFile = originalReadFile // 测试结束后恢复原始函数
    }()

    fakeReader := FakeReadFiler{Content: payload}
    myReadFile = fakeReader.ReadFile // 替换为模拟函数

    result2, err := ObtainTranslationStringsFileChoice2(path)
    fmt.Printf("方法二