在项目里怎么给 GORM 做单元测试

真实的开发场景下我们的项目一般都会使用 ORM ，而不是原生的database/sql来完成数据库操作。在很多使用ORM工具的场景下，也可以使用go-sqlmock库 Mock数据库操作进行测试，今天这篇内容我就以 GORM 为例，讲解怎么给项目中的 ORM 数据库操作做单元测试。,为了场景足够真实，我用 2020 年我更新的 「Go Web 编程入门」项目中的例子给大家演示怎么为使用了 GORM 的 DAO 层逻辑做 Mock 测试。,这里使用的GORM版本为 1.x，有可能在2.x版本下不兼容。,在这个例子中我们有一个与 users 表：,以及几个使用 User 的 DAO 函数：,接下来我们就用 go-sqlmock 工具给这几个 DAO 函数做一下 Mock 测试。,首先我们需要做一下测试的初始化工作，主要是设置Mock的DB连接，因为要给三个方法做Mock测试，最简单的办法是在三个方法里每次都初始化一遍 Mock 的 DB 连接，不过这么做实在是显得有点蠢，这里给大家再介绍一个小技巧。,Go 的测试支持在包内优先执行一个 TestMain(m *testing.M) 函数，可以在这里为 package 下所有测试做一些初始化的工作。,下面是我们为本次测试做的初始化工作。,准备工作做好了，下面正式对 DAO 操作进行Mock测试。,首先对 GORM 的Create 方法进行Mock测试。,因为 sqlmock 使用的是 QueryMatcherEqual 匹配器，所以，预期会执行的 SQL 语句必须精确匹配要执行的SQL(包括符号和空格)。,这个SQL怎么获取呢?其实我们先随便写一个SQL，执行一次测试，在报错信息里就会告知CreateUser操作在写表时 GORM 真正要执行的 SQL 啦。还有一种方法是通过GORM提供的Debug()方法获取到。,比如运行一下下面这个设置了Debug()的创建用户操作，GORM就会打印出执行的语句。,我们执行下这个测试,可以看到，测试函数执行成功，我们还可以故意把SQL改错，做一下反向测试。这个就留给你们自己练习啦，结合上表格测试分别做一下正向和反向单元测试。,GORM 的查询操作的Mock测试跟Create类似。,这里就不在文章里运行演示啦，有兴趣的自己把代码拿下来试一下。,GORM的Update操作我没有测试成功，我这里发出来，大家看一下原因。,运行测试后，会有下面的报错信息：,GORM 在UPDATE 的时候会自动更新updated_at 字段为当前时间，与这里withArgs传递的 time.Now() 参数不一致(毫秒级的差距也不行)。,目前没有办法 Mock 测试 GORM 的UPDATE，除非用 GORM 的 Exec 方法直接执行要更新的SQL，不过那就失去使用ORM的意义了，所以这个先跳过，如果有这方面经验的大佬，可以在留言里指导一下。,这篇内容我们把ORM的 Mock 测试做了一个讲解，这个也是我在学习 Go 单元测试时自己的思考，希望学习到的这些技能能在项目中真实用到。,因为文章中的示例，是以我之前的Go Web 编程教程里的项目里做的测试，源码我也打包更新到了Go Web 编程的项目中啦，公众号私信 gohttp15 就能获得。,如果你觉得有用，可以点赞、在看、分享给更多人，谢谢各位的支持，后面会与时俱进再搞一篇 Go 1.18 Fuzing 测试的使用介绍。