ruochen/ez-api
1
0
Fork 0
mirror of https://github.com/EZ-Api/ez-api.git synced 2026-01-13 17:47:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(log): expand handler, config, and metrics coverage

Browse Source
This commit is contained in:
zenfun
2025-12-21 16:18:38 +08:00
parent c2c65e774b
commit 6e6c669ea0
9 changed files with 112 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/api/admin_issue_key_test.go
View File

@@ -34,7 +34,7 @@ func TestAdmin_IssueChildKeyForMaster_IssuedByAdminAndSynced(t *testing.T) {
syncService := service.NewSyncService(rdb)
masterService := service.NewMasterService(db)
adminHandler := NewAdminHandler(db, masterService, syncService)
adminHandler := NewAdminHandler(db, db, masterService, syncService)
m, _, err := masterService.CreateMaster("m1", "default", 5, 10)
if err != nil {

33
internal/api/log_handler_test.go
View File

@@ -51,7 +51,7 @@ func TestMaster_ListSelfLogs_FiltersByMaster(t *testing.T) {
t.Fatalf("create log2: %v", err)
}
mh := &MasterHandler{db: db}
mh := &MasterHandler{db: db, logDB: db}
withMaster := func(next gin.HandlerFunc) gin.HandlerFunc {
return func(c *gin.Context) {
@@ -69,7 +69,7 @@ func TestMaster_ListSelfLogs_FiltersByMaster(t *testing.T) {
if rr.Code != http.StatusOK {
t.Fatalf("expected 200, got %d body=%s", rr.Code, rr.Body.String())
}
var resp ListLogsResponse
var resp ListMasterLogsResponse
if err := json.Unmarshal(rr.Body.Bytes(), &resp); err != nil {
t.Fatalf("unmarshal: %v", err)
}
@@ -79,6 +79,31 @@ func TestMaster_ListSelfLogs_FiltersByMaster(t *testing.T) {
if resp.Items[0].KeyID != k1.ID {
t.Fatalf("expected key_id %d, got %+v", k1.ID, resp.Items[0])
}
var raw map[string]any
if err := json.Unmarshal(rr.Body.Bytes(), &raw); err != nil {
t.Fatalf("unmarshal raw: %v", err)
}
items, ok := raw["items"].([]any)
if !ok || len(items) == 0 {
t.Fatalf("expected items array in response")
}
first, ok := items[0].(map[string]any)
if !ok {
t.Fatalf("expected item object in response")
}
if _, ok := first["provider_id"]; ok {
t.Fatalf("expected provider_id to be omitted")
}
if _, ok := first["provider_type"]; ok {
t.Fatalf("expected provider_type to be omitted")
}
if _, ok := first["provider_name"]; ok {
t.Fatalf("expected provider_name to be omitted")
}
if _, ok := first["client_ip"]; ok {
t.Fatalf("expected client_ip to be omitted")
}
}
func TestAdmin_DeleteLogs_BeforeFilters(t *testing.T) {
@@ -114,7 +139,7 @@ func TestAdmin_DeleteLogs_BeforeFilters(t *testing.T) {
t.Fatalf("create log3: %v", err)
}
h := &Handler{db: db}
h := &Handler{db: db, logDB: db}
r := gin.New()
r.DELETE("/admin/logs", h.DeleteLogs)
@@ -155,7 +180,7 @@ func TestAdmin_DeleteLogs_RequiresBefore(t *testing.T) {
t.Fatalf("migrate: %v", err)
}
h := &Handler{db: db}
h := &Handler{db: db, logDB: db}
r := gin.New()
r.DELETE("/admin/logs", h.DeleteLogs)

2
internal/api/log_webhook_handler_test.go
View File

@@ -33,7 +33,7 @@ func newTestHandlerWithWebhook(t *testing.T) (*Handler, *miniredis.Miniredis) {
mr := miniredis.RunT(t)
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
sync := service.NewSyncService(rdb)
return NewHandler(db, sync, nil, rdb), mr
return NewHandler(db, db, sync, nil, rdb), mr
}
func TestLogWebhookConfigCRUD(t *testing.T) {

2
internal/api/master_tokens_handler_test.go
View File

@@ -42,7 +42,7 @@ func TestMaster_ListTokens_AndUpdateToken(t *testing.T) {
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
syncSvc := service.NewSyncService(rdb)
masterSvc := service.NewMasterService(db)
h := NewMasterHandler(db, masterSvc, syncSvc)
h := NewMasterHandler(db, db, masterSvc, syncSvc)
withMaster := func(next gin.HandlerFunc) gin.HandlerFunc {
return func(c *gin.Context) {

2
internal/api/model_handler_test.go
View File

@@ -33,7 +33,7 @@ func newTestHandlerWithRedis(t *testing.T) (*Handler, *gorm.DB, *miniredis.Minir
mr := miniredis.RunT(t)
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
sync := service.NewSyncService(rdb)
return NewHandler(db, sync, nil, rdb), db, mr
return NewHandler(db, db, sync, nil, rdb), db, mr
}
func TestCreateModel_DefaultsKindChat_AndWritesModelsMeta(t *testing.T) {

2
internal/api/provider_handler_test.go
View File

@@ -36,7 +36,7 @@ func newTestHandler(t *testing.T) (*Handler, *gorm.DB) {
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
sync := service.NewSyncService(rdb)
return NewHandler(db, sync, nil, rdb), db
return NewHandler(db, db, sync, nil, rdb), db
}
func TestCreateProvider_DefaultsVertexLocationGlobal(t *testing.T) {

4
internal/api/stats_handler_test.go
View File

@@ -72,7 +72,7 @@ func TestMasterStats_AggregatesByKeyAndModel(t *testing.T) {
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
masterSvc := service.NewMasterService(db)
syncSvc := service.NewSyncService(rdb)
h := NewMasterHandler(db, masterSvc, syncSvc)
h := NewMasterHandler(db, db, masterSvc, syncSvc)
withMaster := func(next gin.HandlerFunc) gin.HandlerFunc {
return func(c *gin.Context) {
@@ -163,7 +163,7 @@ func TestAdminStats_AggregatesByProvider(t *testing.T) {
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
masterSvc := service.NewMasterService(db)
syncSvc := service.NewSyncService(rdb)
adminHandler := NewAdminHandler(db, masterSvc, syncSvc)
adminHandler := NewAdminHandler(db, db, masterSvc, syncSvc)
r := gin.New()
r.GET("/admin/stats", adminHandler.GetAdminStats)