testify

在團(tuán)隊(duì)里推行單元測(cè)試的時(shí)候，有一個(gè)反對(duì)的意見(jiàn)是：寫單元測(cè)試耗時(shí)太多。且不論這個(gè)意見(jiàn)對(duì)錯(cuò)，單元測(cè)試確實(shí)不應(yīng)該太費(fèi)時(shí)間。這時(shí)候，一個(gè)好的單測(cè)輔助工具，顯得格外重要。本文推薦的 testify(github.com/stretchr/te…) 包，具有斷言、mock 等功能，能配合標(biāo)準(zhǔn)庫(kù)，使你的單元測(cè)試更加簡(jiǎn)潔易讀。

testify 有三個(gè)主要功能：

• 斷言，在 assert 包和 require 包。
• Mocking，在 mock 包下。
• 測(cè)試組件，在 suite 包下。

assert 包

assert 包提供了一系列很方便的斷言方法，簡(jiǎn)化你的測(cè)試代碼。如

package yours
import (
  "testing"
  "github.com/stretchr/testify/assert"
)
func TestSomething(t *testing.T) {
  // 斷言相等
  assert.Equal(t, 123, 123, "they should be equal")
  // 斷言不等
  assert.NotEqual(t, 123, 456, "they should not be equal")
  // 斷言為 nil 
  assert.Nil(t, object)
}

assert 包的函數(shù)的第一個(gè)參數(shù)為 testing.T，用于執(zhí)行 go test 時(shí)輸出信息。

如果你有很多個(gè)斷言，可以調(diào)用New方法實(shí)例化 Assertions 結(jié)構(gòu)體，然后就可以省略testing.T參數(shù)了。上面的代碼，可以簡(jiǎn)化成

func TestSomething(t *testing.T) {
  // 實(shí)例化 assertion 結(jié)構(gòu)體，下面的斷言都不用傳入 t 作為第一個(gè)參數(shù)了。
  assert := assert.New(t)
  // 斷言相等
  assert.Equal(123, 123, "they should be equal")
  // 斷言不等
  assert.NotEqual(123, 456, "they should not be equal")
  // 斷言為 nil 
  assert.Nil(object)
}

assert 失敗的話，底層調(diào)用 t.Errorf 來(lái)輸出錯(cuò)誤信息。也就是說(shuō)，斷言失敗并不會(huì)中停止測(cè)試。

assert 包的斷言函數(shù)，返回值是 bool 類型，表示斷言的成功或失敗。 我們可以根據(jù)返回值，進(jìn)一步做斷言。如

// 當(dāng) object 不為 nil 的時(shí)候，進(jìn)一步斷言 object.Value 的值
if assert.NotNil(t, object) {
  assert.Equal(t, "Something", object.Value)
}

assert 包提供的斷言類型非常多，包括對(duì)比變量、json、目錄、Http 響應(yīng)等。完整列表見(jiàn)：pkg.go.dev/github.com/…

require 包

require 包提供的函數(shù)和 assert 包是一樣的，區(qū)別是：

• require 包如果斷言失敗，底層調(diào)用 t.FailNow， 會(huì)立刻中斷當(dāng)前的測(cè)試，所以也不會(huì)有返回值。
• assert 包如果斷言失敗，底層調(diào)用 t.Errorf，返回 false，不會(huì)中斷測(cè)試。

mock 包

單元測(cè)試一般僅限于測(cè)試本服務(wù)，對(duì)于別的服務(wù)的調(diào)用（比如數(shù)據(jù)庫(kù)），我們可以創(chuàng)建 Mock 對(duì)象來(lái)模擬對(duì)其他服務(wù)的調(diào)用。

和別的語(yǔ)言一樣，Mock 對(duì)象我們可以通過(guò)工具自動(dòng)生成。mockery(github.com/vektra/mock…) 工具可以根據(jù) golang 的 interface，生成類似下面的 Mock 對(duì)象。

