Go单元测试对数据库CRUD进行Mock测试

前言

最近在实践中也总结了一些如何用表格驱动的方式使用 gock Mock测试外部接口调用。以及怎么对GORM做mock测试，这些等这篇学完基础后，后面再单独写文章给大家介绍。

go-sqlmockminiredisMySQLRedismock

在上一篇《Go单元测试--模拟服务请求和接口返回》中，我们介绍了如何使用httptest和gock工具进行网络测试。

除了网络依赖之外，我们在开发中也会经常用到各种数据库，比如常见的MySQL和Redis等。本文就分别举例来演示如何在编写单元测试的时候对MySQL和Redis进行mock。

go-sqlmock

sql/driver

安装

使用示例

go-sqlmockrecordStats

product_viewers

recordStatssqlmock

上面的代码中，定义了一个执行成功的测试用例和一个执行失败回滚的测试用例，确保我们代码中的每个逻辑分支都能被测试到，提高单元测试覆盖率的同时也保证了代码的健壮性。

执行单元测试，看一下最终的测试结果。

❯ go test -v
=== RUN   TestShouldUpdateStats
--- PASS: TestShouldUpdateStats (0.00s)
=== RUN   TestShouldRollbackStatUpdatesOnFailure
--- PASS: TestShouldRollbackStatUpdatesOnFailure (0.00s)
PASS
ok      golang-unit-test-demo/sqlmock_demo      0.011s

可以看到两个测试用例的结果都符合预期，单元测试通过。

go-sqlmock

miniredis

除了经常用到MySQL外，Redis在日常开发中也会经常用到。接下来的这一小节，我们将一起学习如何在单元测试中mock Redis的相关操作。

net/http/httptest

当我们为一些包含Redis操作的代码编写单元测试时就可以使用它来mock Redis操作。

安装

使用示例

github.com/go-redis/redisDoSomethingWithRedis

miniredisDoSomethingWithRedisminirediss.CheckGet()s.FastForward()

执行执行测试，查看单元测试结果：

❯ go test -v
=== RUN   ;TestDoSomethingWithRedis
--- PASS: TestDoSomethingWithRedis (0.00s)
PASS
ok      golang-unit-test-demo/miniredis_demo    0.052s

miniredis

miniredis

总结

go-sqlmockminiredis