Go 单元测试之Mysql数据库集成测试

文章目录

    • 一、 sqlmock介绍
    • 二、安装
    • 三、基本用法
    • 四、一个小案例
    • 五、Gorm 初始化注意点

一、 sqlmock介绍

sqlmock 是一个用于测试数据库交互的 Go 模拟库。它可以模拟 SQL 查询、插入、更新等操作,并且可以验证 SQL 语句的执行情况,非常适合用于单元测试中。

二、安装

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

三、基本用法

使用 sqlmock 进行 MySQL 数据库集成测试的基本步骤如下:

  1. 创建模拟 DB 连接:
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 可以传递给被测试的函数进行测试
}
  1. 设置模拟 SQL 查询和预期结果:
// 模拟 SQL 查询并设置预期结果
rows := sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "Alice").AddRow(2, "Bob")
mock.ExpectQuery("SELECT id, name FROM users").WillReturnRows(rows)
  1. 调用被测试的函数,并传入模拟的数据库连接:
// 调用被测试的函数,传入模拟的数据库连接
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.DBCreate 方法将用户信息插入到数据库中。如果插入操作遇到唯一性约束错误(例如邮箱或手机号已存在),方法会返回一个特定的错误 ErrUserDuplicate

User 结构体定义了数据库表的结构,其中包含了一些列的定义,如 EmailPhone 被设置为唯一索引。此外,还定义了一些列的类型和约束,如 AboutMe 字段被设置为最大长度为 1024 的字符串类型。

提供了一个使用 GORM 进行数据库操作的 DAO 层,用于处理用户数据的创建。

// internal/user/dao/user.go
package daoimport ("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 = 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:"unique"`Password string//Phone *stringPhone sql.NullString `gorm:"unique"`Birthday sql.NullInt64// 昵称Nickname sql.NullString// 自我介绍AboutMe       sql.NullString `gorm:"type=varchar(1024)"`WechatUnionID sql.NullStringWechatOpenID  sql.NullString `gorm:"unique"`// 创建时间Ctime int64// 更新时间Utime int64
}

接着我们用编写测试用例

