DATA-DOG/go-sqlmock 是一个用于 Go 语言的模拟数据库库，旨在帮助开发者在单元测试中模拟和测试 SQL 查询，而无需连接实际的数据库。它允许开发者模拟与数据库交互的过程，验证 SQL 查询的正确性，同时避免了对实际数据库的依赖，这使得测试过程更加高效、可靠和可控。
go-sqlmock 提供了以下功能：
Life is like a journey
go-sqlmock 模拟了 database/sql 包中与数据库交互的接口，包括执行 SQL 查询、执行 SQL 更新、查询结果等功能。开发者可以使用这个库来模拟查询的返回结果、错误或执行的 SQL。go-sqlmock 可以手动指定查询结果或错误，从而测试程序在不同结果下的行为。go-sqlmock 来单元测试与数据库交互的代码逻辑，例如在使用 database/sql 进行数据库查询时，测试 SQL 查询是否正确，或者测试在不同的查询结果和错误情况下的行为。首先，go-sqlmock 提供了一个 sqlmock.New() 方法来创建一个模拟数据库连接：
package main
import (
"database/sql"
"fmt"
"github.com/DATA-DOG/go-sqlmock"
"log"
)
func main() {
// 创建 mock 数据库连接
db, mock, err := sqlmock.New()
if err != nil {
log.Fatalf("failed to create mock database: %s", err)
}
defer db.Close()
// 设置期望的 SQL 查询
mock.ExpectQuery("SELECT name FROM users WHERE id = ?").
WithArgs(1).
WillReturnRows(sqlmock.NewRows([]string{"name"}).AddRow("John Doe"))
// 执行查询
rows, err := db.Query("SELECT name FROM users WHERE id = ?", 1)
if err != nil {
log.Fatalf("query failed: %s", err)
}
defer rows.Close()
var name string
for rows.Next() {
if err := rows.Scan(&name); err != nil {
log.Fatalf("failed to scan row: %s", err)
}
fmt.Println(name) // 输出 "John Doe"
}
}
你可以通过 ExpectQuery、ExpectExec 等方法来设置期望的 SQL 查询。例如，ExpectQuery 用于设置查询语句，ExpectExec 用于设置更新或删除语句：
mock.ExpectQuery("SELECT name FROM users WHERE id = ?").
WithArgs(1).
WillReturnRows(sqlmock.NewRows([]string{"name"}).AddRow("John Doe"))
ExpectQuery：设置期望的查询语句（SELECT）。WithArgs：指定查询语句中使用的参数（这里是 1）。WillReturnRows：指定查询结果返回的内容（这里返回 "John Doe"）。如果你想模拟执行更新或删除操作，可以使用 ExpectExec：
mock.ExpectExec("UPDATE users SET name = ? WHERE id = ?").
WithArgs("Jane Doe", 1).
WillReturnResult(sqlmock.NewResult(1, 1))
ExpectExec：设置期望的 SQL 执行语句（UPDATE）。WithArgs：指定执行语句的参数。WillReturnResult：指定执行结果，NewResult 的第一个参数是生成的 ID，第二个参数是受影响的行数。你还可以模拟 SQL 执行中的错误：
mock.ExpectQuery("SELECT name FROM users WHERE id = ?").
WithArgs(999).
WillReturnError(fmt.Errorf("user not found"))
这将模拟查询失败，并返回错误 user not found。
测试结束后，你可以使用 mock.ExpectationsWereMet() 来验证是否所有的期望 SQL 都被执行了：
if err := mock.ExpectationsWereMet(); err != nil {
log.Fatalf("there were unmet expectations: %s", err)
}
package main
import (
"database/sql"
"fmt"
"github.com/DATA-DOG/go-sqlmock"
"log"
)
func main() {
// 创建 mock 数据库连接
db, mock, err := sqlmock.New()
if err != nil {
log.Fatalf("failed to create mock database: %s", err)
}
defer db.Close()
// 设置期望查询
mock.ExpectQuery("SELECT name FROM users WHERE id = ?").
WithArgs(1).
WillReturnRows(sqlmock.NewRows([]string{"name"}).AddRow("John Doe"))
// 执行查询
rows, err := db.Query("SELECT name FROM users WHERE id = ?", 1)
if err != nil {
log.Fatalf("query failed: %s", err)
}
defer rows.Close()
var name string
for rows.Next() {
if err := rows.Scan(&name); err != nil {
log.Fatalf("failed to scan row: %s", err)
}
fmt.Println(name) // 输出 "John Doe"
}
// 验证期望的 SQL 查询是否被执行
if err := mock.ExpectationsWereMet(); err != nil {
log.Fatalf("there were unmet expectations: %s", err)
}
}
go-sqlmock 是一个非常强大且灵活的 Go 库，能够在不依赖真实数据库的情况下进行单元测试，模拟 SQL 查询和数据库操作。通过使用 go-sqlmock，开发者可以有效地测试数据库交互代码，验证 SQL 查询、执行更新操作、处理 SQL 错误等，而无需担心实际数据库的复杂性或性能问题。