test: fix flaky test

Signed-off-by: Jian Zeng <[email protected]>
ustclug · Jan 1, 2024 · 7937caa · 7937caa
1 parent f84d6f5
commit 7937caa
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkg/server/main_test.go b/pkg/server/main_test.go
@@ -49,8 +49,12 @@ func NewTestEnv(t *testing.T) *TestEnv {
 	v := validator.New()
 	e.Validator = echoValidator(v.Struct)
 
-	// Connect to a temporary sqlite database.
-	db, err := gorm.Open(sqlite.Open(""), &gorm.Config{
+	dbFile, err := os.CreateTemp("", "yukid*.db")
+	require.NoError(t, err)
+	t.Cleanup(func() {
+		_ = os.Remove(dbFile.Name())
+	})
+	db, err := gorm.Open(sqlite.Open(dbFile.Name()), &gorm.Config{
 		QueryFields: true,
 	})
 	require.NoError(t, err)