package yours
import (
  "testing"
  "github.com/stretchr/testify/mock"
)
// Mock 對(duì)象，會(huì)依賴 mock.Mock
type MyMockedObject struct{
  mock.Mock
}
// 要 mock 的接口的方法
func (m *MyMockedObject) DoSomething(number int) (bool, error) {
  args := m.Called(number)
  return args.Bool(0), args.Error(1)
}

我們可以調(diào)用 Mock 對(duì)象的 On 方法，設(shè)置對(duì)應(yīng)方法的參數(shù)和返回值。如

// 測(cè)試
func TestSomething(t *testing.T) {
  // 實(shí)例化 Mock 對(duì)象
  testObj := new(MyMockedObject)
  // 設(shè)置預(yù)期，當(dāng)調(diào)用 testObj.DoSomething(123)時(shí)，返回 true, nil
  testObj.On("DoSomething", 123).Return(true, nil)
  // 我們要測(cè)試的函數(shù)
  targetFuncThatDoesSomethingWithObj(testObj)
  // 斷言符合預(yù)期，即 testObj.DoSomething(123) 會(huì)被調(diào)用
  testObj.AssertExpectations(t)
}

用On方法設(shè)置預(yù)期（expectations）時(shí)，傳入的參數(shù)可以使用 mock.Anything，表示任何值都行。如

  // 設(shè)置預(yù)期，當(dāng)調(diào)用 DoSomething 時(shí)，返回 ture,nil
  testObj.On("DoSomething", mock.Anything).Return(true, nil)

如果有多個(gè) mock 對(duì)象需要 AssertExpectations，如

testObj1.AssertExpectations(t)
testObj2.AssertExpectations(t)
testObj3.AssertExpectations(t)

可以使用 mock.AssertExpectations 優(yōu)化

mock.AssertExpectations(t, testObj1, testObj2, testObj3)

mock.Mock 還有一些很實(shí)用的常用斷言，如：

• AssertCalled 斷言被調(diào)用
• AssertNotCalled 斷言沒(méi)被調(diào)用
• AssertExpectations 斷言 On 和 Return 設(shè)置的參數(shù)和返回值的方法，有被調(diào)用
• AssertNumberOfCalls 斷言調(diào)用次數(shù)

完整的列表可以查看: pkg.go.dev/github.com/…

suite 包

如果你有別的面向?qū)ο笳Z(yǔ)言的經(jīng)驗(yàn)，用 suite 包寫單元測(cè)試可能更符合你的習(xí)慣。我們可以自定義一個(gè)結(jié)構(gòu)體，它依賴 suite.Suite，它所有的以 Test 開(kāi)頭的函數(shù)，都是一個(gè)測(cè)試。

import (
    "testing"
    "github.com/stretchr/testify/assert"
    "github.com/stretchr/testify/suite"
)
// 依賴 suite.Suite
type ExampleTestSuite struct {
    suite.Suite
    VariableThatShouldStartAtFive int
}
// 每個(gè)測(cè)試運(yùn)行前，會(huì)執(zhí)行
func (suite *ExampleTestSuite) SetupTest() {
    suite.VariableThatShouldStartAtFive = 5
}
// 所有以“Test”開(kāi)頭的方法，都是一個(gè)測(cè)試
func (suite *ExampleTestSuite) TestExample() {
    assert.Equal(suite.T(), 5, suite.VariableThatShouldStartAtFive)
}
// 用于 'go test' 的入口
func TestExampleTestSuite(t *testing.T) {
    suite.Run(t, new(ExampleTestSuite))
}

除了 SetupTest，suite.Suite 還有一些鉤子：

TearDownTest 每個(gè)測(cè)試之后執(zhí)行

SetupSuite Suite 開(kāi)始之前執(zhí)行一次，在所有測(cè)試之前執(zhí)行

TearDownSuite Suite 結(jié)束之后執(zhí)行一次，在所有測(cè)試之后執(zhí)行

更多的介紹可以查看官網(wǎng)：pkg.go.dev/github.com/…

引用

pkg.go.dev/github.com/…

以上就是Go單元測(cè)試?yán)鱰estify使用示例詳解的詳細(xì)內(nèi)容，更多關(guān)于Go 單元測(cè)試testify的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

最新評(píng)論