Golang临时文件测试技巧与使用方法

func TestWithTempFile(t *testing.T) {
  tmpfile, err := os.CreateTemp("", "testfile-*.txt")
  if err != nil {
    t.Fatal(err)
  }

  // 确保测试结束后删除文件
  t.Cleanup(func() {
    os.Remove(tmpfile.Name())
  })

  // 写入测试数据
  _, err = tmpfile.Write([]byte("hello test"))
  if err != nil {
    t.Fatal(err)
  }
  tmpfile.Close()

  // 读取验证
  data, err := os.ReadFile(tmpfile.Name())
  if err != nil {
    t.Fatal(err)
  }
  if string(data) != "hello test" {
    t.Errorf("期望: hello test, 实际: %s", data)
  }
}

func TestMultipleTempFiles(t *testing.T) {
  dir, err := os.MkdirTemp("", "testdir-*")
  if err != nil {
    t.Fatal(err)
  }

  t.Cleanup(func() {
    os.RemoveAll(dir) // 删除整个目录
  })

  file1 := filepath.Join(dir, "file1.txt")
  file2 := filepath.Join(dir, "file2.txt")

  os.WriteFile(file1, []byte("data1"), 0644)
  os.WriteFile(file2, []byte("data2"), 0644)

  // 测试逻辑...
}