File: proxy_test.go

package info (click to toggle)
gcc-avr 1%3A5.4.0%2BAtmel3.6.2-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 589,872 kB
  • sloc: ansic: 2,775,581; ada: 756,757; cpp: 723,977; f90: 117,673; asm: 66,898; makefile: 62,755; xml: 44,466; sh: 29,549; exp: 23,315; objc: 15,216; fortran: 10,901; pascal: 4,185; python: 4,093; perl: 2,969; awk: 2,811; ml: 2,385; cs: 879; yacc: 316; lex: 198; haskell: 112; lisp: 8
file content (50 lines) | stat: -rw-r--r-- 1,111 bytes parent folder | download | duplicates (5) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
// 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)
	}
}