github.com/peggyl/go@v0.0.0-20151008231540-ae315999c2d5/src/net/rpc/client_test.go (about)

     1  // Copyright 2014 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 rpc
     6  
     7  import (
     8  	"errors"
     9  	"fmt"
    10  	"net"
    11  	"runtime"
    12  	"strings"
    13  	"testing"
    14  )
    15  
    16  type shutdownCodec struct {
    17  	responded chan int
    18  	closed    bool
    19  }
    20  
    21  func (c *shutdownCodec) WriteRequest(*Request, interface{}) error { return nil }
    22  func (c *shutdownCodec) ReadResponseBody(interface{}) error       { return nil }
    23  func (c *shutdownCodec) ReadResponseHeader(*Response) error {
    24  	c.responded <- 1
    25  	return errors.New("shutdownCodec ReadResponseHeader")
    26  }
    27  func (c *shutdownCodec) Close() error {
    28  	c.closed = true
    29  	return nil
    30  }
    31  
    32  func TestCloseCodec(t *testing.T) {
    33  	codec := &shutdownCodec{responded: make(chan int)}
    34  	client := NewClientWithCodec(codec)
    35  	<-codec.responded
    36  	client.Close()
    37  	if !codec.closed {
    38  		t.Error("client.Close did not close codec")
    39  	}
    40  }
    41  
    42  // Test that errors in gob shut down the connection. Issue 7689.
    43  
    44  type R struct {
    45  	msg []byte // Not exported, so R does not work with gob.
    46  }
    47  
    48  type S struct{}
    49  
    50  func (s *S) Recv(nul *struct{}, reply *R) error {
    51  	*reply = R{[]byte("foo")}
    52  	return nil
    53  }
    54  
    55  func TestGobError(t *testing.T) {
    56  	if runtime.GOOS == "plan9" {
    57  		t.Skip("skipping test; see https://golang.org/issue/8908")
    58  	}
    59  	defer func() {
    60  		err := recover()
    61  		if err == nil {
    62  			t.Fatal("no error")
    63  		}
    64  		if !strings.Contains("reading body EOF", err.(error).Error()) {
    65  			t.Fatal("expected `reading body EOF', got", err)
    66  		}
    67  	}()
    68  	Register(new(S))
    69  
    70  	listen, err := net.Listen("tcp", "127.0.0.1:0")
    71  	if err != nil {
    72  		panic(err)
    73  	}
    74  	go Accept(listen)
    75  
    76  	client, err := Dial("tcp", listen.Addr().String())
    77  	if err != nil {
    78  		panic(err)
    79  	}
    80  
    81  	var reply Reply
    82  	err = client.Call("S.Recv", &struct{}{}, &reply)
    83  	if err != nil {
    84  		panic(err)
    85  	}
    86  
    87  	fmt.Printf("%#v\n", reply)
    88  	client.Close()
    89  
    90  	listen.Close()
    91  }