Go單元測試實踐

　　單元測試通常用來在日常開發中檢查代碼中存在的問題，是提升代碼質量一種有效手段。在保證代碼功能沒有問題的同時，可以得到預期結果。Golang有許多優秀的框架支持UT，下面列舉日常開發中不同框架對應的UT情況，以便后來人實踐UT。

　　1、Goland提供的簡單UT模板

　　用途：對其中一個函數、方法生成UT

　　介紹：在新手不知道如何寫UT，按照什么規范去編寫UT的時候，不妨采用Goland自帶的模板。

　　tips：在Goland選中要測試的方法、函數--“generate”--"Test for selection"，然后在// TODO: Add test cases中加入自己的測試用例數據即可。

　　缺點：通用性一般，適合邏輯簡單的函數。

      2、Convey

　　用途：較為好用的UT框架，斷言多樣，提供UT的web界面，較為常用

　　官方文檔：link

　　tips：Testify: 斷言庫，斷言功能豐富，簡單易用，通常和其他框架搭配使用

　　code example:

　　

// TestFunc
func CheckVal(val string) bool {
    valList := []string{"AAA", "BBB"}
    for index := range valList {
        if valList[index] == val {
            return true
        }
    }
    return false
}


// Convey特點：斷言函數豐富，提供UT的web界面，較為常用
// Convey website: https://github.com/smartystreets/goconvey/wiki/Documentation

// Testify: 斷言庫，多用它提供的豐富的斷言功能
func TestCheckVal(t *testing.T) {
    convey.Convey("TestCheckVal happy path", t, func(){
        res := CheckVal("AAA")
        convey.So(res, convey.ShouldBeTrue)
        assert.True(t, res)
    })

    convey.Convey("TestCheckVal unhappy path", t, func(){
        res := CheckVal("CCC")
        convey.So(res, convey.ShouldBeFalse)
        assert.False(t, res)
    })
}

 

 

　　3、Gomonkey

　　tips：

　　　　mac電腦在用Gomonkey打樁的時候時常會遇到syscall的問題，是該框架的一個bug，需要配置插件來解決這個問題，具體可以參照國外作者的配置 link

　　code example:

　　

// Gomonkey: 十分常用的打樁工具,是在monkey框架基礎上個人改進的一款UT框架，比較好奇公司喜歡用這類個人庫而不是monkey這種偏官方的庫寫UT
// Gomonkey支持多種打樁特性，但在平常開發中多用來對全局變量打樁；對函數打樁；對方法打樁；對函數變量進行打樁

// 1、為函數打樁 ApplyFunc(origin func, stub func)
func CalVal(a, b int) int {
    return a + b
}

func TestCalVal(t *testing.T) {
    convey.Convey("happy path", t, func() {
        patch := gomonkey.ApplyFunc(CalVal, func(a, b int) int {
            return 3
        })
        defer patch.Reset()

        res := CalVal(1, 2)
        convey.So(res, convey.ShouldEqual, 3) // or use assert
    })
}

// 2、為全局變量打樁 ApplyGlobalVar（&var, stubvalue)

var temp = 10
func TestGlobal(t *testing.T) {
    convey.Convey("happy path", t, func() {
        patch := gomonkey.ApplyGlobalVar(&temp, 15)
        defer patch.Reset()
        convey.So(temp, convey.ShouldEqual, 15)
    })
}

// 3、對函數變量打樁 ApplyFuncVar (&func, stubfunc)   equal to ApplyFunc, but it is the address of func var

// 4、對方法打樁 ApplyMethod（reflect.TypeOf, methodName, stubfunc)

type User struct {
    money int
}

func(u *User) AddMoney(temp int) int{
    u.money += temp
    return u.money
}

func TestAddMoney(t *testing.T) {
    temp := User{}
    patch := gomonkey.ApplyMethod(reflect.TypeOf(&temp), "AddMoney", func(_ *User, temp int) int{
        return 100
    })
    defer patch.Reset()

    convey.Convey("happy path", t, func() {
        Bob := &User{
            money: 20,
        }
        convey.So(Bob.AddMoney(80),convey.ShouldEqual, 100)
    })

}

 

 

　　4、Gomock

　　用途：對接口進行mock，提前定義好返回內容

　　介紹：go官方提供的框架，同時還可以使用mockGen工具來生成接口的mock代碼

　　tips：

　　　　使用mockgen生成接口的mock代碼

　　　　mockgen -source= A.go -destination= A_mock.go -package=main     

　　code example:    

　　

// gomock 多用於接口打樁，結合mockgen工具自動生成打樁代碼，方便調用

type Repo interface {
    GetName(id int) string
}

func TestGetName(t *testing.T) {
    // new mockController
    ctrl := gomock.NewController(t)
    defer ctrl.Finish()

    // mock Repo interface
    mock := mock_go_UT_learn.NewMockRepo(ctrl)
    mock.EXPECT().GetName(gomock.Eq(1)).Return("JD_1")
    if val := mock.GetName(1); val != "JD_1" {
        t.Fatal("expected JD_1, but got", val)
    } else {
        fmt.Println("the result is ", val)
    }

}

//go:generate mockgen -source=./gomock_test.go -destination=./mock/mocktest_mock.go

 

 

　　5、SqlMock

　　用途：模擬數據庫請求，無需關注數據庫連接

　　介紹：實際遇到的情況是SqlMock和GORM的的測試

　　tips： mock.ExpectQuery("content").WillReturnRows(sqlmock.NewRows([]string{"id"}))   在WillReturnRows里不addRow就可以返回空行了

　　code example:

　　

type Goods struct {
    name string
}
type Repository struct {
    db *gorm.DB
}


func(p *Repository)ListAll()([]*Goods, error) {
    var goods []*Goods
    err := p.db.Find(&goods).Error
    return goods, err
}


func TestSql(t *testing.T) {
    // 使用 sqlmock.New() 創建 *sql.DB 的模擬實例和模擬控制器
    db, mock, err := sqlmock.New()
    if nil != err {
        log.Fatalf("Init sqlmock failed, err %v", err)
    }
    // create gorm link with sqlmock
    dbLink, err := gorm.Open(mysql.New(mysql.Config{
        SkipInitializeWithVersion: true,
        Conn: db,
    }), &gorm.Config{})
    if nil != err {
        log.Fatalf("Init DB with sqlmock failed, err %v", err)
    }

    repo := Repository{
        db: dbLink,
    }

    // create mock record
    const sqlSelectAll = `SELECT * FROM "blogs"`
    mock.ExpectQuery(sqlSelectAll).WillReturnRows(sqlmock.NewRows(nil))

    goods, err := repo.ListAll()
    assert.Nil(t, err)
    assert.Nil(t, goods)
}

 

 

　　所有UT代碼地址：this