如何用Golang测试文件读写操作_Golang 文件读写测试实践

func TestWriteAndReadFile(t *testing.T) {
    content := "hello, golang"
    
    // 创建临时文件
    tmpfile, err := os.CreateTemp("", "testfile-*.txt")
    if err != nil {
        t.Fatal(err)
    }
    defer os.Remove(tmpfile.Name()) // 测试后删除
    defer tmpfile.Close()

    // 写入数据
    if _, err := tmpfile.Write([]byte(content)); err != nil {
        t.Error("写入失败:", err)
    }

    // 重置文件指针以便读取
    tmpfile.Seek(0, 0)
    data, err := io.ReadAll(tmpfile)
    if err != nil {
        t.Error("读取失败:", err)
    }

    if string(data) != content {
        t.Errorf("期望 %q，得到 %q", content, string(data))
    }
}

func TestFileServiceWithMock(t *testing.T) {
    var buf bytes.Buffer

    // 模拟写入
    writer := func(data []byte) error {
        _, err := buf.Write(data)
        return err
    }

    reader := func() ([]byte, error) {
        return buf.Bytes(), nil
    }

    // 调用你的服务逻辑
    err := writer([]byte("test"))
    if err != nil {
        t.Fatal(err)
    }

    data, _ := reader()
    if string(data) != "test" {
        t.Errorf("读取内容不符: %s", string(data))
    }
}