You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
3.0 KiB
114 lines
3.0 KiB
package client
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestHTTPClientMakeHTTPDialer(t *testing.T) {
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
_, _ = w.Write([]byte("Hi!\n"))
|
|
})
|
|
ts := httptest.NewServer(handler)
|
|
defer ts.Close()
|
|
|
|
tsTLS := httptest.NewTLSServer(handler)
|
|
defer tsTLS.Close()
|
|
// This silences a TLS handshake error, caused by the dialer just immediately
|
|
// disconnecting, which we can just ignore.
|
|
tsTLS.Config.ErrorLog = log.New(ioutil.Discard, "", 0)
|
|
|
|
for _, testURL := range []string{ts.URL, tsTLS.URL} {
|
|
u, err := newParsedURL(testURL)
|
|
require.NoError(t, err)
|
|
dialFn, err := makeHTTPDialer(testURL)
|
|
require.Nil(t, err)
|
|
|
|
addr, err := dialFn(u.Scheme, u.GetHostWithPath())
|
|
require.NoError(t, err)
|
|
require.NotNil(t, addr)
|
|
}
|
|
}
|
|
|
|
func Test_parsedURL(t *testing.T) {
|
|
type test struct {
|
|
url string
|
|
expectedURL string
|
|
expectedHostWithPath string
|
|
expectedDialAddress string
|
|
}
|
|
|
|
tests := map[string]test{
|
|
"unix endpoint": {
|
|
url: "unix:///tmp/test",
|
|
expectedURL: "unix://.tmp.test",
|
|
expectedHostWithPath: "/tmp/test",
|
|
expectedDialAddress: "/tmp/test",
|
|
},
|
|
|
|
"http endpoint": {
|
|
url: "https://example.com",
|
|
expectedURL: "https://example.com",
|
|
expectedHostWithPath: "example.com",
|
|
expectedDialAddress: "example.com",
|
|
},
|
|
|
|
"http endpoint with port": {
|
|
url: "https://example.com:8080",
|
|
expectedURL: "https://example.com:8080",
|
|
expectedHostWithPath: "example.com:8080",
|
|
expectedDialAddress: "example.com:8080",
|
|
},
|
|
|
|
"http path routed endpoint": {
|
|
url: "https://example.com:8080/rpc",
|
|
expectedURL: "https://example.com:8080/rpc",
|
|
expectedHostWithPath: "example.com:8080/rpc",
|
|
expectedDialAddress: "example.com:8080",
|
|
},
|
|
}
|
|
|
|
for name, tt := range tests {
|
|
tt := tt // suppressing linter
|
|
t.Run(name, func(t *testing.T) {
|
|
parsed, err := newParsedURL(tt.url)
|
|
require.NoError(t, err)
|
|
require.Equal(t, tt.expectedDialAddress, parsed.GetDialAddress())
|
|
require.Equal(t, tt.expectedURL, parsed.GetTrimmedURL())
|
|
require.Equal(t, tt.expectedHostWithPath, parsed.GetHostWithPath())
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMakeHTTPDialerURL(t *testing.T) {
|
|
remotes := []string{"https://foo-bar.com", "http://foo-bar.com"}
|
|
|
|
for _, remote := range remotes {
|
|
u, err := newParsedURL(remote)
|
|
require.NoError(t, err)
|
|
dialFn, err := makeHTTPDialer(remote)
|
|
require.Nil(t, err)
|
|
|
|
addr, err := dialFn(u.Scheme, u.GetHostWithPath())
|
|
require.NoError(t, err)
|
|
require.NotNil(t, addr)
|
|
}
|
|
|
|
errorURLs := []string{"tcp://foo-bar.com", "ftp://foo-bar.com"}
|
|
|
|
for _, errorURL := range errorURLs {
|
|
u, err := newParsedURL(errorURL)
|
|
require.NoError(t, err)
|
|
dialFn, err := makeHTTPDialer(errorURL)
|
|
require.Nil(t, err)
|
|
|
|
addr, err := dialFn(u.Scheme, u.GetHostWithPath())
|
|
require.Error(t, err)
|
|
require.Nil(t, addr)
|
|
}
|
|
}
|