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