mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-10-07 06:32:07 +03:00
MHSanaei
12d51d7195
Most tests opened a throwaway panel DB with database.InitDB, which runs the full AutoMigrate + seed on an empty file every time: ~230ms, and ~850ms under -race because GORM's reflection-heavy migration is what the detector slows most. internal/web/service does this in ~550 of its 830 tests, so the CI race job spent ~10 of its ~14.6 minutes re-migrating empty databases. internal/database/dbtest.InitDB migrates once per test process, then hands each test its own copy of that file (~130ms under -race) and registers the CloseDB cleanup. The copy then goes through InitDB like a panel restart, so every test still starts from the state a fresh install has. Tests that reopen an existing file, migrate a hand-built legacy DB or target Postgres keep calling database.InitDB. Locally under -race: internal/web/service 626s (last CI run) -> 114s, internal/sub 246s -> 35s.
97 lines
3.1 KiB
Go
97 lines
3.1 KiB
Go
package job
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/op/go-logging"
|
|
|
|
"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"
|
|
xuilogger "github.com/mhsanaei/3x-ui/v3/internal/logger"
|
|
"github.com/mhsanaei/3x-ui/v3/internal/web/runtime"
|
|
"github.com/mhsanaei/3x-ui/v3/internal/web/service"
|
|
)
|
|
|
|
func transitiveGuids(t *testing.T) []string {
|
|
t.Helper()
|
|
tree, err := (&service.NodeService{}).GetNodeTree()
|
|
if err != nil {
|
|
t.Fatalf("GetNodeTree: %v", err)
|
|
}
|
|
var out []string
|
|
for _, n := range tree {
|
|
if n.Transitive {
|
|
out = append(out, n.Guid)
|
|
}
|
|
}
|
|
return out
|
|
}
|
|
|
|
// The heartbeat skips a disabled node and never sees a deleted one, so the
|
|
// sub-nodes it had learned from them stayed on the Nodes page for good.
|
|
func TestHeartbeatDropsSubNodesOfNodesItNoLongerProbes(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
retire func(t *testing.T, nodeID int)
|
|
}{
|
|
{"disabled", func(t *testing.T, nodeID int) {
|
|
if err := (&service.NodeService{}).SetEnable(nodeID, false); err != nil {
|
|
t.Fatalf("SetEnable: %v", err)
|
|
}
|
|
}},
|
|
{"deleted", func(t *testing.T, nodeID int) {
|
|
if err := (&service.NodeService{}).Delete(nodeID); err != nil {
|
|
t.Fatalf("Delete: %v", err)
|
|
}
|
|
}},
|
|
}
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
xuilogger.InitLogger(logging.ERROR)
|
|
dbtest.InitDB(t, filepath.Join(t.TempDir(), "x-ui.db"))
|
|
runtime.SetManager(runtime.NewManager(runtime.LocalDeps{APIPort: func() int { return 0 }, SetNeedRestart: func() {}}))
|
|
t.Cleanup(func() { runtime.SetManager(nil) })
|
|
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
switch {
|
|
case strings.HasSuffix(r.URL.Path, "server/status"):
|
|
_, _ = w.Write([]byte(`{"success":true,"obj":{"panelGuid":"direct-guid","xray":{"state":"running"}}}`))
|
|
case strings.HasSuffix(r.URL.Path, "server/descendants"):
|
|
_, _ = w.Write([]byte(`{"success":true,"obj":[{"guid":"sub-guid","parentGuid":"direct-guid","name":"sub","status":"online"}]}`))
|
|
default:
|
|
_, _ = w.Write([]byte(`{"success":true}`))
|
|
}
|
|
}))
|
|
t.Cleanup(srv.Close)
|
|
host, port, _ := strings.Cut(strings.TrimPrefix(srv.URL, "http://"), ":")
|
|
portNum, _ := strconv.Atoi(port)
|
|
node := &model.Node{
|
|
Name: "direct", Scheme: "http", Address: host, Port: portNum, BasePath: "/", ApiToken: "tok",
|
|
Enable: true, Status: "unknown", AllowPrivateAddress: true, TlsVerifyMode: "verify",
|
|
}
|
|
if err := database.GetDB().Create(node).Error; err != nil {
|
|
t.Fatalf("create node: %v", err)
|
|
}
|
|
|
|
hb := NewNodeHeartbeatJob()
|
|
hb.Run()
|
|
if got := transitiveGuids(t); len(got) != 1 || got[0] != "sub-guid" {
|
|
t.Fatalf("sub-nodes after first heartbeat = %v, want [sub-guid]", got)
|
|
}
|
|
|
|
tc.retire(t, node.Id)
|
|
hb.Run()
|
|
if got := transitiveGuids(t); len(got) != 0 {
|
|
t.Fatalf("sub-nodes after the node was %s = %v, want none", tc.name, got)
|
|
}
|
|
})
|
|
}
|
|
}
|