code review 6816116: net: support IPv6 scoped addressing zone

src/pkg/net/tcp_test.go

 // Copyright 2012 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"
         "runtime"
         "testing"
         "time"
 )
 
 func BenchmarkTCPOneShot(b *testing.B) {
         benchmarkTCP(b, false, false)
 }
 
 func BenchmarkTCPOneShotTimeout(b *testing.B) {
@@ skipping 91 matching lines @@
                                 if !sendMsg(c, buf[:]) || !recvMsg(c, buf[:]) {
                                         break
                                 }
                         }
                 }()
         }
         for i := 0; i < cap(sem); i++ {
                 sem <- true
         }
 }
+
+var resolveTCPAddrTests = []struct {
+        net     string
+        litAddr string
+        addr    *TCPAddr
+}{
+        {"tcp", "127.0.0.1:1", &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 1}},
+        {"tcp4", "127.0.0.1:65535", &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 65535}},
+
+        {"tcp", "[::1]:0", &TCPAddr{IP: ParseIP("::1"), Port: 0}},
+        {"tcp", "[::1%1]:1", &TCPAddr{IP: ParseIP("::1"), Port: 1, zoneId: 1}},
+        {"tcp6", "[::1]:2", &TCPAddr{IP: ParseIP("::1"), Port: 2}},
+        {"tcp6", "[::1%2]:3", &TCPAddr{IP: ParseIP("::1"), Port: 3, zoneId: 2}},
+        {"tcp6", "[::1]:4", &TCPAddr{IP: ParseIP("::1"), Port: 4, zoneId: -1}},
+}
+
+func TestResolveTCPAddr(t *testing.T) {
+        for _, tt := range resolveTCPAddrTests {
+                if tt.addr.zoneId == -1 {
+                        ifi := loopbackInterface()
+                        if ifi == nil {
+                                continue
+                        }
+                        i := last(tt.litAddr, ']')
+                        if i > 0 {
+                                tt.litAddr = tt.litAddr[:i] + "%" + ifi.Name + tt.litAddr[i:]
+                                tt.addr.zoneId = ifi.Index
+                        }
+                }
+                addr, err := ResolveTCPAddr(tt.net, tt.litAddr)
+                if err != nil {
+                        t.Fatalf("ResolveTCPAddr(%v, %v) failed: %v", tt.net, tt.litAddr, err)
+                }
+                if !reflect.DeepEqual(addr, tt.addr) {
+                        t.Fatalf("got %#v; expected %#v", addr, tt.addr)
+                }
+        }
+}