48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package rongxing
|
||
|
||
import (
|
||
"net/http"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func TestNewHTTPClient_Direct(t *testing.T) {
|
||
client, err := newHTTPClient(5*time.Second, "")
|
||
if err != nil {
|
||
t.Fatalf("unexpected error: %v", err)
|
||
}
|
||
if client.Timeout != 5*time.Second {
|
||
t.Fatalf("timeout = %v", client.Timeout)
|
||
}
|
||
if client.Transport != nil && client.Transport != http.DefaultTransport {
|
||
// 直连允许 Transport 为 nil（使用 DefaultTransport）
|
||
}
|
||
}
|
||
|
||
func TestNewHTTPClient_Socks5(t *testing.T) {
|
||
client, err := newHTTPClient(3*time.Second, "socks5://rongxing-vpn:1080")
|
||
if err != nil {
|
||
t.Fatalf("unexpected error: %v", err)
|
||
}
|
||
if client.Transport == nil {
|
||
t.Fatal("expected custom transport for socks5")
|
||
}
|
||
}
|
||
|
||
func TestNewHTTPClient_HTTPProxy(t *testing.T) {
|
||
client, err := newHTTPClient(3*time.Second, "http://127.0.0.1:8888")
|
||
if err != nil {
|
||
t.Fatalf("unexpected error: %v", err)
|
||
}
|
||
if client.Transport == nil {
|
||
t.Fatal("expected custom transport for http proxy")
|
||
}
|
||
}
|
||
|
||
func TestNewHTTPClient_UnsupportedScheme(t *testing.T) {
|
||
_, err := newHTTPClient(time.Second, "ftp://x")
|
||
if err == nil {
|
||
t.Fatal("expected error for unsupported scheme")
|
||
}
|
||
}
|