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))
    }
}

type FileWriter interface {
    Write(data []byte) error
}

type FileReader interface {
    Read() ([]byte, error)
}

// 实现
type DiskFile struct {
    path string
}

func (df DiskFile) Write(data []byte) error {
    return os.WriteFile(df.path, data, 0644)
}

func (df DiskFile) Read() ([]byte, error) {
    return os.ReadFile(df.path)
}

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))
    }
}

func TestWriteMultipleFiles(t *testing.T) {
    dir, err := os.MkdirTemp("", "testdir-*")
    if err != nil {
        t.Fatal(err)
    }
    defer os.RemoveAll(dir) // 清理整个目录

    files := map[string]string{
        "a.txt": "data A",
        "b.txt": "data B",
    }

    for name, content := range files {
        path := filepath.Join(dir, name)
        if err := os.WriteFile(path, []byte(content), 0644); err != nil {
            t.Errorf("写入 %s 失败: %v", name, err)
        }
    }

    // 验证是否都写入成功
    for name, want := range files {
        path := filepath.Join(dir, name)
        data, err := os.ReadFile(path)
        if err != nil {
            t.Errorf("读取 %s 失败: %v", name, err)
            continue
        }
        if string(data) != want {
            t.Errorf("文件 %s: 期望 %q, 得到 %q", name, want, string(data))
        }
    }
}