...
Run Format

Source file src/net/ipsock_test.go

Documentation: net

  // 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, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
  
  	{ipv4only, nil, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
  	{ipv4only, []IPAddr{{IP: IPv6loopback}}, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
  
  	{ipv6only, nil, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
  	{ipv6only, []IPAddr{{IP: IPv4(127, 0, 0, 1)}}, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
  }
  
  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, "ADDR")
  		if !reflect.DeepEqual(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)
  			}
  		}
  	}
  }
  

View as plain text