go單元測試的方法有哪些

本篇內(nèi)容介紹了“go單元測試的方法有哪些”的有關(guān)知識，在實(shí)際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！

1.gomock

gomock模擬對象的方式是讓用戶聲明一個(gè)接口，然后使用gomock提供的mockgen工具生成mock對象代碼。要模擬(mock)被測試代碼的依賴對象時(shí)候，即可使用mock出來的對象來模擬和記錄依賴對象的各種行為：比如最常用的返回值，調(diào)用次數(shù)等等。文字?jǐn)⑹鲇悬c(diǎn)抽象，直接上代碼：

dick.go中DickFunc依賴外部對象OutterObj，本示例就是說明如何使用gomock框架控制所依賴的對象。

func DickFunc( outterObj MockInterface,para int)(result int){
    fmt.Println("This init DickFunc")
    fmt.Println("call outter.func:")

    return outterObj.OutterFunc(para)
}

mockgen工具命令是：

mockgen -source {source_file}.go -destination {dest_file}.go

比如，本示例即是：

mockgen -source src_mock.go -destination dst_mock.go

執(zhí)行完后，可在同目錄下找到生成的dst_mock.go文件，可以看到mockgen工具也實(shí)現(xiàn)了接口：

在本示例中只簡單的更改了返回值，拋磚引玉：

func TestDickFunc(t *testing.T ){
   mockCtrl := gomock.NewController(t)
//defer mockCtrl.Finish()

   mockObj := dick.NewMockMockInterface(mockCtrl)
   mockObj.EXPECT().OutterFunc(3).Return(10)

   result :=dick.DickFunc(mockObj,3)
   t.Log("resutl:",result)

}

使用go test命令執(zhí)行這個(gè)單測 

從結(jié)果看：本來應(yīng)該輸出3，最后輸出就是10，和其他語言mock框架相似，生產(chǎn)出來的Mock對象不用自己去重定義這么麻煩。

2.httpexcept

由于go在網(wǎng)絡(luò)架構(gòu)上的優(yōu)秀封裝，使得go在很多網(wǎng)絡(luò)場景被廣泛使用，而http協(xié)議是其中重要部分，在面對http請求的時(shí)候，可以對http的client進(jìn)行測試，算是mock的特殊應(yīng)用場景。

看一個(gè)簡單的示例就輕松的看懂了：

func TestHttp(t *testing.T) {

    handler := FruitServer()

    server := httptest.NewServer(handler)
    defer server.Close()

    e := httpexpect.New(t, server.URL)

    e.GET("/fruits").
        Expect().
        Status(http.StatusOK).JSON().Array().Empty()
}

其中還支持對不同方法(包括Header,Post等)的構(gòu)造以及返回值Json的自定義，更多細(xì)節(jié)查看其官網(wǎng)

3.testify

還有一個(gè)testify使用起來可以說兼容了《一》中的gocheck和gomock，但是其mock使用稍微有點(diǎn)煩雜，使用繼承tetify.Mock(匿名組合)重新實(shí)現(xiàn)需要Mock的接口，在這個(gè)接口里使用者自己使用Called(反射實(shí)現(xiàn))被Mock的接口。

《單元測試的藝術(shù)》中認(rèn)為stub和mock最大的區(qū)別就依賴對象是否和被測對象有交互，而從結(jié)果看就是樁對象不會使測試失敗，它只是為被測對象提供依賴的對象，并不改變測試結(jié)果，而mock則會根據(jù)不同的交互測試要求，很可能會更改測試的結(jié)果。說了這么多理論，但其實(shí)這兩種方法都不是割裂的，所以gomock框架除了像其名字一樣可以模擬對象以外，還提供了樁對象的功能(stub)。以其實(shí)現(xiàn)來說，更像是一個(gè)樁對象的注入。但是因?yàn)榧嫒萘硕鄠€(gè)有用的功能，所以其在社區(qū)最為火爆。

具體用法可參考其github主頁

4.go-sqlmock

還有一種比較常見的場景就是和數(shù)據(jù)庫的交互場景，go-sqlmock是sql模擬（Mock）驅(qū)動器，主要用于測試數(shù)據(jù)庫的交互，go-sqlmock提供了完整的事務(wù)的執(zhí)行測試框架，最新的版本(16.11.02)還支持prepare參數(shù)化提交和執(zhí)行的Mock方案。

比如有這樣的被測函數(shù)：

func recordStats(db *sql.DB, userID, productID int64) (err error) {
    tx, err := db.Begin()
    if err != nil {
        return
    }

    defer func() {
        switch err {
        case nil:
            err = tx.Commit()
        default:
            tx.Rollback()
        }
    }()

    if _, err = tx.Exec("UPDATE products SET views = views + 1"); err != nil {
        return
    }
    if _, err = tx.Exec("INSERT INTO product_viewers (user_id, product_id) VALUES (?, ?)", userID, productID); err != nil {
        return
    }
    return
}

func main() {

    db, err := sql.Open("mysql", "root@/root")
    if err != nil {
        panic(err)
    }
    defer db.Close()

    if err = recordStats(db, 1 , 5 ); err != nil {
        panic(err)
    }
}

單測時(shí)：

func TestShouldUpdateStats(t *testing.T) {
    db, mock, err := sqlmock.New()
    if err != nil {
        t.Fatalf("mock error: '%s' ", err)
    }
    defer db.Close()

    mock.ExpectBegin()
    mock.ExpectExec("UPDATE products").WillReturnResult(sqlmock.NewResult(1, 1))
    mock.ExpectExec("INSERT INTO product_viewers")
          .WithArgs(2, 3)
          .WillReturnResult(sqlmock.NewResult(1, 1))
    mock.ExpectCommit()

    if err = recordStats(db, 2, 3); err != nil {
        t.Errorf("exe error: %s", err)
    }

    if err := mock.ExpectationsWereMet(); err != nil {
        t.Errorf("not implements: %s", err)
    }
}

//測試回滾
func TestShouldRollbackStatUpdatesOnFailure(t *testing.T) {
    db, mock, err := sqlmock.New()
    if err != nil {
        t.Fatalf("mock error: '%s'", err)
    }
    defer db.Close()

    mock.ExpectBegin()
    mock.ExpectExec("UPDATE products").WillReturnResult(sqlmock.NewResult(1, 1))
    mock.ExpectExec("INSERT INTO product_viewers")
           .WithArgs(2, 3)
           .WillReturnError(fmt.Errorf("some error"))
    mock.ExpectRollback()

    // 執(zhí)行被測方法,有錯
    if err = recordStats(db, 2, 3); err == nil {
        t.Errorf("not error")
    }

    // 執(zhí)行被測方法，mock對象
    if err := mock.ExpectationsWereMet(); err != nil {
        t.Errorf("not implements: %s", err)
    }
}

“go單元測試的方法有哪些”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識可以關(guān)注億速云網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實(shí)用文章！