github.com/corona10/go@v0.0.0-20180224231303-7a218942be57/src/net/ipsock_test.go (about)

     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  }