DeviceRepository单测实现

internal/infra/repository/main_test.go

@@ -0,0 +1,38 @@
+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.User{}, &models.Device{})
+	assert.NoError(t, err, "数据库迁移不应出错")
+
+	return db
+}
+
+// TestMain 是一个特殊的函数，它会在包内的所有测试运行之前被调用。
+// 我们可以在这里进行一些全局的设置和清理工作。
+func TestMain(m *testing.M) {
+	// 在所有测试运行前可以执行一些设置代码
+
+	// 运行包中的所有测试
+	code := m.Run()
+
+	// 在所有测试运行后可以执行一些清理代码
+
+	// 退出测试
+	os.Exit(code)
+}