Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
Menu
Open sidebar
陈曦
sub2api
Commits
64b3f3ce
Commit
64b3f3ce
authored
Mar 12, 2026
by
ius
Browse files
test: relocate best-effort usage log stub
parent
6a685727
Changes
2
Hide whitespace changes
Inline
Side-by-side
backend/internal/service/gateway_record_usage_test.go
View file @
64b3f3ce
...
@@ -49,6 +49,31 @@ func newGatewayRecordUsageServiceWithBillingRepoForTest(usageRepo UsageLogReposi
...
@@ -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
{}
...
...
backend/internal/service/openai_gateway_record_usage_test.go
View file @
64b3f3ce
...
@@ -29,31 +29,6 @@ func (s *openAIRecordUsageLogRepoStub) Create(ctx context.Context, log *UsageLog
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment