sockets_test.go

package sockets

import (
	"net/http"
	"net/url"
	"os"
	"testing"
)

var (
	httpProxy  = "http://proxy.example.com"
	httpsProxy = "https://proxy.example.com"
)

func TestConfigureTransportProxy(t *testing.T) {
	// roughly based on defaultHTTPClient in the docker client
	u := &url.URL{
		Scheme: "tcp",
		Host:   "docker.acme.example.com",
	}
	transport := new(http.Transport)
	err := ConfigureTransport(transport, u.Scheme, u.Host)
	if err != nil {
		t.Fatal(err)
	}
	if err := os.Setenv("HTTP_PROXY", httpProxy); err != nil {
		t.Fatal(err)
	}
	if err := os.Setenv("HTTPS_PROXY", httpsProxy); err != nil {
		t.Fatal(err)
	}
	defer func() {
		_ = os.Unsetenv("HTTP_PROXY")
		_ = os.Unsetenv("HTTPS_PROXY")
	}()

	request, err := http.NewRequest(http.MethodGet, "tcp://docker.acme.example.com:2376", nil)
	if err != nil {
		t.Fatal(err)
	}
	proxyURL, err := transport.Proxy(request)
	if err != nil {
		t.Fatal(err)
	}
	if proxyURL.String() != httpProxy {
		t.Fatalf("expected %s, got %s", httpProxy, proxyURL)
	}
}

func TestTCPProxyFromEnvironment(t *testing.T) {
	if err := os.Setenv("HTTP_PROXY", httpProxy); err != nil {
		t.Fatal(err)
	}
	if err := os.Setenv("HTTPS_PROXY", httpsProxy); err != nil {
		t.Fatal(err)
	}
	defer func() {
		_ = os.Unsetenv("HTTP_PROXY")
		_ = os.Unsetenv("HTTPS_PROXY")
	}()

	tests := []struct {
		url      string
		expected *string
	}{
		{
			url:      "example.com",
			expected: nil,
		},
		{
			url:      "ftp://example.com",
			expected: nil,
		},
		{
			url:      "unix:///var/run/docker.sock",
			expected: nil,
		},
		{
			url:      "tcp://example.com:2376",
			expected: &httpProxy,
		},
		{
			url:      "http://example.com:2375",
			expected: &httpProxy,
		},
		{
			url:      "https://example.com:2376",
			expected: &httpsProxy,
		},
	}

	for _, tc := range tests {
		tc := tc
		t.Run(tc.url, func(t *testing.T) {
			request, err := http.NewRequest(http.MethodGet, tc.url, nil)
			if err != nil {
				t.Fatal(err)
			}

			proxyURL, err := TCPProxyFromEnvironment(request)
			if err != nil {
				t.Fatal(err)
			}
			if tc.expected == nil {
				if proxyURL != nil {
					t.Fatalf("expected no proxy, got %s", proxyURL)
				}
			} else if proxyURL.String() != *tc.expected {
				t.Fatalf("expected %s, got %s", *tc.expected, proxyURL)
			}
		})
	}
}