偷偷摘套内射激情视频,久久精品99国产国产精,中文字幕无线乱码人妻,中文在线中文a,性爽19p

Go項目實戰(zhàn)--數(shù)據(jù)Dao層代碼的單元測試實戰(zhàn)

數(shù)據(jù)庫 MongoDB
采用sqlmock類的工具,對Dao要執(zhí)行的SQL作出預(yù)期匹配,同時Mock SQL查詢要返回的數(shù)據(jù),保證Dao方法內(nèi)部的邏輯正常執(zhí)行。

Dao的單元測試

講到數(shù)據(jù)庫的單元測試,一般有那么幾個流派

  • 專門準備一個獨立的數(shù)據(jù)庫,單元測試時讓所有測試用例讀寫這個獨立的數(shù)據(jù)庫,它的優(yōu)點是單測真的去讀寫數(shù)據(jù)庫啦,缺點嘛也顯而易見,一個項目的數(shù)據(jù)庫不是光有表就行,還得準備測試數(shù)據(jù),這個搞起來就有點麻煩,尤其是關(guān)聯(lián)性強的數(shù)據(jù),造起來更麻煩。
  • 讓項目在單元測試時訪問內(nèi)存數(shù)據(jù)庫,它的優(yōu)缺點其實跟上個差不多。
  • 采用sqlmock類的工具,對Dao要執(zhí)行的SQL作出預(yù)期匹配,同時Mock SQL查詢要返回的數(shù)據(jù),保證Dao方法內(nèi)部的邏輯正常執(zhí)行。

我們這里采用的是第三個流派,用 sqlmock 方式來做數(shù)據(jù)庫Dao的單元測試,本節(jié)的內(nèi)容大綱主要如下:

圖片圖片

這里我們會用到DataDog家開發(fā)的go-sqlmock這個工具,先來安裝一下它:

github.com/DATA-DOG/go-sqlmock

安裝過程如下:

圖片圖片

單元測試入口TestMain的設(shè)置

我們計劃在 UserDao 和 OrderDao 中找?guī)讉€典型的方法來做單元測試的實戰(zhàn),這里我們先在新建test/dao/user_test.go,創(chuàng)建完之后還不能馬上開始寫測試用例,我們再來做一下dao層單元測試的基礎(chǔ)工作。

在TestMain方法中初始化go-sqlmock ,這樣整個dao下的測試用例就都能使用它了,TestMain是在當(dāng)前package下最先運行的一個函數(shù),無論你運行哪個測試用例TestMain都會先被Go調(diào)用,所以它常用于測試基礎(chǔ)組件的初始化。

我們的TestMain的代碼如下:

var (
 mock sqlmock.Sqlmock
 err  error
 db   *sql.DB
)

func TestMain(m *testing.M) {
 db, mock, err = sqlmock.New()
if err != nil {
panic(err)
 }
// 把項目使用的DB連接換成sqlmock的DB連接
 dbMasterConn, _ := gorm.Open(mysql.New(mysql.Config{
  Conn:                      db,
  SkipInitializeWithVersion: true,
  DefaultStringSize:         0,
 }))
 dbSlaveConn, _ := gorm.Open(mysql.New(mysql.Config{
  Conn:                      db,
  SkipInitializeWithVersion: true,
  DefaultStringSize:         0,
 }))
 dao2.SetDBMasterConn(dbMasterConn)
 dao2.SetDBSlaveConn(dbSlaveConn)
 os.Exit(m.Run())
}

這里我們創(chuàng)建一個 go-sqlmock 的數(shù)據(jù)庫連接 和 mock對象,mock對象管理 db 預(yù)期要執(zhí)行的SQL,具體初始化中各個參數(shù)的作用,直接看我上面代碼里的注視吧。

因我我們項目里Dao使用的數(shù)據(jù)庫連接在包外不可訪問,所以我在這里給項目dao層里加了 SetDBMasterConn,SetDBSlaveConn兩個方法把我們原本的數(shù)據(jù)庫連接替換成了sqlmock的數(shù)據(jù)庫連接。

基礎(chǔ)設(shè)置完成后,接下來我們分別找Dao的Insert、Update、Select操作來展示怎么給他們做單元測試。

Insert 操作的單元測試

首先給UserDao的CreateUser方法做單元測試,它是用戶注冊接口的邏輯中會用到的Dao方法,其定義如下:

