ipsock_test.go

Documentation: net

		 1  // Copyright 2013 The Go Authors. All rights reserved.
		 2  // Use of this source code is governed by a BSD-style
		 3  // license that can be found in the LICENSE file.
		 4  
		 5  package net
		 6  
		 7  import (
		 8  	"reflect"
		 9  	"testing"
		10  )
		11  
		12  var testInetaddr = func(ip IPAddr) Addr { return &TCPAddr{IP: ip.IP, Port: 5682, Zone: ip.Zone} }
		13  
		14  var addrListTests = []struct {
		15  	filter		func(IPAddr) bool
		16  	ips			 []IPAddr
		17  	inetaddr	func(IPAddr) Addr
		18  	first		 Addr
		19  	primaries addrList
		20  	fallbacks addrList
		21  	err			 error
		22  }{
		23  	{
		24  		nil,
		25  		[]IPAddr{
		26  			{IP: IPv4(127, 0, 0, 1)},
		27  			{IP: IPv6loopback},
		28  		},
		29  		testInetaddr,
		30  		&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
		31  		addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
		32  		addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
		33  		nil,
		34  	},
		35  	{
		36  		nil,
		37  		[]IPAddr{
		38  			{IP: IPv6loopback},
		39  			{IP: IPv4(127, 0, 0, 1)},
		40  		},
		41  		testInetaddr,
		42  		&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
		43  		addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
		44  		addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
		45  		nil,
		46  	},
		47  	{
		48  		nil,
		49  		[]IPAddr{
		50  			{IP: IPv4(127, 0, 0, 1)},
		51  			{IP: IPv4(192, 168, 0, 1)},
		52  		},
		53  		testInetaddr,
		54  		&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
		55  		addrList{
		56  			&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
		57  			&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
		58  		},
		59  		nil,
		60  		nil,
		61  	},
		62  	{
		63  		nil,
		64  		[]IPAddr{
		65  			{IP: IPv6loopback},
		66  			{IP: ParseIP("fe80::1"), Zone: "eth0"},
		67  		},
		68  		testInetaddr,
		69  		&TCPAddr{IP: IPv6loopback, Port: 5682},
		70  		addrList{
		71  			&TCPAddr{IP: IPv6loopback, Port: 5682},
		72  			&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
		73  		},
		74  		nil,
		75  		nil,
		76  	},
		77  	{
		78  		nil,
		79  		[]IPAddr{
		80  			{IP: IPv4(127, 0, 0, 1)},
		81  			{IP: IPv4(192, 168, 0, 1)},
		82  			{IP: IPv6loopback},
		83  			{IP: ParseIP("fe80::1"), Zone: "eth0"},
		84  		},
		85  		testInetaddr,
		86  		&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
		87  		addrList{
		88  			&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
		89  			&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
		90  		},
		91  		addrList{
		92  			&TCPAddr{IP: IPv6loopback, Port: 5682},
		93  			&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
		94  		},
		95  		nil,
		96  	},
		97  	{
		98  		nil,
		99  		[]IPAddr{
	 100  			{IP: IPv6loopback},
	 101  			{IP: ParseIP("fe80::1"), Zone: "eth0"},
	 102  			{IP: IPv4(127, 0, 0, 1)},
	 103  			{IP: IPv4(192, 168, 0, 1)},
	 104  		},
	 105  		testInetaddr,
	 106  		&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
	 107  		addrList{
	 108  			&TCPAddr{IP: IPv6loopback, Port: 5682},
	 109  			&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
	 110  		},
	 111  		addrList{
	 112  			&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
	 113  			&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
	 114  		},
	 115  		nil,
	 116  	},
	 117  	{
	 118  		nil,
	 119  		[]IPAddr{
	 120  			{IP: IPv4(127, 0, 0, 1)},
	 121  			{IP: IPv6loopback},
	 122  			{IP: IPv4(192, 168, 0, 1)},
	 123  			{IP: ParseIP("fe80::1"), Zone: "eth0"},
	 124  		},
	 125  		testInetaddr,
	 126  		&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
	 127  		addrList{
	 128  			&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
	 129  			&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
	 130  		},
	 131  		addrList{
	 132  			&TCPAddr{IP: IPv6loopback, Port: 5682},
	 133  			&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
	 134  		},
	 135  		nil,
	 136  	},
	 137  	{
	 138  		nil,
	 139  		[]IPAddr{
	 140  			{IP: IPv6loopback},
	 141  			{IP: IPv4(127, 0, 0, 1)},
	 142  			{IP: ParseIP("fe80::1"), Zone: "eth0"},
	 143  			{IP: IPv4(192, 168, 0, 1)},
	 144  		},
	 145  		testInetaddr,
	 146  		&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
	 147  		addrList{
	 148  			&TCPAddr{IP: IPv6loopback, Port: 5682},
	 149  			&TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
	 150  		},
	 151  		addrList{
	 152  			&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
	 153  			&TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
	 154  		},
	 155  		nil,
	 156  	},
	 157  
	 158  	{
	 159  		ipv4only,
	 160  		[]IPAddr{
	 161  			{IP: IPv4(127, 0, 0, 1)},
	 162  			{IP: IPv6loopback},
	 163  		},
	 164  		testInetaddr,
	 165  		&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
	 166  		addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
	 167  		nil,
	 168  		nil,
	 169  	},
	 170  	{
	 171  		ipv4only,
	 172  		[]IPAddr{
	 173  			{IP: IPv6loopback},
	 174  			{IP: IPv4(127, 0, 0, 1)},
	 175  		},
	 176  		testInetaddr,
	 177  		&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
	 178  		addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
	 179  		nil,
	 180  		nil,
	 181  	},
	 182  
	 183  	{
	 184  		ipv6only,
	 185  		[]IPAddr{
	 186  			{IP: IPv4(127, 0, 0, 1)},
	 187  			{IP: IPv6loopback},
	 188  		},
	 189  		testInetaddr,
	 190  		&TCPAddr{IP: IPv6loopback, Port: 5682},
	 191  		addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
	 192  		nil,
	 193  		nil,
	 194  	},
	 195  	{
	 196  		ipv6only,
	 197  		[]IPAddr{
	 198  			{IP: IPv6loopback},
	 199  			{IP: IPv4(127, 0, 0, 1)},
	 200  		},
	 201  		testInetaddr,
	 202  		&TCPAddr{IP: IPv6loopback, Port: 5682},
	 203  		addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
	 204  		nil,
	 205  		nil,
	 206  	},
	 207  
	 208  	{nil, nil, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
	 209  
	 210  	{ipv4only, nil, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
	 211  	{ipv4only, []IPAddr{{IP: IPv6loopback}}, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
	 212  
	 213  	{ipv6only, nil, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
	 214  	{ipv6only, []IPAddr{{IP: IPv4(127, 0, 0, 1)}}, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
	 215  }
	 216  
	 217  func TestAddrList(t *testing.T) {
	 218  	if !supportsIPv4() || !supportsIPv6() {
	 219  		t.Skip("both IPv4 and IPv6 are required")
	 220  	}
	 221  
	 222  	for i, tt := range addrListTests {
	 223  		addrs, err := filterAddrList(tt.filter, tt.ips, tt.inetaddr, "ADDR")
	 224  		if !reflect.DeepEqual(err, tt.err) {
	 225  			t.Errorf("#%v: got %v; want %v", i, err, tt.err)
	 226  		}
	 227  		if tt.err != nil {
	 228  			if len(addrs) != 0 {
	 229  				t.Errorf("#%v: got %v; want 0", i, len(addrs))
	 230  			}
	 231  			continue
	 232  		}
	 233  		first := addrs.first(isIPv4)
	 234  		if !reflect.DeepEqual(first, tt.first) {
	 235  			t.Errorf("#%v: got %v; want %v", i, first, tt.first)
	 236  		}
	 237  		primaries, fallbacks := addrs.partition(isIPv4)
	 238  		if !reflect.DeepEqual(primaries, tt.primaries) {
	 239  			t.Errorf("#%v: got %v; want %v", i, primaries, tt.primaries)
	 240  		}
	 241  		if !reflect.DeepEqual(fallbacks, tt.fallbacks) {
	 242  			t.Errorf("#%v: got %v; want %v", i, fallbacks, tt.fallbacks)
	 243  		}
	 244  		expectedLen := len(primaries) + len(fallbacks)
	 245  		if len(addrs) != expectedLen {
	 246  			t.Errorf("#%v: got %v; want %v", i, len(addrs), expectedLen)
	 247  		}
	 248  	}
	 249  }
	 250  
	 251  func TestAddrListPartition(t *testing.T) {
	 252  	addrs := addrList{
	 253  		&IPAddr{IP: ParseIP("fe80::"), Zone: "eth0"},
	 254  		&IPAddr{IP: ParseIP("fe80::1"), Zone: "eth0"},
	 255  		&IPAddr{IP: ParseIP("fe80::2"), Zone: "eth0"},
	 256  	}
	 257  	cases := []struct {
	 258  		lastByte	byte
	 259  		primaries addrList
	 260  		fallbacks addrList
	 261  	}{
	 262  		{0, addrList{addrs[0]}, addrList{addrs[1], addrs[2]}},
	 263  		{1, addrList{addrs[0], addrs[2]}, addrList{addrs[1]}},
	 264  		{2, addrList{addrs[0], addrs[1]}, addrList{addrs[2]}},
	 265  		{3, addrList{addrs[0], addrs[1], addrs[2]}, nil},
	 266  	}
	 267  	for i, tt := range cases {
	 268  		// Inverting the function's output should not affect the outcome.
	 269  		for _, invert := range []bool{false, true} {
	 270  			primaries, fallbacks := addrs.partition(func(a Addr) bool {
	 271  				ip := a.(*IPAddr).IP
	 272  				return (ip[len(ip)-1] == tt.lastByte) != invert
	 273  			})
	 274  			if !reflect.DeepEqual(primaries, tt.primaries) {
	 275  				t.Errorf("#%v: got %v; want %v", i, primaries, tt.primaries)
	 276  			}
	 277  			if !reflect.DeepEqual(fallbacks, tt.fallbacks) {
	 278  				t.Errorf("#%v: got %v; want %v", i, fallbacks, tt.fallbacks)
	 279  			}
	 280  		}
	 281  	}
	 282  }
	 283
View as plain text