Skip to content

Commit f1b4b62

Browse files
committed
fix: 修复ShareService测试失败问题
- 更新share_test.go以匹配新的NewShareService签名 - 添加UserService参数到ShareService构造函数调用 - 添加User模型到测试数据库迁移 - 创建setupTestServices辅助函数简化测试设置 - 所有测试现在正常通过
1 parent 3d85e49 commit f1b4b62

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

internal/services/share_test.go

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -19,19 +19,25 @@ func setupTestDB() *gorm.DB {
1919
}
2020

2121
// 自动迁移
22-
if err := db.AutoMigrate(&models.FileCode{}, &models.UploadChunk{}, &models.KeyValue{}); err != nil {
22+
if err := db.AutoMigrate(&models.FileCode{}, &models.UploadChunk{}, &models.KeyValue{}, &models.User{}); err != nil {
2323
panic("failed to migrate database: " + err.Error())
2424
}
2525
return db
2626
}
2727

28+
func setupTestServices(db *gorm.DB, cfg *config.Config) (*ShareService, *UserService) {
29+
storageManager := storage.NewStorageManager(cfg)
30+
userService := NewUserService(db, cfg)
31+
shareService := NewShareService(db, storageManager, cfg, userService)
32+
return shareService, userService
33+
}
34+
2835
func TestShareText(t *testing.T) {
2936
db := setupTestDB()
3037
cfg := &config.Config{
3138
DataPath: "./test_data",
3239
}
33-
storageManager := storage.NewStorageManager(cfg)
34-
service := NewShareService(db, storageManager, cfg)
40+
service, _ := setupTestServices(db, cfg)
3541

3642
// 测试分享文本
3743
fileCode, err := service.ShareText("Hello World", 1, "day")
@@ -61,8 +67,7 @@ func TestShareText(t *testing.T) {
6167
func TestGenerateCode(t *testing.T) {
6268
db := setupTestDB()
6369
cfg := &config.Config{}
64-
storageManager := storage.NewStorageManager(cfg)
65-
service := NewShareService(db, storageManager, cfg)
70+
service, _ := setupTestServices(db, cfg)
6671

6772
code1 := service.generateCode()
6873
code2 := service.generateCode()
@@ -79,8 +84,7 @@ func TestGenerateCode(t *testing.T) {
7984
func TestParseExpireInfo(t *testing.T) {
8085
db := setupTestDB()
8186
cfg := &config.Config{}
82-
storageManager := storage.NewStorageManager(cfg)
83-
service := NewShareService(db, storageManager, cfg)
87+
service, _ := setupTestServices(db, cfg)
8488

8589
// 测试天数过期
8690
expiredAt, expiredCount, usedCount := service.parseExpireInfo(7, "day")

0 commit comments

Comments
 (0)