Subversion Repositories openrisc

// 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)
        }
}