@@ -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+
2835func 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) {
6167func 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) {
7984func 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