Files
3x-ui-agent/internal/web/service/traffic_runtime_apply_test.go
T
Muse 1035d00fe4 精简版P9(4b): 删除本地核心整条路径(进程管理/buildXrayConfig本地分支/local-only注入/本地核心接口与任务)
- service/xray.go 1741->573行: 只剩 agent 配置构建(GetXrayConfigForAgent/buildAgentXrayConfig=模板+节点入站+egress+转发编译); 删进程生命周期/状态机/RestartXray/StopXray/SetToNeedRestart/GetXrayTraffic/GetOnlineUsers/balancer/TestRoute 与 injectPanelEgress/injectNodeEgresses/injectMtproto/injectAmneziawg/mergeSubscriptionOutbounds
- controller: 43处 SetToNeedRestart 与各 RestartXray 调用清除; 删 balancerStatus/balancerOverride/routeTest/getXrayResult 路由与handler(纯本地核心操作); 前端删 RouteTester 页签与组件
- server service: GetConfigJson 改模板直读; 状态Xray块恒 stopped/空版本; Stop/RestartXrayService 改 no-op(DB导入/恢复流程保留形状)
- setting service: PanelEgressProxyURL/NodeEgressProxyURL 恒直连(桥在本地核心内,已不存在); NodeService.withOutboundBridge 恒直连回退
- runtime: Local 改为全 no-op 桩(DB即真源); NewManager() 去 LocalDeps(APIPort/SetNeedRestart)
- job: 删 XrayTrafficJob/CheckXrayRunningJob/CheckClientIpJob 三文件(共享声明 EventBus/IPWithTimestamp/clientStatsSnapshotMaxClients 迁 shared.go); web.go 不再调度 MtprotoJob/AmneziaWGJob, 删 OnCrash 接线与 stop() 的 sidecar StopAll
- MTProto/AmneziaWG 下线: AddInbound/UpdateInbound 对两协议直接拒绝(存量行保留可删); TG bot 的 restart 命令与在线查询按无本地核心收口
- 测试: GetXrayConfig 系列移植到 agent 构建(finalmask/clients/wireguard/scale); 删退役行为测试(lifecycle/restart/mtproto本地应用/client-ip job/inject系列); 共享测试helper迁 mtproto_test_helpers_test.go 与 job/integration_helpers_test.go
- 既有红(非本轮引入,已在 f6d6520 复现): TestGetXrayConfig_EnabledClientsStillEmitted、TestRouteRegistryContract(P6删 endpoints.ts 遗留)
2026-10-21 18:43:55 +00:00

87 lines
3.0 KiB
Go

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()
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()
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)
}
}