單元測(cè)試 go 函數(shù)時(shí)需注意以下陷阱：避免依賴外部資源，使用樁和模擬來隔離依賴項(xiàng)。檢查錯(cuò)誤，不要忽略它們。使用反射或重命名來測(cè)試私有方法。使用同步原語避免并發(fā)下的競態(tài)條件。

Go 函數(shù)單元測(cè)試的陷阱和注意事項(xiàng)

單元測(cè)試是保證代碼質(zhì)量的關(guān)鍵實(shí)踐。在 Go 中，測(cè)試使用 包。雖然單元測(cè)試相對(duì)簡單，但有一些陷阱和注意事項(xiàng)需要注意。

1. 依賴外部資源

單元測(cè)試應(yīng)該隔離待測(cè)代碼，不依賴外部資源（例如數(shù)據(jù)庫或網(wǎng)絡(luò)調(diào)用）。為此，可以使用樁（stub）、模擬（mock）或測(cè)試雙（test double）來隔離外部依賴項(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
}

2. 忽略錯(cuò)誤

在測(cè)試中忽略錯(cuò)誤很誘人，尤其是在測(cè)試正常代碼路徑時(shí)。然而，這會(huì)導(dǎo)致難以調(diào)試問題，并且可能導(dǎo)致代碼因未處理的錯(cuò)誤而失敗。在可能的情況下，應(yīng)始終檢查錯(cuò)誤并相應(yīng)地處理它們。

示例：

func GetUser(id int) (*User, error) {
    // ... 從數(shù)據(jù)庫中獲取用戶

    // **不要忽略錯(cuò)誤！**
    if err != nil {
        return nil, err
    }

    return user, nil
}

3. 測(cè)試私有方法

Go 語言的私有方法（小寫名稱）通常用于實(shí)現(xiàn)接口方法或隱藏實(shí)現(xiàn)細(xì)節(jié)。然而，它們不能直接從外部測(cè)試。有幾種方法可以測(cè)試私有方法：

使用反射： 從測(cè)試包中使用 包來訪問私有方法。重命名私有方法： 將私有方法重命名為首字母大寫的包級(jí)別方法。

示例（反射）：

func TestPrivateMethod(t *testing.T) {
    // 使用反射訪問私有方法
    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())
    }
}

4. 競態(tài)條件

Go 的并發(fā)性使得競態(tài)條件成為可能。單元測(cè)試應(yīng)注意避免競態(tài)條件，例如通過在并發(fā)Goroutine上使用同步原語（例如sync.Mutex）。

示例（使用 ）：

var userMap sync.Map

func TestConcurrentUserMap(t *testing.T) {
    // 創(chuàng)建 goroutine <a style='color:#f60; text-decoration:underline;' href="/zt/35877.html" target="_blank">并發(fā)訪問</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)
        }
    }
}