Golang程序
|
283行
|
6.64 KB
// Copyright 2013 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 net
import (
"reflect"
"testing"
)
var testInetaddr = func(ip IPAddr) Addr { return &TCPAddr{IP: ip.IP, Port: 5682, Zone: ip.Zone} }
var addrListTests = []struct {
filter func(IPAddr) bool
ips []IPAddr
inetaddr func(IPAddr) Addr
first Addr
primaries addrList
fallbacks addrList
err error
}{
{
nil,
[]IPAddr{
{IP: IPv4(127, 0, 0, 1)},
{IP: IPv6loopback},
},
testInetaddr,
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
nil,
},
{
nil,
[]IPAddr{
{IP: IPv6loopback},
{IP: IPv4(127, 0, 0, 1)},
},
testInetaddr,
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
nil,
},
{
nil,
[]IPAddr{
{IP: IPv4(127, 0, 0, 1)},
{IP: IPv4(192, 168, 0, 1)},
},
testInetaddr,
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
addrList{
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
},
nil,
nil,
},
{
nil,
[]IPAddr{
{IP: IPv6loopback},
{IP: ParseIP("fe80::1"), Zone: "eth0"},
},
testInetaddr,
&TCPAddr{IP: IPv6loopback, Port: 5682},
addrList{
&TCPAddr{IP: IPv6loopback, Port: 5682},
&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
},
nil,
nil,
},
{
nil,
[]IPAddr{
{IP: IPv4(127, 0, 0, 1)},
{IP: IPv4(192, 168, 0, 1)},
{IP: IPv6loopback},
{IP: ParseIP("fe80::1"), Zone: "eth0"},
},
testInetaddr,
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
addrList{
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
},
addrList{
&TCPAddr{IP: IPv6loopback, Port: 5682},
&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
},
nil,
},
{
nil,
[]IPAddr{
{IP: IPv6loopback},
{IP: ParseIP("fe80::1"), Zone: "eth0"},
{IP: IPv4(127, 0, 0, 1)},
{IP: IPv4(192, 168, 0, 1)},
},
testInetaddr,
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
addrList{
&TCPAddr{IP: IPv6loopback, Port: 5682},
&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
},
addrList{
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
},
nil,
},
{
nil,
[]IPAddr{
{IP: IPv4(127, 0, 0, 1)},
{IP: IPv6loopback},
{IP: IPv4(192, 168, 0, 1)},
{IP: ParseIP("fe80::1"), Zone: "eth0"},
},
testInetaddr,
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
addrList{
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
},
addrList{
&TCPAddr{IP: IPv6loopback, Port: 5682},
&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
},
nil,
},
{
nil,
[]IPAddr{
{IP: IPv6loopback},
{IP: IPv4(127, 0, 0, 1)},
{IP: ParseIP("fe80::1"), Zone: "eth0"},
{IP: IPv4(192, 168, 0, 1)},
},
testInetaddr,
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
addrList{
&TCPAddr{IP: IPv6loopback, Port: 5682},
&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
},
addrList{
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
},
nil,
},
{
ipv4only,
[]IPAddr{
{IP: IPv4(127, 0, 0, 1)},
{IP: IPv6loopback},
},
testInetaddr,
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
nil,
nil,
},
{
ipv4only,
[]IPAddr{
{IP: IPv6loopback},
{IP: IPv4(127, 0, 0, 1)},
},
testInetaddr,
&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
nil,
nil,
},
{
ipv6only,
[]IPAddr{
{IP: IPv4(127, 0, 0, 1)},
{IP: IPv6loopback},
},
testInetaddr,
&TCPAddr{IP: IPv6loopback, Port: 5682},
addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
nil,
nil,
},
{
ipv6only,
[]IPAddr{
{IP: IPv6loopback},
{IP: IPv4(127, 0, 0, 1)},
},
testInetaddr,
&TCPAddr{IP: IPv6loopback, Port: 5682},
addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
nil,
nil,
},
{nil, nil, testInetaddr, nil, nil, nil, errNoSuitableAddress},
{ipv4only, nil, testInetaddr, nil, nil, nil, errNoSuitableAddress},
{ipv4only, []IPAddr{{IP: IPv6loopback}}, testInetaddr, nil, nil, nil, errNoSuitableAddress},
{ipv6only, nil, testInetaddr, nil, nil, nil, errNoSuitableAddress},
{ipv6only, []IPAddr{{IP: IPv4(127, 0, 0, 1)}}, testInetaddr, nil, nil, nil, errNoSuitableAddress},
}
func TestAddrList(t *testing.T) {
if !supportsIPv4 || !supportsIPv6 {
t.Skip("both IPv4 and IPv6 are required")
}
for i, tt := range addrListTests {
addrs, err := filterAddrList(tt.filter, tt.ips, tt.inetaddr)
if err != tt.err {
t.Errorf("#%v: got %v; want %v", i, err, tt.err)
}
if tt.err != nil {
if len(addrs) != 0 {
t.Errorf("#%v: got %v; want 0", i, len(addrs))
}
continue
}
first := addrs.first(isIPv4)
if !reflect.DeepEqual(first, tt.first) {
t.Errorf("#%v: got %v; want %v", i, first, tt.first)
}
primaries, fallbacks := addrs.partition(isIPv4)
if !reflect.DeepEqual(primaries, tt.primaries) {
t.Errorf("#%v: got %v; want %v", i, primaries, tt.primaries)
}
if !reflect.DeepEqual(fallbacks, tt.fallbacks) {
t.Errorf("#%v: got %v; want %v", i, fallbacks, tt.fallbacks)
}
expectedLen := len(primaries) + len(fallbacks)
if len(addrs) != expectedLen {
t.Errorf("#%v: got %v; want %v", i, len(addrs), expectedLen)
}
}
}
func TestAddrListPartition(t *testing.T) {
addrs := addrList{
&IPAddr{IP: ParseIP("fe80::"), Zone: "eth0"},
&IPAddr{IP: ParseIP("fe80::1"), Zone: "eth0"},
&IPAddr{IP: ParseIP("fe80::2"), Zone: "eth0"},
}
cases := []struct {
lastByte byte
primaries addrList
fallbacks addrList
}{
{0, addrList{addrs[0]}, addrList{addrs[1], addrs[2]}},
{1, addrList{addrs[0], addrs[2]}, addrList{addrs[1]}},
{2, addrList{addrs[0], addrs[1]}, addrList{addrs[2]}},
{3, addrList{addrs[0], addrs[1], addrs[2]}, nil},
}
for i, tt := range cases {
// Inverting the function's output should not affect the outcome.
for _, invert := range []bool{false, true} {
primaries, fallbacks := addrs.partition(func(a Addr) bool {
ip := a.(*IPAddr).IP
return (ip[len(ip)-1] == tt.lastByte) != invert
})
if !reflect.DeepEqual(primaries, tt.primaries) {
t.Errorf("#%v: got %v; want %v", i, primaries, tt.primaries)
}
if !reflect.DeepEqual(fallbacks, tt.fallbacks) {
t.Errorf("#%v: got %v; want %v", i, fallbacks, tt.fallbacks)
}
}
}
}