使用闭包对GORM调用进行单元测试

package main

import (
    "testing"

    "github.com/stretchr/testify/assert"
    "gorm.io/gorm"
)

// 定义一个结构体，表示数据库中的某个实体
type User struct {
    ID   uint
    Name string
}

// 封装GORM调用的闭包
func testGORMCall(db *gorm.DB) func(*testing.T) {
    return func(t *testing.T) {
        // 在闭包中进行GORM调用
        var user User
        result := db.First(&user, 1)

        // 进行断言和验证
        assert.NoError(t, result.Error)
        assert.Equal(t, "John", user.Name)
    }
}

func TestGORMCall(t *testing.T) {
    // 创建一个内存数据库用于测试
    db, err := gorm.Open("sqlite3", ":memory:")
    if err != nil {
        t.Fatal(err)
    }
    defer db.Close()

    // 在测试函数中调用封装的闭包
    testFunc := testGORMCall(db)
    testFunc(t)
}