mirror of
https://github.com/Sagit-chu/flvx.git
synced 2026-10-01 00:26:38 +08:00
Antigravity
66be07750f
- Extract database layer into model and repo packages - Split repository into focused modules (control, federation, flow, groups, mutations) - Remove monolithic db.go and sqlite/repository.go - Update handlers to use new repository structure - Migrate contract tests to new patterns - Add migration plan documentation
64 lines
1.9 KiB
Go
64 lines
1.9 KiB
Go
package handler
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"go-backend/internal/store/repo"
|
|
)
|
|
|
|
func TestProcessFlowItemTracksPeerShareFlowAndEnforcesLimit(t *testing.T) {
|
|
r, err := repo.Open(filepath.Join(t.TempDir(), "panel.db"))
|
|
if err != nil {
|
|
t.Fatalf("open repo: %v", err)
|
|
}
|
|
defer r.Close()
|
|
|
|
now := time.Now().UnixMilli()
|
|
if err := r.CreatePeerShare(&repo.PeerShare{
|
|
Name: "flow-share",
|
|
NodeID: 1,
|
|
Token: "flow-share-token",
|
|
MaxBandwidth: 3000,
|
|
CurrentFlow: 1000,
|
|
PortRangeStart: 32000,
|
|
PortRangeEnd: 32010,
|
|
IsActive: 1,
|
|
CreatedTime: now,
|
|
UpdatedTime: now,
|
|
}); err != nil {
|
|
t.Fatalf("create peer share: %v", err)
|
|
}
|
|
share, err := r.GetPeerShareByToken("flow-share-token")
|
|
if err != nil || share == nil {
|
|
t.Fatalf("load peer share: %v", err)
|
|
}
|
|
|
|
if err := r.DB().Exec(`
|
|
INSERT INTO peer_share_runtime(id, share_id, node_id, reservation_id, resource_key, binding_id, role, chain_name, service_name, protocol, strategy, port, target, applied, status, created_time, updated_time)
|
|
VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
`, 17, share.ID, share.NodeID, "res-17", "rk-17", "17", "exit", "", "fed_svc_17", "tls", "round", 32001, "", 1, 1, now, now).Error; err != nil {
|
|
t.Fatalf("insert peer_share_runtime: %v", err)
|
|
}
|
|
|
|
h := &Handler{repo: r}
|
|
h.processFlowItem(flowItem{N: "fed_svc_17", U: 1200, D: 900})
|
|
|
|
updatedShare, err := r.GetPeerShare(share.ID)
|
|
if err != nil || updatedShare == nil {
|
|
t.Fatalf("reload share: %v", err)
|
|
}
|
|
if updatedShare.CurrentFlow != 3100 {
|
|
t.Fatalf("expected current_flow=3100, got %d", updatedShare.CurrentFlow)
|
|
}
|
|
|
|
runtime, err := r.GetPeerShareRuntimeByID(17)
|
|
if err != nil || runtime == nil {
|
|
t.Fatalf("reload runtime: %v", err)
|
|
}
|
|
if runtime.Status != 0 {
|
|
t.Fatalf("expected runtime status=0 after limit enforcement, got %d", runtime.Status)
|
|
}
|
|
}
|