package daoimport ("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,不是 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: "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 不需要 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 不会在初始化的过程中发起额外的调用。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/824547.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

基于SpringBoot+Vue社区医院服务平台(源码+文档+包运行)

一.系统概述 随着信息技术在管理上越来越深入而广泛的应用,管理信息系统的实施在技术上已逐步成熟。本文介绍了社区医院信息平台的开发全过程。通过分析社区医院信息平台管理的不足,创建了一个计算机管理社区医院信息平台的方案。文章介绍了社区医院信息…

如何在Linux CentOS部署宝塔面板并实现固定公网地址访问内网宝塔

文章目录 一、使用官网一键安装命令安装宝塔二、简单配置宝塔,内网穿透三、使用固定公网地址访问宝塔 宝塔面板作为建站运维工具,适合新手,简单好用。当我们在家里/公司搭建了宝塔,没有公网IP,但是想要在外也可以访问内…

QAnything部署Mac m1环境

本次安装时Qanything已经更新到了v1.3.3,支持纯python安装。安装过程比较简单,如下: QAnything/README_zh.md at qanything-python-v1.3.1 netease-youdao/QAnything GitHub 首先需要用Anaconda3创建隔离环境,简要说明下Anaco…

春藤实业启动SAP S/4HANA Cloud Public Edition项目,与工博科技携手数字化转型之路

3月11日,广东省春藤实业有限公司(以下简称“春藤实业”)SAP S/4HANA Cloud Public Edition(以下简称“SAP ERP公有云”)项目正式启动。春藤实业董事长陈董、联络协调项目经理慕总、内部推行项目经理陈总以及工博董事长…

酒店水电能源计量管理系统

酒店水电能源计量管理系统是一种针对酒店行业设计的能源管理系统,旨在实现对水电能源的计量、监测和管理。本文将从系统特点、构成以及带来的效益三个方面展开介绍。 系统特点 1.多元化计量:该系统能够对酒店内的水、电能源进行多元化计量,…

软件项目总体测试计划(Word原件2024)

一、 前言 (一) 背景 (二) 目的 (三) 测试目标 (四) 适用范围与读者对象 (五) 术语与缩写 二、 软件测试实施流程 (一) 测试工作总体流…

2024年MathorCup数学建模C题物流网络分拣中心货量预测及人员排班解题文档与程序

2024年第十四届MathorCup高校数学建模挑战赛 C题 物流网络分拣中心货量预测及人员排班 原题再现: 电商物流网络在订单履约中由多个环节组成,图1是一个简化的物流网络示意图。其中,分拣中心作为网络的中间环节,需要将包按照不同流…

【Python基础】MySQL

文章目录 [toc]创建数据库创建数据表数据插入数据查询数据更新 个人主页:丷从心 系列专栏:Python基础 学习指南:Python学习指南 创建数据库 import pymysqldef create_database():db pymysql.connect(hostlocalhost, userroot, passwordr…

Maven多模块管理

Maven多模块管理 在了解怎么进行Maven多模块管理之前,先聊聊为什么要进行Maven多模块管理 为什么要Maven多模块管理? 在传统的单体架构开发下,一个项目中的依赖只需要使用一个pom.xml文件管理即可。但是随着微服务的流行,将原有…

【IT运维入门(ITHW)系列】之「快速部署」第二期清单(持续更新)

ITHW是Information Technology Hello World的缩写简拼。意在提供IT领域的入门相关知识,近期给大家带来的是主流技术选型的快速部署系列,意在最大程度地简化部署过程,以便能快速体验或测试相关技术选型。 「快速部署」第一期清单 ITHW快捷部署…

Python零基础从小白打怪升级中~~~~~~~多线程

线程安全和锁 一、全局解释器锁 首先需要明确的一点是GIL并不是Python的特性,它是在实现Python解析器(CPython)时所引入的一个概念。 GIL全称global interpreter lock,全局解释器锁。 每个线程在执行的时候都需要先获取GIL,保证同一时刻只…

MR-JE-70A 三菱MR-JE伺服驱动器(750W通用型)

三菱MR-JE伺服驱动器(750W通用型) MR-JE-70A外部连接,MR-JE-70A用户手册,MR-JE-70A 三相或单相AC220V三菱通用型伺服放大器750W,配套电机HG-SN52J-S100、HG-KN73J-S100。 MR-JE-70A参数说明:伺服驱动器通用型750W,三相或单相AC200V~240V 三…

Dual-AMN论文阅读

Boosting the Speed of Entity Alignment 10: Dual Attention Matching Network with Normalized Hard Sample Mining 将实体对齐速度提高 10 倍:具有归一化硬样本挖掘的双重注意力匹配网络 ABSTRACT 寻找多源知识图谱(KG)中的等效实体是知识图谱集成的关键步骤&…

使用Flask和Flask-JWT-Extended保护API免受跨站请求攻击

在本文中,我们将探讨如何使用Flask和Flask-JWT-Extended库来保护您的API免受跨站请求攻击(CSRF)。我们将首先简要介绍CSRF攻击的概念,然后详细说明如何使用Flask-JWT-Extended库来保护您的API。 什么是跨站请求攻击(C…

CSS文本属性与字体属性

目录 文本属性 文本颜色 文本对齐 修饰文本 文本缩进 行高 字体属性 字体系列 字体大小 字体粗细 字体样式 字体/文本综合属性写法 Chrome调试工具的使用 文本属性 文本颜色 在CSS中使用color 属性用于定义文本的颜色,使用background-color设置一个盒…

Adobe Acrobat PDF 2024

Adobe Acrobat PDF 2024正式发布!支持Windows和macOS系统,新界面做了轻微调整。 下载地址 Windows客户端:https://www.123pan.com/s/f43eVv-GKZKd.html macOS客户端:https://www.123pan.com/s/f43eVv-PKZKd.html

带缓存的输入输出流(I/O)

文章目录 前言一、带缓冲的输入输出流是什么?二、使用方法 1.BufferedInputStream与BufferedOutputStream类2.BufferedReader与BufferedWriter类总结 前言 输入输出流可以视为,从A点把货物搬运至B点。那么带缓冲的意思可以视为用货车把A点的货物搬运至B点…

[ ROS入门]

00ROS实现流程大致分为5步 先创建一个工作空间;再创建一个功能包;编辑源文件;编辑配置文件;编译并执行。 01创建工作空间 02进入src创建ros包并添加依赖 上述命令,会在工作空间下生成一个功能包,该功能包依…

关于外网java后端服务访问内网minio中间件,因连接minio超时,启动失败问题

注:服务器情况:2台服务器,内网服务器包含(activemq、minio、nginx、redis、mysql、后端java服务)。外网服务器只有后端java服务,访问内网的中间件(内网服务器开放了部分指定端口) 问…

Unity应用开机自启动

使用说明 以代码设置的方式设置Unity应用开机自启动。 将下面脚本挂载到场景物体,通过UI按钮开启应用自启动和取消应用自启动,设置下次运行应用生效。 所用到的Dll下载地址:Interop.IWshRuntimeLibrary 脚本代码 using System; using Syst…