如何把网站做好,做360网站优化快速,怎么开发网赌软件,长沙建站公司招聘文章目录 一、 sqlmock介绍二、安装三、基本用法四、一个小案例五、Gorm 初始化注意点 一、 sqlmock介绍
sqlmock 是一个用于测试数据库交互的 Go 模拟库。它可以模拟 SQL 查询、插入、更新等操作#xff0c;并且可以验证 SQL 语句的执行情况#xff0c;非常适合用于单元测试… 文章目录 一、 sqlmock介绍二、安装三、基本用法四、一个小案例五、Gorm 初始化注意点 一、 sqlmock介绍
sqlmock 是一个用于测试数据库交互的 Go 模拟库。它可以模拟 SQL 查询、插入、更新等操作并且可以验证 SQL 语句的执行情况非常适合用于单元测试中。
二、安装
go get github.com/DATA-DOG/go-sqlmock三、基本用法
使用 sqlmock 进行 MySQL 数据库集成测试的基本步骤如下
创建模拟 DB 连接
import (database/sqltestinggithub.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 查询和预期结果
// 模拟 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)
}四、一个小案例
这里我们定义了一个 GORMUserDAO 结构体它实现了 UserDAO 接口用于与用户表进行交互。这个结构体通过 gorm.DB 实例与数据库进行通信。
具体来说GORMUserDAO 提供了 Insert 方法用于在数据库中创建新用户。这个方法接受一个 User 类型的结构体作为参数该结构体定义了用户的基本信息包括 ID、邮箱、密码、手机号、生日、昵称、自我介绍、微信 UnionID 和 OpenID 等字段。
在 Insert 方法中首先获取当前时间戳以毫秒为单位并设置用户的创建时间和更新时间。然后使用 gorm.DB 的 Create 方法将用户信息插入到数据库中。如果插入操作遇到唯一性约束错误例如邮箱或手机号已存在方法会返回一个特定的错误 ErrUserDuplicate。
User 结构体定义了数据库表的结构其中包含了一些列的定义如 Email 和 Phone 被设置为唯一索引。此外还定义了一些列的类型和约束如 AboutMe 字段被设置为最大长度为 1024 的字符串类型。
提供了一个使用 GORM 进行数据库操作的 DAO 层用于处理用户数据的创建。
// internal/user/dao/user.go
package daoimport (contextdatabase/sqlerrorsgithub.com/go-sql-driver/mysqlgorm.io/gormtime
)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 nowu.Ctime nowerr : dao.db.WithContext(ctx).Create(u).Errorif mysqlErr, ok : err.(*mysql.MySQLError); ok {const uniqueConflictsErrNo uint16 1062if 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:uniquePassword string//Phone *stringPhone sql.NullString gorm:uniqueBirthday sql.NullInt64// 昵称Nickname sql.NullString// 自我介绍AboutMe sql.NullString gorm:typevarchar(1024)WechatUnionID sql.NullStringWechatOpenID sql.NullString gorm:unique// 创建时间Ctime int64// 更新时间Utime int64
}
接着我们用编写测试用例
package daoimport (contextdatabase/sqlerrorsgithub.com/DATA-DOG/go-sqlmockgithub.com/go-sql-driver/mysqlgithub.com/stretchr/testify/assertgithub.com/stretchr/testify/requiregormMysql gorm.io/driver/mysqlgorm.io/gormtesting
)func TestGORMUserDAO_Insert(t *testing.T) {//testCases : []struct {name string// 为什么不用 ctrl ?// 因为你这里是 sqlmock不是 gomockmock func(t *testing.T) *sql.DBctx context.Contextuser UserwantErr 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: 123qq.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 不需要 pingDisableAutomaticPing: true,// 这个是什么呢SkipDefaultTransaction: true,})d : NewUserDAO(db)u : tc.usererr d.Insert(tc.ctx, u)assert.Equal(t, tc.wantErr, err)})}
}
五、Gorm 初始化注意点
这里运行测试的代码也有点与众不同在初始化 GORM 的时候需要额外设置三个参数。
SkipInitializeWithVersion如果为 false那么 GORM 在初始化的时候会先调用 show version。DisableAutomiticPing为 true 不允许 Ping 数据库。SkipDefaultTransaction为 false 的时候即便是一个单一增删改语句GORM 也会开启事务。
这三个选项禁用之后就可以确保 GORM 不会在初始化的过程中发起额外的调用。
