升级补丁
单元测试涉及数据库交互的 go 函数时，可以通过以下步骤进行：设置测试数据库，如 sqlite；创建测试函数，如 getuser；编写单元测试，验证 getuser 函数的输出，覆盖不同的测试用例。
如何对涉及数据库交互的 Golang 函数进行单元测试
在编写涉及数据库交互的 Golang 函数时，进行单元测试至关重要。以下是使用 database/sql 包进行单元测试的步骤：
1. 设置测试数据库
立即学习“go语言免费学习笔记（深入）”；
对于此示例，我们将使用 [SQLite](https://www.sqlite.org/) 作为测试数据库。
import ( "database/sql" "log" "os" "testing" "github.com/mattn/go-sqlite3")var ( testDB *sql.DB)func setupTestDB(t *testing.T) { dbPath := ":memory:" conn, err := sqlite3.Open(dbPath) if err != nil { t.Fatalf("Could not open test database: %v", err) } testDB, err = sql.Open("sqlite3", conn) if err != nil { t.Fatalf("Could not connect to test database: %v", err) }}登录后复制2. 创建测试函数
我们将创建要测试的函数：
func GetUser(id int) (*User, error) { row := testDB.QueryRow( "SELECT * FROM users WHERE id = ?", id, ) var user User err := row.Scan(&user.ID, &user.Name, &user.Email) if err != nil { return nil, err } return &user, nil}登录后复制3. 编写单元测试
现在，让我们编写一个测试函数来验证 GetUser 函数：
func TestGetUser(t *testing.T) { setupTestDB(t) defer testDB.Close() // 编写测试用例 type testCase struct { id int input int want *User err bool } testCases := []testCase{ {id: 1, input: 1, want: &User{ID: 1, Name: "John Doe", Email: "john@example.com"}, err: false}, {id: 2, input: 2, want: nil, err: true}, } for _, tc := range testCases { user, err := GetUser(tc.input) if (tc.err && err == nil) || (!tc.err && err != nil) { t.Errorf("Expected error=%t, got %v", tc.err, err) } if tc.err { continue } if tc.want.ID != user.ID || tc.want.Name != user.Name || tc.want.Email != user.Email { t.Errorf("Expected result: %v, got: %v", tc.want, user) } }}登录后复制实战案例
以下是一份使用 database/sql 包执行单元测试的完整示例：
// test_user.gopackage mainimport ( "database/sql" "log" "os" "testing" "time" _ "github.com/mattn/go-sqlite3")type User struct { ID int Name string Email string Birthday time.Time}func setupTestDB(t *testing.T) { dbPath := ":memory:" conn, err := sqlite3.Open(dbPath) if err != nil { t.Fatalf("Could not open test database: %v", err) } testDB, err = sql.Open("sqlite3", conn) if err != nil { t.Fatalf("Could not connect to test database: %v", err) } _, err = testDB.Exec( `CREATE TABLE users ( id INTEGER PRIMARY KEY, name TEXT NOT NULL, email TEXT UNIQUE, birthday DATETIME )`, ) if err != nil { t.Fatalf("Could not create test table: %v", err) }}func TestInsertUser(t *testing.T) { setupTestDB(t) defer testDB.Close() type testCase struct { input *User want *User err bool } testCases := []testCase{ { input: &User{Name: "John Doe", Email: "john@example.com", Birthday: time.Date(1980, time.January, 1, 0, 0, 0, 0, time.UTC)}, want: &User{ID: 1, Name: "John Doe", Email: "john@example.com", Birthday: time.Date(1980, time.January, 1, 0, 0, 0, 0, time.UTC)}, err: false, }, { input: &User{Name: "Jane Doe", Email: "jane@example.com", Birthday: time.Date(1990, time.March, 8, 0, 0, 0, 0, time.UTC)}, want: &User{ID: 2, Name: "Jane Doe", Email: "jane@example.com", Birthday: time.Date(1990, time.March, 8, 0, 0, 0, 0, time.UTC)}, err: false, }, { input: &User{Name: "Invalid Name", Email: "invalid@email.com", Birthday: time.Date(2000, time.April, 1, 0, 0, 0, 0, time.UTC)}, want: nil, err: true, }, } for _, tc := range testCases { user, err := InsertUser(tc.input) if (tc.err && err == nil) || (!tc.err && err != nil) { t.Errorf("Expected error=%t, got %v", tc.err, err) } if tc.err { continue } if tc.want.ID != user.ID || tc.want.Name != user.Name || tc.want.Email != user.Email || tc.want.Birthday != user.Birthday {登录后复制以上就是如何对涉及数据库交互的 Golang 函数进行单元测试？的详细内容！
