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 (
        "errors"
        "net"
        "reflect"
        "testing"
)

type recordingProxy struct {
        addrs []string
}

func (r *recordingProxy) Dial(network, addr string) (net.Conn, error) {
        r.addrs = append(r.addrs, addr)
        return nil, errors.New("recordingProxy")
}

func TestPerHost(t *testing.T) {
        var def, bypass recordingProxy
        perHost := NewPerHost(&def, &bypass)
        perHost.AddFromString("localhost,*.zone,127.0.0.1,10.0.0.1/8,1000::/16")

        expectedDef := []string{
                "example.com:123",
                "1.2.3.4:123",
                "[1001::]:123",
        }
        expectedBypass := []string{
                "localhost:123",
                "zone:123",
                "foo.zone:123",
                "127.0.0.1:123",
                "10.1.2.3:123",
                "[1000::]:123",
        }

        for _, addr := range expectedDef {
                perHost.Dial("tcp", addr)
        }
        for _, addr := range expectedBypass {
                perHost.Dial("tcp", addr)
        }

        if !reflect.DeepEqual(expectedDef, def.addrs) {
                t.Errorf("Hosts which went to the default proxy didn't match. Got %v, want %v", def.addrs, expectedDef)
        }
        if !reflect.DeepEqual(expectedBypass, bypass.addrs) {
                t.Errorf("Hosts which went to the bypass proxy didn't match. Got %v, want %v", bypass.addrs, expectedBypass)
        }
}