ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-08-31 19:14:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Change signature of socks5 dialer

Browse Source
This commit is contained in:
9seconds
2021-03-05 18:26:13 +03:00
parent 5b46152945
commit ecfd550a95
3 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mtglib/network/socks5_test.go
+4 -2
View File
@@ -14,6 +14,7 @@ import (
type Socks5TestSuite struct {
HTTPServerTestSuite
baseDialer network.Dialer
socksListener net.Listener
socksProxy *socks5.Server
}
@@ -29,6 +30,7 @@ func (suite *Socks5TestSuite) SetupSuite() {
suite.socksProxy, _ = socks5.New(&socksConf)
suite.socksListener, _ = net.Listen("tcp", "127.0.0.1:0")
suite.baseDialer, _ = network.NewDefaultDialer(0, 0)
go suite.socksProxy.Serve(suite.socksListener)
}
@@ -45,7 +47,7 @@ func (suite *Socks5TestSuite) TestRequestFailed() {
User: url.UserPassword("user2", "password"),
Host: suite.socksListener.Addr().String(),
}
dialer, _ := network.NewSocks5Dialer(proxyURL, 0, 0)
dialer, _ := network.NewSocks5Dialer(proxyURL, suite.baseDialer)
httpClient := http.Client{
Transport: &http.Transport{
@@ -64,7 +66,7 @@ func (suite *Socks5TestSuite) TestRequestOk() {
User: url.UserPassword("user", "password"),
Host: suite.socksListener.Addr().String(),
}
dialer, _ := network.NewSocks5Dialer(proxyURL, 0, 0)
dialer, _ := network.NewSocks5Dialer(proxyURL, suite.baseDialer)
httpClient := http.Client{
Transport: &http.Transport{