79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
|
//go:build unit
|
|||
|
|
|
|||
|
|
package testutil
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/user-management-system/internal/service"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// NewTestUser 创建一个可用的测试用户，可通过 opts 覆盖默认值。
|
|||
|
|
func NewTestUser(opts ...func(*service.User)) *service.User {
|
|||
|
|
u := &service.User{
|
|||
|
|
ID: 1,
|
|||
|
|
Email: "test@example.com",
|
|||
|
|
Username: "testuser",
|
|||
|
|
Role: "user",
|
|||
|
|
Balance: 100.0,
|
|||
|
|
Concurrency: 5,
|
|||
|
|
Status: service.StatusActive,
|
|||
|
|
CreatedAt: time.Now(),
|
|||
|
|
UpdatedAt: time.Now(),
|
|||
|
|
}
|
|||
|
|
for _, opt := range opts {
|
|||
|
|
opt(u)
|
|||
|
|
}
|
|||
|
|
return u
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewTestAccount 创建一个可用的测试账户，可通过 opts 覆盖默认值。
|
|||
|
|
func NewTestAccount(opts ...func(*service.Account)) *service.Account {
|
|||
|
|
a := &service.Account{
|
|||
|
|
ID: 1,
|
|||
|
|
Name: "test-account",
|
|||
|
|
Platform: service.PlatformAnthropic,
|
|||
|
|
Status: service.StatusActive,
|
|||
|
|
Schedulable: true,
|
|||
|
|
Concurrency: 5,
|
|||
|
|
Priority: 1,
|
|||
|
|
}
|
|||
|
|
for _, opt := range opts {
|
|||
|
|
opt(a)
|
|||
|
|
}
|
|||
|
|
return a
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewTestAPIKey 创建一个可用的测试 API Key，可通过 opts 覆盖默认值。
|
|||
|
|
func NewTestAPIKey(opts ...func(*service.APIKey)) *service.APIKey {
|
|||
|
|
groupID := int64(1)
|
|||
|
|
k := &service.APIKey{
|
|||
|
|
ID: 1,
|
|||
|
|
UserID: 1,
|
|||
|
|
Key: "sk-test-key-12345678",
|
|||
|
|
Name: "test-key",
|
|||
|
|
GroupID: &groupID,
|
|||
|
|
Status: service.StatusActive,
|
|||
|
|
CreatedAt: time.Now(),
|
|||
|
|
UpdatedAt: time.Now(),
|
|||
|
|
}
|
|||
|
|
for _, opt := range opts {
|
|||
|
|
opt(k)
|
|||
|
|
}
|
|||
|
|
return k
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewTestGroup 创建一个可用的测试分组，可通过 opts 覆盖默认值。
|
|||
|
|
func NewTestGroup(opts ...func(*service.Group)) *service.Group {
|
|||
|
|
g := &service.Group{
|
|||
|
|
ID: 1,
|
|||
|
|
Platform: service.PlatformAnthropic,
|
|||
|
|
Status: service.StatusActive,
|
|||
|
|
Hydrated: true,
|
|||
|
|
}
|
|||
|
|
for _, opt := range opts {
|
|||
|
|
opt(g)
|
|||
|
|
}
|
|||
|
|
return g
|
|||
|
|
}
|