單元測試 go 函數時需注意以下陷阱：避免依賴外部資源，使用樁和模擬來隔離依賴項。檢查錯誤，不要忽略它們。使用反射或重命名來測試私有方法。使用同步原語避免并發下的競態條件。
Go 函數單元測試的陷阱和注意事項
單元測試是保證代碼質量的關鍵實踐。在 Go 中，測試使用 testing 包。雖然單元測試相對簡單，但有一些陷阱和注意事項需要注意。
1. 依賴外部資源
單元測試應該隔離待測代碼，不依賴外部資源（例如數據庫或網絡調用）。為此，可以使用樁（stub）、模擬（mock）或測試雙（test double）來隔離外部依賴項。
示例（樁）：
type DatabaseClient interface {
GetUser(id int) (*User, error)
}
// DbClientStub 是 DatabaseClient 的樁
type DbClientStub struct {
GetResult *User
GetError error
}
func (s *DbClientStub) GetUser(id int) (*User, error) {
return s.GetResult, s.GetError
}
登錄后復制
2. 忽略錯誤
在測試中忽略錯誤很誘人，尤其是在測試正常代碼路徑時。然而，這會導致難以調試問題，并且可能導致代碼因未處理的錯誤而失敗。在可能的情況下，應始終檢查錯誤并相應地處理它們。
示例：
func GetUser(id int) (*User, error) {
// ... 從數據庫中獲取用戶
// **不要忽略錯誤！**
if err != nil {
return nil, err
}
return user, nil
}
登錄后復制
3. 測試私有方法
Go 語言的私有方法（小寫名稱）通常用于實現接口方法或隱藏實現細節。然而，它們不能直接從外部測試。有幾種方法可以測試私有方法：
使用反射： 從測試包中使用 reflect 包來訪問私有方法。重命名私有方法： 將私有方法重命名為首字母大寫的包級別方法。
示例（反射）：
func TestPrivateMethod(t *testing.T) {
// 使用反射訪問私有方法
value := reflect.ValueOf(myPackage.myPrivateMethod)
result := value.Call([]reflect.Value{reflect.ValueOf(123)})
// 檢查結果
if result[0].Int() != 456 {
t.Errorf("Expected 456, got %d", result[0].Int())
}
}
登錄后復制
4. 競態條件
Go 的并發性使得競態條件成為可能。單元測試應注意避免競態條件，例如通過在并發Goroutine上使用同步原語（例如sync.Mutex）。
示例（使用 sync.Mutex）：
var userMap sync.Map
func TestConcurrentUserMap(t *testing.T) {
// 創建 goroutine <a style='color:#f60; text-decoration:underline;' href="https://www.php.cn/zt/35877.html" target="_blank">并發訪問</a> userMap
for i := 0; i < 1000; i++ {
go func(i int) {
userMap.LoadOrStore(i, "User"+strconv.Itoa(i))
}(i)
}
// 等待所有 goroutine 完成
time.Sleep(time.Millisecond)
// 驗證 userMap 是否包含所有預期的鍵
for i := 0; i < 1000; i++ {
if _, ok := userMap.Load(i); !ok {
t.Errorf("userMap doesn't contain key %d", i)
}
}
}
登錄后復制