func (ud *UserDao) CreateUser(userInfo *do.UserBaseInfo, userPasswordHash string) (*model.User, error) {
 userModel := new(model.User)
 err := util.CopyProperties(userModel, userInfo)
if err != nil {
  err = errcode.Wrap("UserDaoCreateUserError", err)
returnnil, err
 }
 userModel.Password = userPasswordHash

 err = DBMaster().WithContext(ud.ctx).Create(userModel).Error
if err != nil {
  err = errcode.Wrap("UserDaoCreateUserError", err)
returnnil, err
 }
return userModel, nil
}

這里就不再對CreateUser這個方法里都是什么做展開了,大家直接看項目代碼吧,它的單元測試如下:

func TestUserDao_CreateUser(t *testing.T) {
    userInfo := &do.UserBaseInfo{
        Nickname:  "Slang",
        LoginName: "slang@go-mall.com",
        Verified:  0,
        Avatar:    "",
        Slogan:    "happy!",
        IsBlocked: 0,
        CreatedAt: time.Now(),
        UpdatedAt: time.Now(),
    }
    passwordHash, _ := util.BcryptPassword("123456")
    userIsDel := 0

    ud := dao2.NewUserDao(context.TODO())
    mock.ExpectBegin()
    mock.ExpectExec(regexp.QuoteMeta("INSERT INTO `users`")).
    WithArgs(userInfo.Nickname, userInfo.LoginName, passwordHash, userInfo.Verified, userInfo.Avatar,
               userInfo.Slogan, userIsDel, userInfo.IsBlocked, userInfo.CreatedAt, userInfo.UpdatedAt).
    WillReturnResult(sqlmock.NewResult(1, 1))
    mock.ExpectCommit()
    userObj, err := ud.CreateUser(userInfo, passwordHash)
    assert.Nil(t, err)
    assert.Equal(t, userInfo.LoginName, userObj.LoginName)
}

這里我們首先自己初始化了一個CreateUser會用到的數(shù)據(jù)userInfo和passwordHash,然后使用 ExpectExec 指定預(yù)期要執(zhí)行的SQL以及預(yù)期返回的結(jié)果。

這里我來說明一下sqlmock 默認使用 sqlmock.QueryMatcherRegex 作為默認的SQL匹配器,該匹配器使用mock.ExpectQuery 和 mock.ExpectExec 的參數(shù)作為正則表達式與真正執(zhí)行的SQL語句進行匹配,如果使用QueryMatcherEqual 作為匹配器的話,那么我們寫預(yù)期SQL時就要寫完整的SQL了。

我推薦用默認的匹配器就行,因為接下來的WithArgs中我們還要給SQL的 ? 占位符提供參數(shù)值,這個參數(shù)值如果數(shù)量或者類型匹配不上的話,單測依然是無法通過的。

WillReturnResult(sqlmock.NewResult(1, 1)) 這行的意思是SQL執(zhí)行后返回的 lastInsertId 是 1, 受影響行數(shù)也是 1。

拿到結(jié)果之后我們再做assert斷言,判斷結(jié)果是否符合預(yù)期。符合預(yù)期則通過,不符合的話測試用例會失敗。大家可以自己嘗試修改一下這個用例看它執(zhí)行失敗的效果。

Select 查詢的單元測試

關(guān)于SQL查詢的單元測試,和上面的區(qū)別是我們會Mock返回的結(jié)果集,這里我們拿的是OrderDao的GetUserOrders做的單元測試,代碼如下。

func TestOrderDao_GetUserOrders(t *testing.T) {
    orderDel := soft_delete.DeletedAt(0)
    now := time.Now()
    emptyPayTime := time.Date(1970, time.January, 1, 0, 0, 0, 0, time.UTC)

    orders := []*model.Order{
        {1, "12345675555", "", 1, 1, 100, 100, 0, 0, emptyPayTime, orderDel, now, now},
        {2, "12345675556", "", 1, 1, 100, 100, 0, 0, emptyPayTime, orderDel, now, now},
    }
    od := dao2.NewOrderDao(context.TODO())
    var userId int64 = 1
    offset := 10
    limit := 50
    mock.ExpectQuery(regexp.QuoteMeta("SELECT * FROM `orders`")).WithArgs(userId, orderDel, limit, offset).
    WillReturnRows(
        sqlmock.NewRows([]string{"id", "order_no", "pay_trans_id", "pay_type", "user_id", "bill_money", "pay_money",
                                  "pay_state", "order_status", "paid_at", "is_del", "created_at", "updated_at"}).
        AddRow(
            orders[0].ID, orders[0].OrderNo, orders[0].PayTransId, orders[0].PayType, orders[0].UserId, orders[0].BillMoney, orders[0].PayMoney,
            orders[0].PayState, orders[0].OrderStatus, orders[0].PaidAt, orders[0].IsDel, orders[0].CreatedAt, orders[0].UpdatedAt,
        ).AddRow(
            orders[1].ID, orders[1].OrderNo, orders[1].PayTransId, orders[1].PayType, orders[1].UserId, orders[1].BillMoney, orders[1].PayMoney,
            orders[1].PayState, orders[1].OrderStatus, orders[1].PaidAt, orders[1].IsDel, orders[1].CreatedAt, orders[1].UpdatedAt,
        ),
    )
    mock.ExpectQuery(regexp.QuoteMeta("SELECT count(*) FROM `orders`")).WithArgs(userId, orderDel).
    WillReturnRows(sqlmock.NewRows([]string{"COUNT(*)"}).AddRow(2))
    gotOrders, totalRow, err := od.GetUserOrders(userId, offset, limit)
    assert.Nil(t, err)
    assert.Equal(t, orders, gotOrders)
    assert.Equal(t, totalRow, int64(2))
}

