mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-03-30 04:44:49 +00:00
test: relocate best-effort usage log stub
This commit is contained in:
@@ -49,6 +49,31 @@ func newGatewayRecordUsageServiceWithBillingRepoForTest(usageRepo UsageLogReposi
|
|||||||
return svc
|
return svc
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type openAIRecordUsageBestEffortLogRepoStub struct {
|
||||||
|
UsageLogRepository
|
||||||
|
|
||||||
|
bestEffortErr error
|
||||||
|
createErr error
|
||||||
|
bestEffortCalls int
|
||||||
|
createCalls int
|
||||||
|
lastLog *UsageLog
|
||||||
|
lastCtxErr error
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *openAIRecordUsageBestEffortLogRepoStub) CreateBestEffort(ctx context.Context, log *UsageLog) error {
|
||||||
|
s.bestEffortCalls++
|
||||||
|
s.lastLog = log
|
||||||
|
s.lastCtxErr = ctx.Err()
|
||||||
|
return s.bestEffortErr
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *openAIRecordUsageBestEffortLogRepoStub) Create(ctx context.Context, log *UsageLog) (bool, error) {
|
||||||
|
s.createCalls++
|
||||||
|
s.lastLog = log
|
||||||
|
s.lastCtxErr = ctx.Err()
|
||||||
|
return false, s.createErr
|
||||||
|
}
|
||||||
|
|
||||||
func TestGatewayServiceRecordUsage_BillingUsesDetachedContext(t *testing.T) {
|
func TestGatewayServiceRecordUsage_BillingUsesDetachedContext(t *testing.T) {
|
||||||
usageRepo := &openAIRecordUsageLogRepoStub{inserted: false, err: context.DeadlineExceeded}
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: false, err: context.DeadlineExceeded}
|
||||||
userRepo := &openAIRecordUsageUserRepoStub{}
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
||||||
|
|||||||
@@ -29,31 +29,6 @@ func (s *openAIRecordUsageLogRepoStub) Create(ctx context.Context, log *UsageLog
|
|||||||
return s.inserted, s.err
|
return s.inserted, s.err
|
||||||
}
|
}
|
||||||
|
|
||||||
type openAIRecordUsageBestEffortLogRepoStub struct {
|
|
||||||
UsageLogRepository
|
|
||||||
|
|
||||||
bestEffortErr error
|
|
||||||
createErr error
|
|
||||||
bestEffortCalls int
|
|
||||||
createCalls int
|
|
||||||
lastLog *UsageLog
|
|
||||||
lastCtxErr error
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *openAIRecordUsageBestEffortLogRepoStub) CreateBestEffort(ctx context.Context, log *UsageLog) error {
|
|
||||||
s.bestEffortCalls++
|
|
||||||
s.lastLog = log
|
|
||||||
s.lastCtxErr = ctx.Err()
|
|
||||||
return s.bestEffortErr
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *openAIRecordUsageBestEffortLogRepoStub) Create(ctx context.Context, log *UsageLog) (bool, error) {
|
|
||||||
s.createCalls++
|
|
||||||
s.lastLog = log
|
|
||||||
s.lastCtxErr = ctx.Err()
|
|
||||||
return false, s.createErr
|
|
||||||
}
|
|
||||||
|
|
||||||
type openAIRecordUsageBillingRepoStub struct {
|
type openAIRecordUsageBillingRepoStub struct {
|
||||||
UsageBillingRepository
|
UsageBillingRepository
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user