hangyu.tao/crm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix_trial

Browse Source
This commit is contained in:
hangyu.tao 2026-01-16 19:04:44 +08:00
parent 8cbad66b32
commit c8f6341bc9
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
internal/storage/trial_period_storage_test.go
View File

@ -27,10 +27,10 @@ func TestUpdateTrialPeriodWithIsTrial(t *testing.T) {
endTime := now.Add(7 * 24 * time.Hour) endTime := now.Add(7 * 24 * time.Hour)
trialPeriod := models.TrialPeriod{ trialPeriod := models.TrialPeriod{
CustomerID: "test-customer-001", CustomerName: "test-customer-001",
StartTime: startTime, StartTime: startTime,
EndTime: endTime, EndTime: endTime,
IsTrial: true, IsTrial: true,
} }
createdPeriod, err := storage.CreateTrialPeriod(trialPeriod) createdPeriod, err := storage.CreateTrialPeriod(trialPeriod)
@ -111,10 +111,10 @@ func TestUpdateTrialPeriodWithStartEndTime(t *testing.T) {
endTime := now.Add(7 * 24 * time.Hour) endTime := now.Add(7 * 24 * time.Hour)
trialPeriod := models.TrialPeriod{ trialPeriod := models.TrialPeriod{
CustomerID: "test-customer-002", CustomerName: "test-customer-002",
StartTime: startTime, StartTime: startTime,
EndTime: endTime, EndTime: endTime,
IsTrial: true, IsTrial: true,
} }
createdPeriod, err := storage.CreateTrialPeriod(trialPeriod) createdPeriod, err := storage.CreateTrialPeriod(trialPeriod)
@ -190,10 +190,10 @@ func TestGetAllTrialPeriods(t *testing.T) {
now := time.Now() now := time.Now()
for i := 0; i < 3; i++ { for i := 0; i < 3; i++ {
trialPeriod := models.TrialPeriod{ trialPeriod := models.TrialPeriod{
CustomerID: "test-customer-bulk", CustomerName: "test-customer-bulk",
StartTime: now.Add(time.Duration(i) * 24 * time.Hour), StartTime: now.Add(time.Duration(i) * 24 * time.Hour),
EndTime: now.Add(time.Duration(i+7) * 24 * time.Hour), EndTime: now.Add(time.Duration(i+7) * 24 * time.Hour),
IsTrial: i%2 == 0, // Alternate true/false IsTrial: i%2 == 0, // Alternate true/false
} }
_, err := storage.CreateTrialPeriod(trialPeriod) _, err := storage.CreateTrialPeriod(trialPeriod)
if err != nil { if err != nil {
@ -235,10 +235,10 @@ func TestDeleteTrialPeriod(t *testing.T) {
// Create a trial period // Create a trial period
now := time.Now() now := time.Now()
trialPeriod := models.TrialPeriod{ trialPeriod := models.TrialPeriod{
CustomerID: "test-customer-delete", CustomerName: "test-customer-delete",
StartTime: now, StartTime: now,
EndTime: now.Add(7 * 24 * time.Hour), EndTime: now.Add(7 * 24 * time.Hour),
IsTrial: true, IsTrial: true,
} }
createdPeriod, err := storage.CreateTrialPeriod(trialPeriod) createdPeriod, err := storage.CreateTrialPeriod(trialPeriod)