mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-09-28 10:12: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.
95 lines
3.6 KiB
Go
95 lines
3.6 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"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"
|
|
)
|
|
|
|
// restoreVisionFlowForEligibleInbound must re-add Vision to a client whose flow
|
|
// was stripped while the XHTTP inbound was not yet vlessenc-encrypted, but only
|
|
// when the client's intended flow (its flow_override on a sibling) is Vision,
|
|
// only on now-eligible inbounds, and never overwriting an explicit flow.
|
|
func TestRestoreVisionFlowForEligibleInbound(t *testing.T) {
|
|
dbDir := t.TempDir()
|
|
t.Setenv("XUI_DB_FOLDER", dbDir)
|
|
dbtest.InitDB(t, filepath.Join(dbDir, "x-ui.db"))
|
|
db := database.GetDB()
|
|
|
|
const vision = "xtls-rprx-vision"
|
|
const realityStream = `{"network":"tcp","security":"reality"}`
|
|
const xhttpEnc = `{"network":"xhttp","security":"reality"}`
|
|
const encSettings = `"decryption":"mlkem768x25519plus.native.0rtt.KEY","encryption":"mlkem768x25519plus.native.0rtt.KEY"`
|
|
|
|
cs := &ClientService{}
|
|
ibSvc := &InboundService{}
|
|
|
|
// Sibling reality inbound where the client legitimately has Vision.
|
|
sibling := &model.Inbound{
|
|
Tag: "sib", Enable: true, Port: 51001, Protocol: model.VLESS, StreamSettings: realityStream,
|
|
Settings: `{"clients":[{"id":"u1","email":"keep@x","flow":"` + vision + `","subId":"s1","enable":true}]}`,
|
|
}
|
|
if err := db.Create(sibling).Error; err != nil {
|
|
t.Fatalf("create sibling: %v", err)
|
|
}
|
|
keep, _ := ibSvc.GetClients(sibling)
|
|
if err := cs.SyncInbound(nil, sibling.Id, keep); err != nil {
|
|
t.Fatalf("sync sibling: %v", err)
|
|
}
|
|
|
|
// A client with no intended Vision anywhere — must NOT be touched.
|
|
other := &model.Inbound{
|
|
Tag: "oth", Enable: true, Port: 51002, Protocol: model.VLESS, StreamSettings: realityStream,
|
|
Settings: `{"clients":[{"id":"u2","email":"none@x","subId":"s2","enable":true}]}`,
|
|
}
|
|
if err := db.Create(other).Error; err != nil {
|
|
t.Fatalf("create other: %v", err)
|
|
}
|
|
oc, _ := ibSvc.GetClients(other)
|
|
if err := cs.SyncInbound(nil, other.Id, oc); err != nil {
|
|
t.Fatalf("sync other: %v", err)
|
|
}
|
|
|
|
// The now-eligible XHTTP inbound: keep@x has empty flow (was stripped),
|
|
// none@x has empty flow (no Vision anywhere), set@x has an explicit empty
|
|
// stays empty unless intended Vision.
|
|
target := `{` + encSettings + `,"clients":[` +
|
|
`{"id":"u1","email":"keep@x","flow":"","subId":"s1","enable":true},` +
|
|
`{"id":"u2","email":"none@x","flow":"","subId":"s2","enable":true}` +
|
|
`]}`
|
|
|
|
out, changed := ibSvc.restoreVisionFlowForEligibleInbound(nil, target, xhttpEnc, model.VLESS)
|
|
if !changed {
|
|
t.Fatal("expected changed=true")
|
|
}
|
|
var parsed map[string]any
|
|
if err := json.Unmarshal([]byte(out), &parsed); err != nil {
|
|
t.Fatalf("parse out: %v", err)
|
|
}
|
|
flows := map[string]string{}
|
|
for _, c := range parsed["clients"].([]any) {
|
|
cm := c.(map[string]any)
|
|
flows[cm["email"].(string)], _ = cm["flow"].(string)
|
|
}
|
|
if flows["keep@x"] != vision {
|
|
t.Errorf("keep@x flow = %q, want Vision (intended on sibling)", flows["keep@x"])
|
|
}
|
|
if flows["none@x"] != "" {
|
|
t.Errorf("none@x flow = %q, want empty (no Vision intent)", flows["none@x"])
|
|
}
|
|
|
|
// Ineligible inbound (xhttp without encryption) must be a no-op.
|
|
noenc := `{"clients":[{"id":"u1","email":"keep@x","flow":"","subId":"s1","enable":true}]}`
|
|
if _, ch := ibSvc.restoreVisionFlowForEligibleInbound(nil, noenc, `{"network":"xhttp","security":"reality"}`, model.VLESS); ch {
|
|
t.Error("ineligible xhttp (no vlessenc) must not change")
|
|
}
|
|
// Non-VLESS must be a no-op.
|
|
if _, ch := ibSvc.restoreVisionFlowForEligibleInbound(nil, target, xhttpEnc, model.VMESS); ch {
|
|
t.Error("non-VLESS must not change")
|
|
}
|
|
}
|