code review 5136052: net: add shutdown: TCPConn.CloseWrite and CloseRead

src/pkg/net/net_test.go

 // Copyright 2009 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 (
         "flag"
+        "os"
         "regexp"
         "testing"
 )
 
 var runErrorTest = flag.Bool("run_error_test", false, "let TestDialError check for dns errors")
 
 type DialErrorTest struct {
         Net     string
         Raddr   string
         Pattern string
@@ skipping 93 matching lines @@
                         t.Errorf("#%d: expected <nil>, got %q (error)", i, e)
                 }
                 if e != nil && e.(*DNSError).Error != tt.ErrPrefix {
                         t.Errorf("#%d: expected %q, got %q (mismatched error)", i, tt.ErrPrefix, e.(*DNSError).Error)
                 }
                 if a != tt.Reverse {
                         t.Errorf("#%d: expected %q, got %q (reverse address)", i, tt.Reverse, a)
                 }
         }
 }
+
+func TestShutdown(t *testing.T) {
+        l, err := Listen("tcp", "127.0.0.1:0")
+        if err != nil {
+                if l, err = Listen("tcp6", "[::1]:0"); err != nil {
+                        t.Fatalf("ListenTCP on :0: %v", err)
+                }
+        }
+
+        go func() {
+                c, err := l.Accept()
+                if err != nil {
+                        t.Fatalf("Accept: %v", err)
+                }
+                var buf [10]byte
+                n, err := c.Read(buf[:])
+                if n != 0 || err != os.EOF {
+                        t.Fatalf("server Read = %d, %v; want 0, os.EOF", n, err)
+                }
+                c.Write([]byte("response"))
+                c.Close()
+        }()
+
+        c, err := Dial("tcp", l.Addr().String())
+        if err != nil {
+                t.Fatalf("Dial: %v", err)
+        }
+        defer c.Close()
+
+        err = c.(*TCPConn).CloseWrite()
+        if err != nil {
+                t.Fatalf("CloseWrite: %v", err)
+        }
+        var buf [10]byte
+        n, err := c.Read(buf[:])
+        if err != nil {
+                t.Fatalf("client Read: %d, %v", n, err)
+        }
+        got := string(buf[:n])
+        if got != "response" {
+                t.Errorf("read = %q, want \"response\"", got)
+        }
+}