rsc.io/go@v0.0.0-20150416155037-e040fd465409/src/net/ipraw_test.go (about)

     1  // Copyright 2009 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  // The full stack test cases for IPConn have been moved to the
    13  // following:
    14  //	golang.org/x/net/ipv4
    15  //	golang.org/x/net/ipv6
    16  //	golang.org/x/net/icmp
    17  
    18  type resolveIPAddrTest struct {
    19  	network       string
    20  	litAddrOrName string
    21  	addr          *IPAddr
    22  	err           error
    23  }
    24  
    25  var resolveIPAddrTests = []resolveIPAddrTest{
    26  	{"ip", "127.0.0.1", &IPAddr{IP: IPv4(127, 0, 0, 1)}, nil},
    27  	{"ip4", "127.0.0.1", &IPAddr{IP: IPv4(127, 0, 0, 1)}, nil},
    28  	{"ip4:icmp", "127.0.0.1", &IPAddr{IP: IPv4(127, 0, 0, 1)}, nil},
    29  
    30  	{"ip", "::1", &IPAddr{IP: ParseIP("::1")}, nil},
    31  	{"ip6", "::1", &IPAddr{IP: ParseIP("::1")}, nil},
    32  	{"ip6:ipv6-icmp", "::1", &IPAddr{IP: ParseIP("::1")}, nil},
    33  	{"ip6:IPv6-ICMP", "::1", &IPAddr{IP: ParseIP("::1")}, nil},
    34  
    35  	{"ip", "::1%en0", &IPAddr{IP: ParseIP("::1"), Zone: "en0"}, nil},
    36  	{"ip6", "::1%911", &IPAddr{IP: ParseIP("::1"), Zone: "911"}, nil},
    37  
    38  	{"", "127.0.0.1", &IPAddr{IP: IPv4(127, 0, 0, 1)}, nil}, // Go 1.0 behavior
    39  	{"", "::1", &IPAddr{IP: ParseIP("::1")}, nil},           // Go 1.0 behavior
    40  
    41  	{"l2tp", "127.0.0.1", nil, UnknownNetworkError("l2tp")},
    42  	{"l2tp:gre", "127.0.0.1", nil, UnknownNetworkError("l2tp:gre")},
    43  	{"tcp", "1.2.3.4:123", nil, UnknownNetworkError("tcp")},
    44  }
    45  
    46  func TestResolveIPAddr(t *testing.T) {
    47  	if !testableNetwork("ip+nopriv") {
    48  		t.Skip("ip+nopriv test")
    49  	}
    50  
    51  	origTestHookLookupIP := testHookLookupIP
    52  	defer func() { testHookLookupIP = origTestHookLookupIP }()
    53  	testHookLookupIP = lookupLocalhost
    54  
    55  	for i, tt := range resolveIPAddrTests {
    56  		addr, err := ResolveIPAddr(tt.network, tt.litAddrOrName)
    57  		if err != tt.err {
    58  			t.Errorf("#%d: %v", i, err)
    59  		} else if !reflect.DeepEqual(addr, tt.addr) {
    60  			t.Errorf("#%d: got %#v; want %#v", i, addr, tt.addr)
    61  		}
    62  		if err != nil {
    63  			continue
    64  		}
    65  		rtaddr, err := ResolveIPAddr(addr.Network(), addr.String())
    66  		if err != nil {
    67  			t.Errorf("#%d: %v", i, err)
    68  		} else if !reflect.DeepEqual(rtaddr, addr) {
    69  			t.Errorf("#%d: got %#v; want %#v", i, rtaddr, addr)
    70  		}
    71  	}
    72  }
    73  
    74  var ipConnLocalNameTests = []struct {
    75  	net   string
    76  	laddr *IPAddr
    77  }{
    78  	{"ip4:icmp", &IPAddr{IP: IPv4(127, 0, 0, 1)}},
    79  	{"ip4:icmp", &IPAddr{}},
    80  	{"ip4:icmp", nil},
    81  }
    82  
    83  func TestIPConnLocalName(t *testing.T) {
    84  	for _, tt := range ipConnLocalNameTests {
    85  		if !testableNetwork(tt.net) {
    86  			t.Logf("skipping %s test", tt.net)
    87  			continue
    88  		}
    89  		c, err := ListenIP(tt.net, tt.laddr)
    90  		if err != nil {
    91  			t.Fatalf("ListenIP failed: %v", err)
    92  		}
    93  		defer c.Close()
    94  		if la := c.LocalAddr(); la == nil {
    95  			t.Fatal("IPConn.LocalAddr failed")
    96  		}
    97  	}
    98  }
    99  
   100  func TestIPConnRemoteName(t *testing.T) {
   101  	if !testableNetwork("ip:tcp") {
   102  		t.Skip("ip:tcp test")
   103  	}
   104  
   105  	raddr := &IPAddr{IP: IPv4(127, 0, 0, 1).To4()}
   106  	c, err := DialIP("ip:tcp", &IPAddr{IP: IPv4(127, 0, 0, 1)}, raddr)
   107  	if err != nil {
   108  		t.Fatalf("DialIP failed: %v", err)
   109  	}
   110  	defer c.Close()
   111  	if !reflect.DeepEqual(raddr, c.RemoteAddr()) {
   112  		t.Fatalf("got %#v, expected %#v", c.RemoteAddr(), raddr)
   113  	}
   114  }