|
|
@@ -0,0 +1,83 @@
|
|
|
+package sub
|
|
|
+
|
|
|
+import (
|
|
|
+ "strings"
|
|
|
+ "testing"
|
|
|
+
|
|
|
+ "github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
|
|
+)
|
|
|
+
|
|
|
+func TestHostToExternalProxyMap_VlessRoute(t *testing.T) {
|
|
|
+ with := hostToExternalProxyMap(&model.Host{VlessRoute: "443"}, "d.example.com", 443)
|
|
|
+ if with["vlessRoute"] != "443" {
|
|
|
+ t.Fatalf(`ep["vlessRoute"] = %v, want "443"`, with["vlessRoute"])
|
|
|
+ }
|
|
|
+ without := hostToExternalProxyMap(&model.Host{}, "d.example.com", 443)
|
|
|
+ if _, ok := without["vlessRoute"]; ok {
|
|
|
+ t.Fatalf("empty VlessRoute must not add the key: %v", without["vlessRoute"])
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+// seedSubInbound's client UUID is 11111111-2222-4333-8444-<port>, so route 443
|
|
|
+// -> 01bb, 53 -> 0035, and a route-less host keeps 4333.
|
|
|
+func TestSub_HostVlessRoute_RawMultiHost(t *testing.T) {
|
|
|
+ seedSubDB(t)
|
|
|
+ ib := seedSubInbound(t, "s1", "vr", 4500, 1, wsTLSStream)
|
|
|
+ seedHost(t, &model.Host{InboundId: ib.Id, SortOrder: 1, Remark: "A", Address: "a.cdn.com", Port: 8443, Security: "tls", VlessRoute: "443"})
|
|
|
+ seedHost(t, &model.Host{InboundId: ib.Id, SortOrder: 2, Remark: "B", Address: "b.cdn.com", Port: 8443, Security: "tls", VlessRoute: "53"})
|
|
|
+ seedHost(t, &model.Host{InboundId: ib.Id, SortOrder: 3, Remark: "C", Address: "c.cdn.com", Port: 8443, Security: "tls"})
|
|
|
+
|
|
|
+ links, _, _, _, err := NewSubService("").GetSubs("s1", "req.example.com")
|
|
|
+ if err != nil {
|
|
|
+ t.Fatalf("GetSubs: %v", err)
|
|
|
+ }
|
|
|
+ parts := strings.Split(strings.Join(links, "\n"), "\n")
|
|
|
+ if len(parts) != 3 {
|
|
|
+ t.Fatalf("want 3 host links, got %d: %v", len(parts), parts)
|
|
|
+ }
|
|
|
+ if !strings.Contains(parts[0], "vless://11111111-2222-01bb-8444-") {
|
|
|
+ t.Fatalf("host A (route 443) must encode 01bb: %s", parts[0])
|
|
|
+ }
|
|
|
+ if !strings.Contains(parts[1], "vless://11111111-2222-0035-8444-") {
|
|
|
+ t.Fatalf("host B (route 53) must encode 0035: %s", parts[1])
|
|
|
+ }
|
|
|
+ if !strings.Contains(parts[2], "vless://11111111-2222-4333-8444-") {
|
|
|
+ t.Fatalf("host C (no route) must keep the original 3rd group: %s", parts[2])
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func TestSub_HostVlessRoute_JSON(t *testing.T) {
|
|
|
+ seedSubDB(t)
|
|
|
+ ib := seedSubInbound(t, "s1", "vrj", 4501, 1, wsTLSStream)
|
|
|
+ seedHost(t, &model.Host{InboundId: ib.Id, SortOrder: 1, Remark: "J", Address: "j.cdn.com", Port: 8443, Security: "tls", VlessRoute: "443"})
|
|
|
+
|
|
|
+ js := NewSubJsonService("", "", "", NewSubService(""))
|
|
|
+ out, _, err := js.GetJson("s1", "req.example.com")
|
|
|
+ if err != nil {
|
|
|
+ t.Fatalf("GetJson: %v", err)
|
|
|
+ }
|
|
|
+ if !strings.Contains(out, "11111111-2222-01bb-8444-") {
|
|
|
+ t.Fatalf("json outbound id should encode route 443 (01bb):\n%s", out)
|
|
|
+ }
|
|
|
+ if strings.Contains(out, "11111111-2222-4333-8444-") {
|
|
|
+ t.Fatalf("original id 3rd group must be replaced in json:\n%s", out)
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func TestSub_HostVlessRoute_Clash(t *testing.T) {
|
|
|
+ seedSubDB(t)
|
|
|
+ ib := seedSubInbound(t, "s1", "vrc", 4502, 1, wsTLSStream)
|
|
|
+ seedHost(t, &model.Host{InboundId: ib.Id, SortOrder: 1, Remark: "C", Address: "c.cdn.com", Port: 8443, Security: "tls", VlessRoute: "443"})
|
|
|
+
|
|
|
+ clash := NewSubClashService(false, "", NewSubService(""))
|
|
|
+ yaml, _, err := clash.GetClash("s1", "req.example.com")
|
|
|
+ if err != nil {
|
|
|
+ t.Fatalf("GetClash: %v", err)
|
|
|
+ }
|
|
|
+ if !strings.Contains(yaml, "11111111-2222-01bb-8444-") {
|
|
|
+ t.Fatalf("clash proxy uuid should encode route 443 (01bb):\n%s", yaml)
|
|
|
+ }
|
|
|
+ if strings.Contains(yaml, "11111111-2222-4333-8444-") {
|
|
|
+ t.Fatalf("original uuid 3rd group must be replaced in clash:\n%s", yaml)
|
|
|
+ }
|
|
|
+}
|