Go 单位测试之Mysql数据库集成测试 - ToB企服应用市场:ToB评测及商务社交产业平台

go get github.com/DATA-DOG/go-sqlmock

复制代码

import (
"database/sql"
"testing"
"github.com/DATA-DOG/go-sqlmock"
)
func TestMyDBFunction(t *testing.T) {
db, mock, err := sqlmock.New()
if err != nil {
t.Fatalf("Error creating mock database: %v", err)
}
defer db.Close()
// 使用 mock 来替代真实的数据库连接
// db 可以传递给被测试的函数进行测试
}

复制代码

// 模拟 SQL 查询并设置预期结果
rows := sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "Alice").AddRow(2, "Bob")
mock.ExpectQuery("SELECT id, name FROM users").WillReturnRows(rows)

复制代码

// 调用被测试的函数，传入模拟的数据库连接
result := MyDBFunction(db)
// 验证结果是否符合预期
if result != expected {
t.Errorf("Expected %d, got %d", expected, result)
}

复制代码

// internal/user/dao/user.go
package dao
import (
"context"
"database/sql"
"errors"
"github.com/go-sql-driver/mysql"
"gorm.io/gorm"
"time"
)
var (
ErrUserDuplicate = errors.New("邮箱冲突")
)
type UserDAO interface {
Insert(ctx context.Context, u User) error
}
type GORMUserDAO struct {
db *gorm.DB
}
func NewUserDAO(db *gorm.DB) UserDAO {
return &GORMUserDAO{
db: db,
}
}
func (dao *GORMUserDAO) Insert(ctx context.Context, u User) error {
// 存毫秒数
now := time.Now().UnixMilli()
u.Utime = now
u.Ctime = now
err := dao.db.WithContext(ctx).Create(&u).Error
if mysqlErr, ok := err.(*mysql.MySQLError); ok {
const uniqueConflictsErrNo uint16 = 1062
if mysqlErr.Number == uniqueConflictsErrNo {
// 邮箱冲突 or 手机号码冲突
return ErrUserDuplicate
}
}
return err
}
// User 直接对应数据库表结构
// 有些人叫做 entity，有些人叫做 model，有些人叫做 PO(persistent object)
type User struct {
Id int64 `gorm:"primaryKey,autoIncrement"`
// 设置为唯一索引
Email sql.NullString `gorm:"unique"`
Password string
//Phone *string
Phone sql.NullString `gorm:"unique"`
Birthday sql.NullInt64
// 昵称
Nickname sql.NullString
// 自我介绍
AboutMe sql.NullString `gorm:"type=varchar(1024)"`
WechatUnionID sql.NullString
WechatOpenID sql.NullString `gorm:"unique"`
// 创建时间
Ctime int64
// 更新时间
Utime int64
}

复制代码

package dao
import (
"context"
"database/sql"
"errors"
"github.com/DATA-DOG/go-sqlmock"
"github.com/go-sql-driver/mysql"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
gormMysql "gorm.io/driver/mysql"
"gorm.io/gorm"
"testing"
)
func TestGORMUserDAO_Insert(t *testing.T) {
//
testCases := []struct {
name string
// 为什么不用 ctrl ?
// 因为你这里是 sqlmock，不是 gomock
mock func(t *testing.T) *sql.DB
ctx context.Context
user User
wantErr error
}{
{
name: "插入成功",
mock: func(t *testing.T) *sql.DB {
mockDB, mock, err := sqlmock.New()
res := sqlmock.NewResult(3, 1)
// 这边预期的是正则表达式
// 这个写法的意思就是，只要是 INSERT 到 users 的语句
mock.ExpectExec("INSERT INTO `users` .*").
WillReturnResult(res)
require.NoError(t, err)
return mockDB
},
user: User{
Email: sql.NullString{
String: "123@qq.com",
Valid: true,
},
},
},
{
name: "邮箱冲突",
mock: func(t *testing.T) *sql.DB {
mockDB, mock, err := sqlmock.New()
// 这边预期的是正则表达式
// 这个写法的意思就是，只要是 INSERT 到 users 的语句
mock.ExpectExec("INSERT INTO `users` .*").
WillReturnError(&mysql.MySQLError{
Number: 1062,
})
require.NoError(t, err)
return mockDB
},
user: User{},
wantErr: ErrUserDuplicate,
},
{
name: "数据库错误",
mock: func(t *testing.T) *sql.DB {
mockDB, mock, err := sqlmock.New()
// 这边预期的是正则表达式
// 这个写法的意思就是，只要是 INSERT 到 users 的语句
mock.ExpectExec("INSERT INTO `users` .*").
WillReturnError(errors.New("数据库错误"))
require.NoError(t, err)
return mockDB
},
user: User{},
wantErr: errors.New("数据库错误"),
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
db, err := gorm.Open(gormMysql.New(gormMysql.Config{
Conn: tc.mock(t),
// SELECT VERSION;
SkipInitializeWithVersion: true,
}), &gorm.Config{
// 你 mock DB 不需要 ping
DisableAutomaticPing: true,
// 这个是什么呢？
SkipDefaultTransaction: true,
})
d := NewUserDAO(db)
u := tc.user
err = d.Insert(tc.ctx, u)
assert.Equal(t, tc.wantErr, err)
})
}
}

复制代码