8f5cd667de
Go currently doesn't support the socks5h protocol, but it does know how to handle it if given a URL starting with socks5:// instead. Let's rewrite the URLs internally to make things just magically work.
144 lines
3.9 KiB
Go
144 lines
3.9 KiB
Go
package lfshttp
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestHttpsProxyFromGitConfig(t *testing.T) {
|
|
c, err := NewClient(NewContext(nil, map[string]string{
|
|
"HTTPS_PROXY": "https://proxy-from-env:8080",
|
|
}, map[string]string{
|
|
"http.proxy": "https://proxy-from-git-config:8080",
|
|
}))
|
|
require.Nil(t, err)
|
|
|
|
req, err := http.NewRequest("GET", "https://some-host.com:123/foo/bar", nil)
|
|
require.Nil(t, err)
|
|
|
|
proxyURL, err := proxyFromClient(c)(req)
|
|
assert.Equal(t, "proxy-from-git-config:8080", proxyURL.Host)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestProxyForURL(t *testing.T) {
|
|
c, err := NewClient(NewContext(nil, nil, map[string]string{
|
|
"http.proxy": "https://proxy-for-everyone:8080",
|
|
"http.https://some-host.com:123.proxy": "https://proxy-for-some-host:8080",
|
|
}))
|
|
require.Nil(t, err)
|
|
|
|
req, err := http.NewRequest("GET", "https://some-host.com:123/foo/bar", nil)
|
|
require.Nil(t, err)
|
|
|
|
proxyURL, err := proxyFromClient(c)(req)
|
|
assert.Equal(t, "proxy-for-some-host:8080", proxyURL.Host)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestHttpProxyFromGitConfig(t *testing.T) {
|
|
c, err := NewClient(NewContext(nil, map[string]string{
|
|
"HTTPS_PROXY": "https://proxy-from-env:8080",
|
|
}, map[string]string{
|
|
"http.proxy": "http://proxy-from-git-config:8080",
|
|
}))
|
|
require.Nil(t, err)
|
|
|
|
req, err := http.NewRequest("GET", "http://some-host.com:123/foo/bar", nil)
|
|
require.Nil(t, err)
|
|
|
|
proxyURL, err := proxyFromClient(c)(req)
|
|
assert.Equal(t, "proxy-from-git-config:8080", proxyURL.Host)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestProxyFromEnvironment(t *testing.T) {
|
|
c, err := NewClient(NewContext(nil, map[string]string{
|
|
"HTTPS_PROXY": "https://proxy-from-env:8080",
|
|
}, nil))
|
|
require.Nil(t, err)
|
|
|
|
req, err := http.NewRequest("GET", "https://some-host.com:123/foo/bar", nil)
|
|
require.Nil(t, err)
|
|
|
|
proxyURL, err := proxyFromClient(c)(req)
|
|
assert.Equal(t, "proxy-from-env:8080", proxyURL.Host)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestProxyIsNil(t *testing.T) {
|
|
c, _ := NewClient(nil)
|
|
|
|
req, err := http.NewRequest("GET", "http://some-host.com:123/foo/bar", nil)
|
|
require.Nil(t, err)
|
|
|
|
proxyURL, err := proxyFromClient(c)(req)
|
|
assert.Nil(t, proxyURL)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestProxyNoProxy(t *testing.T) {
|
|
c, err := NewClient(NewContext(nil, map[string]string{
|
|
"NO_PROXY": "some-host",
|
|
}, map[string]string{
|
|
"http.proxy": "https://proxy-from-git-config:8080",
|
|
}))
|
|
require.Nil(t, err)
|
|
|
|
req, err := http.NewRequest("GET", "https://some-host:8080", nil)
|
|
require.Nil(t, err)
|
|
|
|
proxyURL, err := proxyFromClient(c)(req)
|
|
assert.Nil(t, proxyURL)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestProxyNoProxyWithWildcard(t *testing.T) {
|
|
c, err := NewClient(NewContext(nil, map[string]string{
|
|
"NO_PROXY": "*.example.com",
|
|
}, map[string]string{
|
|
"http.proxy": "https://proxy-from-git-config:8080",
|
|
}))
|
|
require.Nil(t, err)
|
|
|
|
req, err := http.NewRequest("GET", "https://foo.example.com:8080", nil)
|
|
require.Nil(t, err)
|
|
|
|
proxyURL, err := proxyFromClient(c)(req)
|
|
assert.Nil(t, proxyURL)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestSocksProxyFromEnvironment(t *testing.T) {
|
|
c, err := NewClient(NewContext(nil, map[string]string{
|
|
"HTTPS_PROXY": "socks5://proxy-from-env:3128",
|
|
}, nil))
|
|
require.Nil(t, err)
|
|
|
|
req, err := http.NewRequest("GET", "https://some-host.com:123/foo/bar", nil)
|
|
require.Nil(t, err)
|
|
|
|
proxyURL, err := proxyFromClient(c)(req)
|
|
assert.Equal(t, "socks5", proxyURL.Scheme)
|
|
assert.Equal(t, "proxy-from-env:3128", proxyURL.Host)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestSocks5hProxyFromEnvironment(t *testing.T) {
|
|
c, err := NewClient(NewContext(nil, map[string]string{
|
|
"HTTPS_PROXY": "socks5h://proxy-from-env:3128",
|
|
}, nil))
|
|
require.Nil(t, err)
|
|
|
|
req, err := http.NewRequest("GET", "https://some-host.com:123/foo/bar", nil)
|
|
require.Nil(t, err)
|
|
|
|
proxyURL, err := proxyFromClient(c)(req)
|
|
assert.Equal(t, "socks5", proxyURL.Scheme)
|
|
assert.Equal(t, "proxy-from-env:3128", proxyURL.Host)
|
|
assert.Nil(t, err)
|
|
}
|