oadmin/sub2api
1
0
Fork 0
Code Issues Pull Requests 1 Actions 1 Packages Projects Releases Wiki Activity

fix(test): 修复集成测试中 Create 方法的返回值处理

Browse Source
This commit is contained in:
IanShaw027
2026-01-04 17:27:32 +08:00
parent 603b361fb9
commit 7e6a197ddb
1 changed files with 27 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
backend/internal/repository/usage_log_repo_integration_test.go
View File

@@ -50,7 +50,8 @@ func (s *UsageLogRepoSuite) createUsageLog(user *service.User, apiKey *service.A
ActualCost: cost, ActualCost: cost,
CreatedAt: createdAt, CreatedAt: createdAt,
} }
s.Require().NoError(s.repo.Create(s.ctx, log)) _, err := s.repo.Create(s.ctx, log)
s.Require().NoError(err)
return log return log
} }
@@ -72,7 +73,7 @@ func (s *UsageLogRepoSuite) TestCreate() {
ActualCost: 0.4, ActualCost: 0.4,
} }
err := s.repo.Create(s.ctx, log) _, err = s.repo.Create(s.ctx, log)
s.Require().NoError(err, "Create") s.Require().NoError(err, "Create")
s.Require().NotZero(log.ID) s.Require().NotZero(log.ID)
} }
@@ -239,7 +240,8 @@ func (s *UsageLogRepoSuite) TestDashboardStats_TodayTotalsAndPerformance() {
DurationMs: &d1, DurationMs: &d1,
CreatedAt: maxTime(todayStart.Add(2*time.Minute), now.Add(-2*time.Minute)), CreatedAt: maxTime(todayStart.Add(2*time.Minute), now.Add(-2*time.Minute)),
} }
s.Require().NoError(s.repo.Create(s.ctx, logToday), "Create logToday") _, err = s.repo.Create(s.ctx, logToday)
s.Require().NoError(err, "Create logToday")
logOld := &service.UsageLog{ logOld := &service.UsageLog{
UserID: userOld.ID, UserID: userOld.ID,
@@ -253,7 +255,8 @@ func (s *UsageLogRepoSuite) TestDashboardStats_TodayTotalsAndPerformance() {
DurationMs: &d2, DurationMs: &d2,
CreatedAt: todayStart.Add(-1 * time.Hour), CreatedAt: todayStart.Add(-1 * time.Hour),
} }
s.Require().NoError(s.repo.Create(s.ctx, logOld), "Create logOld") _, err = s.repo.Create(s.ctx, logOld)
s.Require().NoError(err, "Create logOld")
logPerf := &service.UsageLog{ logPerf := &service.UsageLog{
UserID: userToday.ID, UserID: userToday.ID,
@@ -267,7 +270,8 @@ func (s *UsageLogRepoSuite) TestDashboardStats_TodayTotalsAndPerformance() {
DurationMs: &d3, DurationMs: &d3,
CreatedAt: now.Add(-30 * time.Second), CreatedAt: now.Add(-30 * time.Second),
} }
s.Require().NoError(s.repo.Create(s.ctx, logPerf), "Create logPerf") _, err = s.repo.Create(s.ctx, logPerf)
s.Require().NoError(err, "Create logPerf")
stats, err := s.repo.GetDashboardStats(s.ctx) stats, err := s.repo.GetDashboardStats(s.ctx)
s.Require().NoError(err, "GetDashboardStats") s.Require().NoError(err, "GetDashboardStats")
@@ -479,7 +483,8 @@ func (s *UsageLogRepoSuite) TestListByModelAndTimeRange() {
ActualCost: 0.5, ActualCost: 0.5,
CreatedAt: base, CreatedAt: base,
} }
s.Require().NoError(s.repo.Create(s.ctx, log1)) _, err = s.repo.Create(s.ctx, log1)
s.Require().NoError(err)
log2 := &service.UsageLog{ log2 := &service.UsageLog{
UserID: user.ID, UserID: user.ID,
@@ -492,7 +497,8 @@ func (s *UsageLogRepoSuite) TestListByModelAndTimeRange() {
ActualCost: 0.6, ActualCost: 0.6,
CreatedAt: base.Add(30 * time.Minute), CreatedAt: base.Add(30 * time.Minute),
} }
s.Require().NoError(s.repo.Create(s.ctx, log2)) _, err = s.repo.Create(s.ctx, log2)
s.Require().NoError(err)
log3 := &service.UsageLog{ log3 := &service.UsageLog{
UserID: user.ID, UserID: user.ID,
@@ -505,7 +511,8 @@ func (s *UsageLogRepoSuite) TestListByModelAndTimeRange() {
ActualCost: 0.7, ActualCost: 0.7,
CreatedAt: base.Add(1 * time.Hour), CreatedAt: base.Add(1 * time.Hour),
} }
s.Require().NoError(s.repo.Create(s.ctx, log3)) _, err = s.repo.Create(s.ctx, log3)
s.Require().NoError(err)
startTime := base.Add(-1 * time.Hour) startTime := base.Add(-1 * time.Hour)
endTime := base.Add(2 * time.Hour) endTime := base.Add(2 * time.Hour)
@@ -591,7 +598,8 @@ func (s *UsageLogRepoSuite) TestGetUserModelStats() {
ActualCost: 0.5, ActualCost: 0.5,
CreatedAt: base, CreatedAt: base,
} }
s.Require().NoError(s.repo.Create(s.ctx, log1)) _, err = s.repo.Create(s.ctx, log1)
s.Require().NoError(err)
log2 := &service.UsageLog{ log2 := &service.UsageLog{
UserID: user.ID, UserID: user.ID,
@@ -604,7 +612,8 @@ func (s *UsageLogRepoSuite) TestGetUserModelStats() {
ActualCost: 0.2, ActualCost: 0.2,
CreatedAt: base.Add(1 * time.Hour), CreatedAt: base.Add(1 * time.Hour),
} }
s.Require().NoError(s.repo.Create(s.ctx, log2)) _, err = s.repo.Create(s.ctx, log2)
s.Require().NoError(err)
startTime := base.Add(-1 * time.Hour) startTime := base.Add(-1 * time.Hour)
endTime := base.Add(2 * time.Hour) endTime := base.Add(2 * time.Hour)
@@ -684,7 +693,8 @@ func (s *UsageLogRepoSuite) TestGetModelStatsWithFilters() {
ActualCost: 0.5, ActualCost: 0.5,
CreatedAt: base, CreatedAt: base,
} }
s.Require().NoError(s.repo.Create(s.ctx, log1)) _, err = s.repo.Create(s.ctx, log1)
s.Require().NoError(err)
log2 := &service.UsageLog{ log2 := &service.UsageLog{
UserID: user.ID, UserID: user.ID,
@@ -697,7 +707,8 @@ func (s *UsageLogRepoSuite) TestGetModelStatsWithFilters() {
ActualCost: 0.2, ActualCost: 0.2,
CreatedAt: base.Add(1 * time.Hour), CreatedAt: base.Add(1 * time.Hour),
} }
s.Require().NoError(s.repo.Create(s.ctx, log2)) _, err = s.repo.Create(s.ctx, log2)
s.Require().NoError(err)
startTime := base.Add(-1 * time.Hour) startTime := base.Add(-1 * time.Hour)
endTime := base.Add(2 * time.Hour) endTime := base.Add(2 * time.Hour)
@@ -739,7 +750,8 @@ func (s *UsageLogRepoSuite) TestGetAccountUsageStats() {
ActualCost: 0.4, ActualCost: 0.4,
CreatedAt: base.Add(12 * time.Hour), CreatedAt: base.Add(12 * time.Hour),
} }
s.Require().NoError(s.repo.Create(s.ctx, log1)) _, err = s.repo.Create(s.ctx, log1)
s.Require().NoError(err)
log2 := &service.UsageLog{ log2 := &service.UsageLog{
UserID: user.ID, UserID: user.ID,
@@ -752,7 +764,8 @@ func (s *UsageLogRepoSuite) TestGetAccountUsageStats() {
ActualCost: 0.15, ActualCost: 0.15,
CreatedAt: base.Add(36 * time.Hour), // next day CreatedAt: base.Add(36 * time.Hour), // next day
} }
s.Require().NoError(s.repo.Create(s.ctx, log2)) _, err = s.repo.Create(s.ctx, log2)
s.Require().NoError(err)
startTime := base startTime := base
endTime := base.Add(72 * time.Hour) endTime := base.Add(72 * time.Hour)