查看计划简单信息和单测

internal/infra/repository/main_test.go

@@ -14,11 +14,11 @@ import (
 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, "连接内存数据库不应出错")
+	assert.NoError(t, err, "连接内存数据库时发生错误")
 
 	// 自动迁移所有需要的表结构
-	err = db.AutoMigrate(&models.User{}, &models.Device{})
-	assert.NoError(t, err, "数据库迁移不应出错")
+	err = db.AutoMigrate(&models.User{}, &models.Device{}, &models.SubPlan{}, &models.Task{}, &models.Plan{})
+	assert.NoError(t, err, "数据库迁移时发生错误")
 
 	return db
 }