Ant-Browser/backend/internal/proxy/speedtest_test.go

package proxy

import (
	"encoding/base64"
	"strings"
	"testing"

	"github.com/metacubex/mihomo/component/resolver"
)

func TestProxyConfigToMappingStandardProxy(t *testing.T) {
	t.Parallel()

	mapping, err := proxyConfigToMapping("http://user:pass@example.com:8080/path")
	if err != nil {
		t.Fatalf("proxyConfigToMapping returned error: %v", err)
	}

	if got := mapping["type"]; got != "http" {
		t.Fatalf("type = %v, want http", got)
	}
	if got := mapping["server"]; got != "example.com" {
		t.Fatalf("server = %v, want example.com", got)
	}
	if got := mapping["port"]; got != 8080 {
		t.Fatalf("port = %v, want 8080", got)
	}
	if got := mapping["username"]; got != "user" {
		t.Fatalf("username = %v, want user", got)
	}
	if got := mapping["password"]; got != "pass" {
		t.Fatalf("password = %v, want pass", got)
	}
}

func TestEnableMihomoIPv6(t *testing.T) {
	resolver.DisableIPv6 = true
	t.Cleanup(func() { resolver.DisableIPv6 = true })

	enableMihomoIPv6()

	if resolver.DisableIPv6 {
		t.Fatal("expected mihomo IPv6 resolver to be enabled")
	}
}

func TestProxyConfigToMappingEscapedCredentials(t *testing.T) {
	t.Parallel()

	mapping, err := proxyConfigToMapping("http://user%40mail:p%40ss%3Aword@example.com:8080")
	if err != nil {
		t.Fatalf("proxyConfigToMapping returned error: %v", err)
	}
	if got := mapping["username"]; got != "user@mail" {
		t.Fatalf("username = %v, want user@mail", got)
	}
	if got := mapping["password"]; got != "p@ss:word" {
		t.Fatalf("password = %v, want p@ss:word", got)
	}
}

func TestProxyEndpointDropsCredentials(t *testing.T) {
	t.Parallel()

	endpoint, err := proxyEndpoint("http://user:pass@example.com:8080")
	if err != nil {
		t.Fatalf("proxyEndpoint returned error: %v", err)
	}
	if endpoint != "example.com:8080" {
		t.Fatalf("endpoint = %q, want example.com:8080", endpoint)
	}
}

func TestProxyConfigToMappingClashYAML(t *testing.T) {
	t.Parallel()

	src := "proxies:\n  - type: vmess\n    server: test.example.com\n    port: 443\n"
	mapping, err := proxyConfigToMapping(src)
	if err != nil {
		t.Fatalf("proxyConfigToMapping returned error: %v", err)
	}

	if got := mapping["type"]; got != "vmess" {
		t.Fatalf("type = %v, want vmess", got)
	}
	if got := mapping["server"]; got != "test.example.com" {
		t.Fatalf("server = %v, want test.example.com", got)
	}
	if got := mapping["port"]; got != 443 {
		t.Fatalf("port = %v, want 443", got)
	}
	if got := mapping["name"]; got != "speedtest-proxy" {
		t.Fatalf("name = %v, want speedtest-proxy", got)
	}
}

func TestProxyConfigToMappingSSURI(t *testing.T) {
	t.Parallel()

	userinfo := base64.RawURLEncoding.EncodeToString([]byte("aes-128-gcm:secret"))
	mapping, err := proxyConfigToMapping("ss://" + userinfo + "@ptxlv6-1.hxx.top:43001#node")
	if err != nil {
		t.Fatalf("proxyConfigToMapping returned error: %v", err)
	}
	if got := mapping["type"]; got != "ss" {
		t.Fatalf("type = %v, want ss", got)
	}
	if got := mapping["server"]; got != "ptxlv6-1.hxx.top" {
		t.Fatalf("server = %v, want ptxlv6-1.hxx.top", got)
	}
	if got := mapping["port"]; got != 43001 {
		t.Fatalf("port = %v, want 43001", got)
	}
	if got := mapping["cipher"]; got != "aes-128-gcm" {
		t.Fatalf("cipher = %v, want aes-128-gcm", got)
	}
	if got := mapping["password"]; got != "secret" {
		t.Fatalf("password = %v, want secret", got)
	}
}

func TestProxyEndpointSSURIIPv6(t *testing.T) {
	t.Parallel()

	raw := base64.RawURLEncoding.EncodeToString([]byte("aes-128-gcm:secret@[2001:db8::1]:43001"))
	endpoint, err := proxyEndpoint("ss://" + raw)
	if err != nil {
		t.Fatalf("proxyEndpoint returned error: %v", err)
	}
	if endpoint != "[2001:db8::1]:43001" {
		t.Fatalf("endpoint = %q, want [2001:db8::1]:43001", endpoint)
	}
}

func TestProxyConfigToMappingUnsupportedURI(t *testing.T) {
	t.Parallel()

	if _, err := proxyConfigToMapping("vmess://example"); err == nil {
		t.Fatal("expected unsupported URI error")
	}
}

func TestDefaultProxyCheckURLsAreConfigured(t *testing.T) {
	t.Parallel()

	if strings.TrimSpace(DefaultSpeedTestURL) == "" {
		t.Fatalf("DefaultSpeedTestURL must not be empty")
	}
	if strings.TrimSpace(DefaultIPHealthURL) == "" {
		t.Fatalf("DefaultIPHealthURL must not be empty")
	}
}