mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-10-06 14:12:09 +03:00
chore(nodes): run the master+node scopes in parallel and document both layers
Each enrollment scope owns its panels, ports and temp dirs, so the two run side by side; the only shared state, the process-global database handle the harness borrows for setup and for simulating a lost inbound, is now behind a mutex. On Linux the suite drops from 188s to 95s. CLAUDE.md now names the fast contract layer (node_contract_test.go, in make test-go) next to the multi-tick nodee2e layer.
This commit is contained in:
@@ -1,79 +0,0 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http/httptest"
|
||||
"net/url"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/database"
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/database/dbtest"
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/util/crypto"
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/web/runtime"
|
||||
)
|
||||
|
||||
// A node enrolled with an admin-scope token (the -getApiToken default) must
|
||||
// still store the clients its master pushes; it used to keep its own list.
|
||||
func TestMasterPushWithAdminTokenAppliesClients(t *testing.T) {
|
||||
gin.SetMode(gin.TestMode)
|
||||
dbDir := t.TempDir()
|
||||
t.Setenv("XUI_DB_FOLDER", dbDir)
|
||||
dbtest.InitDB(t, filepath.Join(dbDir, "x-ui.db"))
|
||||
prev := runtime.GetManager()
|
||||
runtime.SetManager(runtime.NewManager(runtime.LocalDeps{APIPort: func() int { return 0 }, SetNeedRestart: func() {}}))
|
||||
t.Cleanup(func() { runtime.SetManager(prev) })
|
||||
|
||||
const token = "admin-node-token"
|
||||
if err := database.GetDB().Create(&model.ApiToken{
|
||||
Name: "node", Token: crypto.HashTokenSHA256(token), Enabled: true, Scope: model.ApiScopeAdmin,
|
||||
}).Error; err != nil {
|
||||
t.Fatalf("seed token: %v", err)
|
||||
}
|
||||
var owner model.User
|
||||
if err := database.GetDB().First(&owner).Error; err != nil {
|
||||
t.Fatalf("load panel user: %v", err)
|
||||
}
|
||||
const stream = `{"network":"tcp","security":"none","tcpSettings":{"header":{"type":"none"}}}`
|
||||
stored := &model.Inbound{
|
||||
UserId: owner.Id, Tag: "in-46001", Protocol: model.VLESS, Port: 46001, Enable: true,
|
||||
Settings: `{"clients":[],"decryption":"none"}`, StreamSettings: stream, Sniffing: `{}`,
|
||||
}
|
||||
if err := database.GetDB().Create(stored).Error; err != nil {
|
||||
t.Fatalf("seed node inbound: %v", err)
|
||||
}
|
||||
|
||||
engine := gin.New()
|
||||
a := &APIController{}
|
||||
api := engine.Group("/panel/api")
|
||||
api.Use(a.checkAPIAuth, a.enforceTokenScope)
|
||||
NewInboundController(api.Group("/inbounds"))
|
||||
srv := httptest.NewServer(engine)
|
||||
defer srv.Close()
|
||||
|
||||
u, _ := url.Parse(srv.URL)
|
||||
port, _ := strconv.Atoi(u.Port())
|
||||
master := runtime.NewRemote(&model.Node{
|
||||
Id: 1, Name: "n1", Scheme: "http", Address: u.Hostname(), Port: port,
|
||||
BasePath: "/", ApiToken: token, Enable: true, AllowPrivateAddress: true,
|
||||
}, nil)
|
||||
|
||||
pushed := *stored
|
||||
pushed.Settings = `{"clients":[{"id":"7fa0b7d1-9b5f-47ad-bef2-6cb0c4a624be","email":"alice","enable":true,"subId":"s-alice"}],"decryption":"none"}`
|
||||
if err := master.UpdateInbound(context.Background(), &pushed, &pushed); err != nil {
|
||||
t.Fatalf("master push: %v", err)
|
||||
}
|
||||
|
||||
var got model.Inbound
|
||||
if err := database.GetDB().First(&got, stored.Id).Error; err != nil {
|
||||
t.Fatalf("reload node inbound: %v", err)
|
||||
}
|
||||
if !strings.Contains(got.Settings, `"alice"`) {
|
||||
t.Fatalf("node kept its own client list after a master push: %s", got.Settings)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user