39 lines
1.2 KiB
Go
39 lines
1.2 KiB
Go
package repository_test
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
|
"github.com/stretchr/testify/assert"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// setupTestDB 是一个共享的辅助函数，用于为集成测试创建一个干净的、内存中的 SQLite 数据库实例。
|
|
func setupTestDB(t *testing.T) *gorm.DB {
|
|
// "file::memory:?cache=shared" 是 GORM 连接内存 SQLite 的标准方式，确保在同一测试中的不同连接可以访问相同的数据。
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
|
|
assert.NoError(t, err, "连接内存数据库时发生错误")
|
|
|
|
// 自动迁移所有需要的表结构
|
|
err = db.AutoMigrate(models.GetAllModels()...)
|
|
assert.NoError(t, err, "数据库迁移时发生错误")
|
|
|
|
return db
|
|
}
|
|
|
|
// TestMain 是一个特殊的函数，它会在包内的所有测试运行之前被调用。
|
|
// 我们可以在这里进行一些全局的设置和清理工作。
|
|
func TestMain(m *testing.M) {
|
|
// 在所有测试运行前可以执行一些设置代码
|
|
|
|
// 运行包中的所有测试
|
|
code := m.Run()
|
|
|
|
// 在所有测试运行后可以执行一些清理代码
|
|
|
|
// 退出测试
|
|
os.Exit(code)
|
|
}
|