清理資源是go測試的關(guān)鍵環(huán)節(jié)，忽視可能導致測試失敗或影響生產(chǎn)環(huán)境。針對文件操作，應(yīng)使用ioutil.tempdir或os.createtemp創(chuàng)建臨時目錄并測試后用defer os.removeall(tempdir)刪除；對于數(shù)據(jù)庫連接，需使用專用測試庫并在測試前后清空數(shù)據(jù)表及關(guān)閉連接；測試中啟動的http服務(wù)或tcp監(jiān)聽器必須在測試結(jié)束時通過srv.shutdown(ctx)優(yōu)雅關(guān)閉；此外，異步任務(wù)如goroutine和定時器應(yīng)帶上context并調(diào)用timer.stop()確保釋放。以上步驟依次保障了文件、數(shù)據(jù)庫、網(wǎng)絡(luò)及其他資源的正確回收。
寫Go測試的時候，資源清理往往容易被忽略。尤其是涉及文件、數(shù)據(jù)庫連接、網(wǎng)絡(luò)服務(wù)時，不及時釋放可能導致后續(xù)測試失敗甚至影響生產(chǎn)環(huán)境。下面幾個策略能幫你有效管理測試后的資源回收。
臨時目錄和文件要及時刪除
在測試中創(chuàng)建的臨時文件或目錄，最好用ioutil.TempDir或者os.CreateTemp來生成，這樣能保證系統(tǒng)自動分配路徑，避免沖突。
但關(guān)鍵點是：測試完成后必須手動刪除。可以配合defer os.RemoveAll(tempDir)使用，確保即使測試出錯也能清理掉。
舉個例子：
func TestSomething(t *testing.T) { tempDir, err := os.MkdirTemp("", "test-*") if err != nil { t.Fatal(err) } defer os.RemoveAll(tempDir) // 測試邏輯使用tempDir }
這樣做的好處是：
立即學習“go語言免費學習筆記（深入）”；
- 不會留下垃圾文件
- 每次測試都是干凈的環(huán)境
- 避免權(quán)限問題導致下次測試失敗
數(shù)據(jù)庫連接與數(shù)據(jù)要斷開并清空
如果測試過程中連接了真實數(shù)據(jù)庫（比如mysql、postgresql），除了關(guān)閉連接外，還建議清空測試用的數(shù)據(jù)表。
否則多次運行測試會導致數(shù)據(jù)堆積，進而引發(fā)唯一索引沖突、查詢結(jié)果異常等問題。
常見做法有：
- 使用專用測試數(shù)據(jù)庫或schema
- 測試開始前清空相關(guān)表（如delete FROM users;）
- 測試結(jié)束后關(guān)閉連接，并確認連接池已釋放
例如：
func setupDB() (*sql.DB, func()) { db, err := sql.Open("mysql", "user:pass@/testdb") if err != nil { panic(err) } _, _ = db.Exec("DELETE FROM users") return db, func() { db.Close() } }
這樣做之后，每次測試都能在一個“干凈”的數(shù)據(jù)庫狀態(tài)下運行。
網(wǎng)絡(luò)監(jiān)聽和服務(wù)要關(guān)閉
如果你在測試里啟動了一個HTTP服務(wù)或TCP監(jiān)聽器，記得在測試結(jié)束時關(guān)閉它。否則端口一直被占用，會影響其他測試甚至本地開發(fā)。
標準做法是在啟動服務(wù)器后立即用defer關(guān)閉：
func TestHTTPServer(t *testing.T) { srv := &http.Server{Addr: ":8080"} go func() { if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed { t.Log(err) } }() defer func() { ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() srv.Shutdown(ctx) }() // 發(fā)起請求測試 }
注意幾點：
- 使用Shutdown()而不是Close()，更優(yōu)雅地關(guān)閉連接
- 加上超時機制，防止卡住
- 盡量使用隨機端口，減少沖突可能
其他資源別忘了：鎖、goroutine、定時器
有時候測試會啟動goroutine或設(shè)置定時任務(wù)，這些很容易被遺漏。一個沒處理好的timer或者阻塞的goroutine，會讓整個測試掛住。
建議：
比如定時器的例子：
timer := time.NewTimer(time.Second) defer timer.Stop()
還有像文件鎖、系統(tǒng)信號監(jiān)聽等資源，也要在測試完成后恢復原始狀態(tài)。
基本上就這些。資源清理看起來不復雜，但在測試中很容易被忽略。特別是多人協(xié)作項目，如果不統(tǒng)一規(guī)范，很容易埋下隱患。所以寫測試的時候多花幾行代碼做清理，后期省很多排查時間。
