本文介绍了在 Go 单元测试中如何安全、有效地使用临时目录来存放测试所需的临时文件。通过使用 testing.T.TempDir() 方法，可以创建在测试结束后自动清理的临时目录，避免手动管理临时文件带来的潜在问题。
在编写 Go 单元测试时，经常需要创建一些临时文件，例如配置文件、测试数据等。这些文件仅在测试期间使用，测试结束后应该被清理掉，以避免占用磁盘空间或干扰其他测试。手动创建和删除临时文件容易出错，而且在不同平台上可能存在兼容性问题。Go 1.15 之后，testing 包提供了一个更方便、更可靠的方法：T.TempDir()。
使用 testing.T.TempDir() 创建临时目录
T.TempDir() 方法是 testing.T 类型的一个方法，它会在测试执行期间创建一个唯一的临时目录，并在测试结束后自动清理该目录。这意味着你无需手动删除临时文件，可以专注于测试逻辑的编写。
package mypackage
import (
"io/ioutil"
"os"
"path/filepath"
"testing"
)
func LoadData(path string) (string, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return "", err
}
return string(b), nil
}
func TestLoadData(t *testing.T) {
// 创建临时目录
tempDir := t.TempDir()
// 创建临时文件
tempFile := filepath.Join(tempDir, "testdata.txt")
err := ioutil.WriteFile(tempFile, []byte("test data"), 0644)
if err != nil {
t.Fatalf("Failed to create temp file: %v", err)
}
// 调用被测函数
data, err := LoadData(tempFile)
if err != nil {
t.Fatalf("Failed to load data: %v", err)
}
// 断言结果
if data != "test data" {
t.Errorf("Expected 'test data', got '%s'", data)
}
// 注意：不需要手动删除临时文件，T.TempDir() 会自动清理
}示例说明：
- t.TempDir() 创建一个临时目录，并将目录路径赋值给 tempDir 变量。
- filepath.Join(tempDir, "testdata.txt") 构建临时文件的完整路径。
- ioutil.WriteFile() 将测试数据写入临时文件。
- LoadData() 函数读取临时文件中的数据。
- 最后，使用断言验证读取的数据是否正确。
优点：
- 自动清理： 无需手动删除临时文件，避免资源泄露。
- 跨平台： 适用于各种操作系统，无需考虑平台差异。
- 线程安全： 在并行测试中，每个测试都会获得独立的临时目录，避免并发冲突。
- 唯一性： 每次调用 t.TempDir() 都会创建一个新的临时目录，保证测试的隔离性。
替代方案：
在 Go 1.15 之前，可以使用 ioutil.TempFile 或 os.TempDir 手动创建临时文件和目录。但是，这些方法需要手动管理临时文件的生命周期，容易出错。
注意事项：
- t.TempDir() 只能在测试函数内部调用。
- 临时目录的生命周期与测试函数的生命周期相同。当测试函数执行完毕后，临时目录及其内容将被自动删除。
- 如果在测试函数中使用了 t.Fatal() 或 t.FailNow() 等方法提前终止测试，临时目录仍然会被自动清理。
总结：
testing.T.TempDir() 是在 Go 单元测试中创建临时目录的最佳实践。它简化了临时文件的管理，提高了测试代码的可靠性和可维护性。建议在编写 Go 单元测试时，优先使用 t.TempDir() 来创建临时目录。
