fix: fix creating tmp dir in unit test

pkg/utils/yaml_test.go

@@ -18,16 +18,19 @@ package utils
 
 import (
 	"os"
+	"path/filepath"
 	"runtime"
 	"testing"
 	"time"
 
 	"github.com/cloudwego/kitex/internal/test"
 )
 
-const (
-	TestYamlFile = "/tmp/test.yaml"
-)
+func getTestYamlFile(t *testing.T) string {
+	tempDir := t.TempDir()
+
+	return filepath.Join(tempDir, "test.yaml")
+}
 
 var cfg *YamlConfig
 
@@ -89,22 +92,26 @@ func TestMain(m *testing.M) {
 		return
 	}
 	t := &testing.T{}
-	createTestYamlFile(t, TestYamlFile)
 
-	cfg, _ = ReadYamlConfigFile(TestYamlFile)
+	testYamlFile := getTestYamlFile(t)
+
+	createTestYamlFile(t, testYamlFile)
+
+	cfg, _ = ReadYamlConfigFile(testYamlFile)
 
 	exit := m.Run()
-	deleteTestYamlFile(t, TestYamlFile)
+	deleteTestYamlFile(t, testYamlFile)
 	os.Exit(exit)
 }
 
 func Test_ReadYamlConfigFile(t *testing.T) {
-	createTestYamlFile(t, TestYamlFile)
+	testYamlFile := getTestYamlFile(t)
+	createTestYamlFile(t, testYamlFile)
 	defer func() {
-		deleteTestYamlFile(t, TestYamlFile)
+		deleteTestYamlFile(t, testYamlFile)
 	}()
 
-	cfg, err := ReadYamlConfigFile(TestYamlFile)
+	cfg, err := ReadYamlConfigFile(testYamlFile)
 	test.Assert(t, err == nil)
 	addr, ok := cfg.GetString("Address")
 	test.Assert(t, ok && addr == ":8888")

tool/internal_pkg/generator/template_test.go

@@ -15,14 +15,21 @@
 package generator
 
 import (
+	"os"
+	"path/filepath"
 	"testing"
 
 	"github.com/cloudwego/kitex/internal/test"
 )
 
 func TestNilSafe(t *testing.T) {
 	var q *TemplateExtension
-	fn := "/tmp/kitex-template-nil.json"
+
+	tempDir := t.TempDir()
+
+	defer os.RemoveAll(tempDir)
+
+	fn := filepath.Join(tempDir, "kitex-template-nil.json")
 
 	err := q.ToJSONFile(fn)
 	test.Assert(t, err == nil, err)
@@ -56,7 +63,12 @@ func TestMarshal(t *testing.T) {
 		},
 	}
 
-	fn := "/tmp/kitex-template.json"
+	tempDir := t.TempDir()
+
+	defer os.RemoveAll(tempDir)
+
+	fn := filepath.Join(tempDir, "kitex-template.json")
+
 	err := p.ToJSONFile(fn)
 	test.Assert(t, err == nil, err)