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:"), &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)
}