libgo/go/net/rpc/client_test.go

   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         "testing"
  10 )
  11
  12 type shutdownCodec struct {
  13         responded chan int
  14         closed    bool
  15 }
  16
  17 func (c *shutdownCodec) WriteRequest(*Request, interface{}) error { return nil }
  18 func (c *shutdownCodec) ReadResponseBody(interface{}) error       { return nil }
  19 func (c *shutdownCodec) ReadResponseHeader(*Response) error {
  20         c.responded <- 1
  21         return errors.New("shutdownCodec ReadResponseHeader")
  22 }
  23 func (c *shutdownCodec) Close() error {
  24         c.closed = true
  25         return nil
  26 }
  27
  28 func TestCloseCodec(t *testing.T) {
  29         codec := &shutdownCodec{responded: make(chan int)}
  30         client := NewClientWithCodec(codec)
  31         <-codec.responded
  32         client.Close()
  33         if !codec.closed {
  34                 t.Error("client.Close did not close codec")
  35         }
  36 }