mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-09-01 07:57:14 +00:00
fix(traffic): apply maintenance side effects only after the commit lands (#6200)
* fix(traffic): apply maintenance after durable commit * fix(traffic): apply all runtime maintenance after commit --------- Co-authored-by: n0ctal <293235942+n0ctal@users.noreply.github.com>
This commit is contained in:
@@ -0,0 +1,86 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/database"
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/web/runtime"
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/xray"
|
||||
)
|
||||
|
||||
func TestTrafficDisableImmediatelyUpdatesNodeRuntime(t *testing.T) {
|
||||
setupConflictDB(t)
|
||||
nodeID, fake := setupNodeRuntime(t)
|
||||
client := model.Client{Email: "spent-node", Enable: true}
|
||||
ib := nodeInbound(t, nodeID, 46301, []model.Client{client})
|
||||
if err := database.GetDB().Create(&xray.ClientTraffic{
|
||||
InboundId: ib.Id, Email: client.Email, Enable: true, Up: 100, Total: 100,
|
||||
}).Error; err != nil {
|
||||
t.Fatalf("seed traffic: %v", err)
|
||||
}
|
||||
|
||||
if _, _, _, err := (&InboundService{}).addTrafficLocked(nil, nil); err != nil {
|
||||
t.Fatalf("addTrafficLocked: %v", err)
|
||||
}
|
||||
if got := fake.updateInbound.Load(); got != 1 {
|
||||
t.Fatalf("remote UpdateInbound calls = %d, want 1 after commit", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTrafficDisableRefreshesLocalMTProtoSidecar(t *testing.T) {
|
||||
setupConflictDB(t)
|
||||
mgr := runtime.NewManager(runtime.LocalDeps{APIPort: func() int { return 0 }})
|
||||
fake := &fakeNodeRuntime{}
|
||||
mgr.SetLocalRuntimeOverride(fake)
|
||||
runtime.SetManager(mgr)
|
||||
t.Cleanup(func() { runtime.SetManager(nil) })
|
||||
|
||||
seedInboundConflict(t, "mt-spent", "", 46302, model.MTProto, "",
|
||||
`{"clients":[{"email":"spent-mt","secret":"`+mtprotoTestSecretA+`","enable":true}]}`)
|
||||
ib := loadInboundByTag(t, "mt-spent")
|
||||
clients, err := (&InboundService{}).GetClients(ib)
|
||||
if err != nil {
|
||||
t.Fatalf("GetClients: %v", err)
|
||||
}
|
||||
if err := (&ClientService{}).SyncInbound(nil, ib.Id, clients); err != nil {
|
||||
t.Fatalf("SyncInbound: %v", err)
|
||||
}
|
||||
seedClientTraffic(t, ib.Id, "spent-mt", true)
|
||||
if err := database.GetDB().Model(&xray.ClientTraffic{}).Where("email = ?", "spent-mt").
|
||||
Updates(map[string]any{"up": 100, "total": 100}).Error; err != nil {
|
||||
t.Fatalf("deplete traffic: %v", err)
|
||||
}
|
||||
|
||||
if _, _, _, err := (&InboundService{}).addTrafficLocked(nil, nil); err != nil {
|
||||
t.Fatalf("addTrafficLocked: %v", err)
|
||||
}
|
||||
if got := fake.updateInbound.Load(); got != 1 {
|
||||
t.Fatalf("MTProto sidecar UpdateInbound calls = %d, want 1 after commit", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDelDepletedClientsCleansRuntimeAfterCommit(t *testing.T) {
|
||||
setupConflictDB(t)
|
||||
mgr := runtime.NewManager(runtime.LocalDeps{APIPort: func() int { return 0 }})
|
||||
fake := &fakeNodeRuntime{}
|
||||
mgr.SetLocalRuntimeOverride(fake)
|
||||
runtime.SetManager(mgr)
|
||||
t.Cleanup(func() { runtime.SetManager(nil) })
|
||||
|
||||
seedInboundConflict(t, "depleted-only", "", 46303, model.VLESS, `{"network":"tcp"}`,
|
||||
`{"clients":[{"email":"gone","enable":true}]}`)
|
||||
ib := loadInboundByTag(t, "depleted-only")
|
||||
seedClientTraffic(t, ib.Id, "gone", true)
|
||||
if err := database.GetDB().Model(&xray.ClientTraffic{}).Where("email = ?", "gone").
|
||||
Updates(map[string]any{"up": 100, "total": 100, "reset": 0}).Error; err != nil {
|
||||
t.Fatalf("deplete traffic: %v", err)
|
||||
}
|
||||
|
||||
if err := (&InboundService{}).DelDepletedClients(-1); err != nil {
|
||||
t.Fatalf("DelDepletedClients: %v", err)
|
||||
}
|
||||
if got := fake.delInbound.Load(); got != 1 {
|
||||
t.Fatalf("runtime DelInbound calls = %d, want 1 after commit", got)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user