test(service): add golden file validation and master key tests

- Update TESTING.md to reflect current testing status and future plans
- Add golden file comparison for provider snapshot validation in sync_test.go
- Introduce master_test.go for testing Master/Key functionality
- Add testdata directory for contract testing snapshots
This commit is contained in:
zenfun
2025-12-14 23:37:16 +08:00
parent d0011f3eb2
commit 71c183a480
4 changed files with 113 additions and 8 deletions

View File

@@ -0,0 +1,69 @@
package service
import (
"testing"
"github.com/ez-api/ez-api/internal/model"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)
func newTestDB(t *testing.T) *gorm.DB {
t.Helper()
db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
if err != nil {
t.Fatalf("open sqlite: %v", err)
}
if err := db.AutoMigrate(&model.Master{}, &model.Key{}); err != nil {
t.Fatalf("migrate: %v", err)
}
return db
}
func TestMasterService_CreateAndValidateMasterKey(t *testing.T) {
db := newTestDB(t)
svc := NewMasterService(db)
m, raw, err := svc.CreateMaster("m1", "default", 2, 10)
if err != nil {
t.Fatalf("CreateMaster: %v", err)
}
if raw == "" {
t.Fatalf("expected raw master key")
}
if m.MasterKeyDigest == "" {
t.Fatalf("expected master key digest to be set")
}
got, err := svc.ValidateMasterKey(raw)
if err != nil {
t.Fatalf("ValidateMasterKey: %v", err)
}
if got.ID != m.ID {
t.Fatalf("expected master id %d, got %d", m.ID, got.ID)
}
}
func TestMasterService_IssueChildKey_RespectsLimit(t *testing.T) {
db := newTestDB(t)
svc := NewMasterService(db)
m, _, err := svc.CreateMaster("m1", "default", 1, 10)
if err != nil {
t.Fatalf("CreateMaster: %v", err)
}
_, raw1, err := svc.IssueChildKey(m.ID, "default", "chat:write")
if err != nil {
t.Fatalf("IssueChildKey #1: %v", err)
}
if raw1 == "" {
t.Fatalf("expected raw child key")
}
_, _, err = svc.IssueChildKey(m.ID, "default", "chat:write")
if err == nil {
t.Fatalf("expected child key limit error")
}
}

View File

@@ -2,6 +2,9 @@ package service
import (
"encoding/json"
"os"
"path/filepath"
"reflect"
"testing"
"github.com/alicebob/miniredis/v2"
@@ -10,6 +13,16 @@ import (
)
func TestSyncProvider_WritesSnapshotAndRouting(t *testing.T) {
goldenPath := filepath.Join("testdata", "provider_snapshot.json")
goldenRaw, err := os.ReadFile(goldenPath)
if err != nil {
t.Fatalf("read golden %s: %v", goldenPath, err)
}
var golden map[string]any
if err := json.Unmarshal(goldenRaw, &golden); err != nil {
t.Fatalf("parse golden json: %v", err)
}
mr := miniredis.RunT(t)
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
@@ -40,8 +53,10 @@ func TestSyncProvider_WritesSnapshotAndRouting(t *testing.T) {
if err := json.Unmarshal([]byte(raw), &snap); err != nil {
t.Fatalf("invalid snapshot json: %v", err)
}
if snap["google_location"] != "global" {
t.Fatalf("expected google_location=global, got %v", snap["google_location"])
for k, v := range golden {
if !reflect.DeepEqual(snap[k], v) {
t.Fatalf("snapshot mismatch for %q: got=%#v want=%#v", k, snap[k], v)
}
}
routeKey := "route:group:default:gemini-3-pro-preview"

View File

@@ -0,0 +1,13 @@
{
"id": 42,
"name": "p1",
"type": "vertex-express",
"base_url": "",
"api_key": "",
"google_location": "global",
"group": "default",
"models": ["gemini-3-pro-preview"],
"status": "active",
"auto_ban": true
}