mirror of
https://github.com/autc04/Retro68.git
synced 2024-11-28 21:49:33 +00:00
82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
// Copyright 2009 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package http
|
|
|
|
import (
|
|
"net/url"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
// TODO(mattn):
|
|
// test ProxyAuth
|
|
|
|
var UseProxyTests = []struct {
|
|
host string
|
|
match bool
|
|
}{
|
|
// Never proxy localhost:
|
|
{"localhost", false},
|
|
{"127.0.0.1", false},
|
|
{"127.0.0.2", false},
|
|
{"[::1]", false},
|
|
{"[::2]", true}, // not a loopback address
|
|
|
|
{"barbaz.net", false}, // match as .barbaz.net
|
|
{"foobar.com", false}, // have a port but match
|
|
{"foofoobar.com", true}, // not match as a part of foobar.com
|
|
{"baz.com", true}, // not match as a part of barbaz.com
|
|
{"localhost.net", true}, // not match as suffix of address
|
|
{"local.localhost", true}, // not match as prefix as address
|
|
{"barbarbaz.net", true}, // not match because NO_PROXY have a '.'
|
|
{"www.foobar.com", false}, // match because NO_PROXY includes "foobar.com"
|
|
}
|
|
|
|
func TestUseProxy(t *testing.T) {
|
|
ResetProxyEnv()
|
|
os.Setenv("NO_PROXY", "foobar.com, .barbaz.net")
|
|
for _, test := range UseProxyTests {
|
|
if useProxy(test.host+":80") != test.match {
|
|
t.Errorf("useProxy(%v) = %v, want %v", test.host, !test.match, test.match)
|
|
}
|
|
}
|
|
}
|
|
|
|
var cacheKeysTests = []struct {
|
|
proxy string
|
|
scheme string
|
|
addr string
|
|
key string
|
|
}{
|
|
{"", "http", "foo.com", "|http|foo.com"},
|
|
{"", "https", "foo.com", "|https|foo.com"},
|
|
{"http://foo.com", "http", "foo.com", "http://foo.com|http|"},
|
|
{"http://foo.com", "https", "foo.com", "http://foo.com|https|foo.com"},
|
|
}
|
|
|
|
func TestCacheKeys(t *testing.T) {
|
|
for _, tt := range cacheKeysTests {
|
|
var proxy *url.URL
|
|
if tt.proxy != "" {
|
|
u, err := url.Parse(tt.proxy)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
proxy = u
|
|
}
|
|
cm := connectMethod{proxy, tt.scheme, tt.addr}
|
|
if got := cm.key().String(); got != tt.key {
|
|
t.Fatalf("{%q, %q, %q} cache key = %q; want %q", tt.proxy, tt.scheme, tt.addr, got, tt.key)
|
|
}
|
|
}
|
|
}
|
|
|
|
func ResetProxyEnv() {
|
|
for _, v := range []string{"HTTP_PROXY", "http_proxy", "NO_PROXY", "no_proxy"} {
|
|
os.Setenv(v, "")
|
|
}
|
|
ResetCachedEnvironment()
|
|
}
|