Go语言文件读写测试方法详解

func TestWriteAndReadFile(t *testing.T) {
    tmpDir := t.TempDir()
    filePath := filepath.Join(tmpDir, "test.txt")

    // 写入文件
    dataToWrite := "hello, world"
    err := os.WriteFile(filePath, []byte(dataToWrite), 0644)
    if err != nil {
        t.Fatalf("写入文件失败: %v", err)
    }

    // 读取文件
    data, err := os.ReadFile(filePath)
    if err != nil {
        t.Fatalf("读取文件失败: %v", err)
    }

    if string(data) != dataToWrite {
        t.Errorf("期望 %q，实际 %q", dataToWrite, string(data))
    }
}

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

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

type DiskIO struct{}

func (d DiskIO) Read(file string) ([]byte, error) {
    return os.ReadFile(file)
}

func (d DiskIO) Write(file string, data []byte) error {
    return os.WriteFile(file, data, 0644)
}

type MockIO struct {
    data map[string][]byte
}

func (m MockIO) Read(file string) ([]byte, error) {
    d, ok := m.data[file]
    if !ok {
        return nil, os.ErrNotExist
    }
    return d, nil
}

func (m MockIO) Write(file string, data []byte) error {
    if m.data == nil {
        m.data = make(map[string][]byte)
    }
    m.data[file] = append([]byte{}, data...)
    return nil
}

func TestFileProcessor(t *testing.T) {
    mock := MockIO{data: make(map[string][]byte)}
    mock.Write("config.txt", []byte("timeout=30"))

    data, _ := mock.Read("config.txt")
    if string(data) != "timeout=30" {
        t.Error("读取数据不符")
    }
}

func TestLoadConfig(t *testing.T) {
    f, err := ioutil.TempFile("", "config-*.json")
    if err != nil {
        t.Fatal(err)
    }
    defer os.Remove(f.Name()) // 测试完清理
    defer f.Close()

    _, err = f.WriteString(`{"port": 8080}`)
    if err != nil {
        t.Fatal(err)
    }

    config, err := LoadConfig(f.Name())
    if err != nil {
        t.Fatal(err)
    }
    if config.Port != 8080 {
        t.Errorf("期望端口 8080，实际 %d", config.Port)
    }
}