這里我用 ExpectQuery 指定了兩個預(yù)期要執(zhí)行的SQL是為什么呢?因為GetUserOrders方法即返回了用戶訂單列表還返回了數(shù)據(jù)分頁用的totalRaws變量,大家可以試試把它刪掉看看這個單元測試能不能執(zhí)行成功,這里我可以告訴你結(jié)果會成功但又沒完全成功,會有一條Warning警告,報告出有一個執(zhí)行的SQL沒有做預(yù)期匹配。

執(zhí)行單元測試時可以用上面我教的命令,也可以用IDE自帶的測試按鈕跑來跑這個測試用例。

Update操作的單元測試

Update操作的單元測試于Insert操作的類似,我們選用OrderDao的UpdateOrderStatus 方法來做單元測試。

func TestOrderDao_UpdateOrderStatus(t *testing.T) {
 orderNewStatus := 1
 var orderId int64 = 1
 orderDel := 0
 mock.ExpectBegin()
 mock.ExpectExec(regexp.QuoteMeta("UPDATE `orders` SET")).
  WithArgs(orderNewStatus, AnyTime{}, orderId, orderDel).
  WillReturnResult(sqlmock.NewResult(1, 1))
 mock.ExpectCommit()
 od := dao2.NewOrderDao(context.TODO())
 err := od.UpdateOrderStatus(orderId, orderNewStatus)
 assert.Nil(t, err)
}

這里的AnyTime是咱們自定義的一個類型

type AnyTime struct{}

func (a AnyTime) Match(v driver.Value) bool {
 // Match 方法中:判斷字段值只要是time.Time 類型,就能驗證通過
 _, ok := v.(time.Time)
 return ok
}

其實在使用SQL完全匹配模式時才必須用它,因為參數(shù)提供的Time.Now()做為UpdatedAt的時間,這與SQL執(zhí)行時真正的UpdateAt時間是有很小的差異的,這個時候我們可以提供AnyTime做為更新時間,這樣sqlmock在做預(yù)期SQL和實際SQL的匹配時,遇到了AnyTime類型的預(yù)期值,就會按照這里指定的規(guī)則,判斷字段值只要是time.Time 類型就能驗證通過。

總結(jié)

本節(jié)代碼版本為c19.1

git fetch --tags
git checkout tags/c19.1

訪問 https://github.com/go-study-lab/go-mall/compare/c18...c19.1 可在線查看詳細的代碼更新。

責(zé)任編輯:武曉燕 來源: 網(wǎng)管叨bi叨
相關(guān)推薦

2025-05-07 09:06:03

2025-05-15 09:05:19

Go項目BDD測試

2022-04-08 09:01:56

腳本Go應(yīng)用單元

2021-04-23 07:33:10

SpringSecurity單元

2017-01-14 23:42:49

單元測試框架軟件測試

2022-08-02 08:07:24

單元測試代碼重構(gòu)

2024-01-09 08:08:12

Go單元測試系統(tǒng)

2023-07-27 08:16:51

數(shù)據(jù)訪問層項目

2023-10-28 10:10:41

2023-07-26 08:58:45

Golang單元測試

2022-10-26 08:00:49

單元測試React

2011-05-16 16:52:09

單元測試徹底測試

2017-03-30 07:56:30

測試前端代碼

2009-06-26 17:48:38

JSF項目單元測試JSFUnit

2011-07-27 17:02:12

Xcode iPhone 單元測試

2021-09-18 15:40:03

Vue單元測試命令

2023-07-28 10:27:48

Java單元測試

2017-01-14 23:26:17

單元測試JUnit測試

2017-01-16 12:12:29

單元測試JUnit

2011-06-14 15:56:42

單元測試
點贊
收藏

51CTO技術(shù)棧公眾號