| // Copyright 2011 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 proxy |
| |
| import ( |
| "net" |
| "net/url" |
| "testing" |
| ) |
| |
| type testDialer struct { |
| network, addr string |
| } |
| |
| func (t *testDialer) Dial(network, addr string) (net.Conn, error) { |
| t.network = network |
| t.addr = addr |
| return nil, t |
| } |
| |
| func (t *testDialer) Error() string { |
| return "testDialer " + t.network + " " + t.addr |
| } |
| |
| func TestFromURL(t *testing.T) { |
| u, err := url.Parse("socks5://user:password@1.2.3.4:5678") |
| if err != nil { |
| t.Fatalf("failed to parse URL: %s", err) |
| } |
| |
| tp := &testDialer{} |
| proxy, err := FromURL(u, tp) |
| if err != nil { |
| t.Fatalf("FromURL failed: %s", err) |
| } |
| |
| conn, err := proxy.Dial("tcp", "example.com:80") |
| if conn != nil { |
| t.Error("Dial unexpected didn't return an error") |
| } |
| if tp, ok := err.(*testDialer); ok { |
| if tp.network != "tcp" || tp.addr != "1.2.3.4:5678" { |
| t.Errorf("Dialer connected to wrong host. Wanted 1.2.3.4:5678, got: %v", tp) |
| } |
| } else { |
| t.Errorf("Unexpected error from Dial: %s", err) |
| } |
| } |