Go语言测试：如何优雅地模拟文件读取操作

package main

import (
    "bytes"
    "fmt"
    "io"
    "io/ioutil" // 示例沿用ioutil，生产环境建议使用os.ReadFile或io.ReadAll
)

// ProcessFileContentsOption1 接受一个io.Reader接口，用于处理文件内容
func ProcessFileContentsOption1(rdr io.Reader) ([]string, error) {
    // 使用ioutil.ReadAll从io.Reader中读取所有内容
    if contents, err := ioutil.ReadAll(rdr); err == nil {
        // 模拟对内容的进一步处理，这里简单地将其包装成一个字符串切片
        return []string{string(contents)}, nil
    } else {
        return nil, err
    }
}

func main() {
    payload := "这是一个模拟的文件内容"

    // 模拟测试场景：使用bytes.Buffer作为io.Reader
    buf := bytes.NewBufferString(payload)
    result1, err := ProcessFileContentsOption1(buf)
    fmt.Printf("ProcessFileContentsOption1 结果: %#v, 错误: %#v\n", result1, err)

    // 实际场景：如果需要从真实文件读取，可以这样传入*os.File
    // file, err := os.Open("your_file.txt")
    // if err != nil {
    //     log.Fatal(err)
    // }
    // defer file.Close()
    // resultReal, err := ProcessFileContentsOption1(file)
    // fmt.Printf("ProcessFileContentsOption1 真实文件结果: %#v, 错误: %#v\n", resultReal, err)
}

package main

import (
    "bytes"
    "fmt"
    "io/ioutil" // 示例沿用ioutil
)

// ReadFileFunc 是一个函数类型，与ioutil.ReadFile的签名一致
type ReadFileFunc func(filename string) ([]byte, error)

// myReadFile 是一个包级变量，默认指向ioutil.ReadFile
var myReadFile ReadFileFunc = ioutil.ReadFile

// FakeReadFiler 结构体，包含模拟的文件内容
type FakeReadFiler struct {
    Str string
}

// ReadFile 方法实现了ReadFileFunc的签名，用于模拟文件读取
func (f FakeReadFiler) ReadFile(filename string) ([]byte, error) {
    // 这里可以根据filename返回不同的内容或错误
    buf := bytes.NewBufferString(f.Str)
    return ioutil.ReadAll(buf) // 模拟从缓冲区读取
}

// ProcessFileContentsOption2 使用包级变量myReadFile来读取文件
func ProcessFileContentsOption2(path string) ([]string, error) {
    if contents, err := myReadFile(path); err == nil {
        return []string{string(contents)}, nil
    } else {
        return nil, err
    }
}

func main() {
    payload := "这是通过包级变量模拟的文件内容"
    path := "/dev/null" // 路径在这里不重要，因为我们模拟了读取行为

    // 默认情况下，myReadFile会调用ioutil.ReadFile
    // resultDefault, err := ProcessFileContentsOption2(path)
    // fmt.Printf("默认结果: %#v, 错误: %#v\n", resultDefault, err)

    // 模拟测试场景：将myReadFile替换为FakeReadFiler的ReadFile方法
    fake := FakeReadFiler{Str: payload}
    myReadFile = fake.ReadFile // 替换包级变量
    result2, err := ProcessFileContentsOption2(path)
    fmt.Printf("ProcessFileContentsOption2 结果: %#v, 错误: %#v\n", result2, err)

    // 重要：测试结束后通常需要恢复myReadFile，以避免影响其他测试
    // myReadFile = ioutil.ReadFile
}