Files
3x-ui/internal/sub/service_exclude_from_sub_test.go
T
Duxxie c3394fefed feat(sub): opt-in dns and sniffer from remote Clash rules (#6868)
Public Mihomo templates ship rules, rule-providers and proxy-groups together
with a dns block those rules depend on (fake-ip ranges, nameserver policy).
Pasting a template inline keeps its dns, but pointing the global routing rules
at its URL dropped it, so URL users had to give up auto-refresh to keep it.

A new switch, subClashRemoteDNS (off by default), also takes the remote dns and
sniffer maps. It is opt-in because a remote dns block decides where clients send
every lookup. A malformed dns or sniffer value is skipped, not the whole template.
The component-test teardown now outlasts AntD Form's 10ms error debounce, which
otherwise fired after jsdom was gone.
2026-10-24 04:21:04 +02:00

78 lines
2.5 KiB
Go

package sub
import (
"strings"
"testing"
"github.com/mhsanaei/3x-ui/v3/internal/database"
"github.com/mhsanaei/3x-ui/v3/internal/xray"
)
// An excluded inbound keeps serving its clients, so every renderer must drop
// its links yet still count its clients' usage in the Subscription-Userinfo header.
func TestExcludedInboundHidesLinksButKeepsUsage(t *testing.T) {
seedSubDB(t)
db := database.GetDB()
shown := seedSubInbound(t, "sub-excl", "shown", 24401, 1, `{"network":"tcp","security":"none"}`)
hidden := seedSubInbound(t, "sub-excl", "hidden", 24402, 2, `{"network":"tcp","security":"none"}`)
if err := db.Model(hidden).Update("exclude_from_sub", true).Error; err != nil {
t.Fatalf("mark excluded: %v", err)
}
for _, row := range []*xray.ClientTraffic{
{InboundId: shown.Id, Email: "shown@e", Up: 100, Down: 200, Enable: true},
{InboundId: hidden.Id, Email: "hidden@e", Up: 1000, Down: 2000, Enable: true},
} {
if err := db.Create(row).Error; err != nil {
t.Fatalf("seed traffic %s: %v", row.Email, err)
}
}
const wantHeader = "upload=1100; download=2200; "
assertOnlyShown := func(t *testing.T, out string) {
t.Helper()
if !strings.Contains(out, "24401") {
t.Fatalf("output lost the shown inbound:\n%s", out)
}
if strings.Contains(out, "24402") {
t.Fatalf("output leaked the excluded inbound:\n%s", out)
}
}
t.Run("raw", func(t *testing.T) {
links, _, _, traffic, err := NewSubService("").GetSubs("sub-excl", "req.example.com")
if err != nil {
t.Fatalf("GetSubs: %v", err)
}
if len(links) != 1 {
t.Fatalf("links = %q, want only the shown inbound's link", links)
}
assertOnlyShown(t, links[0])
if traffic.Up != 1100 || traffic.Down != 2200 {
t.Fatalf("usage = up %d/down %d, want 1100/2200 including the excluded inbound's client", traffic.Up, traffic.Down)
}
})
t.Run("clash", func(t *testing.T) {
out, header, err := NewSubClashService(false, "", false, NewSubService("")).GetClash("sub-excl", "req.example.com")
if err != nil {
t.Fatalf("GetClash: %v", err)
}
assertOnlyShown(t, out)
if !strings.HasPrefix(header, wantHeader) {
t.Fatalf("header = %q, want prefix %q", header, wantHeader)
}
})
t.Run("json", func(t *testing.T) {
out, header, err := NewSubJsonService("", "", "", "", NewSubService("")).GetJson("sub-excl", "req.example.com", false)
if err != nil {
t.Fatalf("GetJson: %v", err)
}
assertOnlyShown(t, out)
if !strings.HasPrefix(header, wantHeader) {
t.Fatalf("header = %q, want prefix %q", header, wantHeader)
}
})
}