feat: add http & socks5 proxy

This commit is contained in:
Zhiqiang Li
2023-09-26 12:45:30 +08:00
parent a02b32fd62
commit 2abc77d1ef
4 changed files with 104 additions and 2 deletions

41
util/http_proxy_test.go Normal file
View File

@@ -0,0 +1,41 @@
package util
import (
"github.com/stretchr/testify/assert"
"net/http"
"testing"
)
func TestHttpProxy(t *testing.T) {
proxyAddress := "http://127.0.0.1:1087"
transport, err := NewHttpProxy(proxyAddress)
assert.NoError(t, err)
assert.NotNil(t, transport)
client := &http.Client{
Transport: transport,
}
resp, err := client.Get("https://www.google.com")
assert.NoError(t, err)
assert.NotNil(t, resp)
assert.Equal(t, 200, resp.StatusCode)
}
func TestSocksProxy(t *testing.T) {
proxyAddress := "socks5://127.0.0.1:1080"
transport, err := NewSocksProxy(proxyAddress)
assert.NoError(t, err)
assert.NotNil(t, transport)
client := &http.Client{
Transport: transport,
}
resp, err := client.Get("https://www.google.com")
assert.NoError(t, err)
assert.NotNil(t, resp)
assert.Equal(t, 200, resp.StatusCode)
}