Browse Source

Add a flag for which host to hit in TestTransport

Brad Fitzpatrick 11 years ago
parent
commit
07ee680c59
1 changed files with 6 additions and 4 deletions
  1. 6 4
      transport_test.go

+ 6 - 4
transport_test.go

@@ -12,15 +12,17 @@ import (
 	"testing"
 )
 
-var extNet = flag.Bool("extnet", false, "do external network tests")
+var (
+	extNet        = flag.Bool("extnet", false, "do external network tests")
+	transportHost = flag.String("transporthost", "http2.golang.org", "hostname to use for TestTransport")
+)
 
 func TestTransport(t *testing.T) {
 	if !*extNet {
 		t.Skip("skipping external network test")
 	}
-	req, _ := http.NewRequest("GET", "https://http2.golang.org/", nil)
-	var rt http.RoundTripper = &Transport{}
-	//rt = http.DefaultTransport
+	req, _ := http.NewRequest("GET", "https://"+*transportHost+"/", nil)
+	rt := &Transport{}
 	res, err := rt.RoundTrip(req)
 	if err != nil {
 		t.Fatalf("%v", err)