diff options
author | Russ Cox <rsc@golang.org> | 2009-03-23 18:50:35 -0700 |
---|---|---|
committer | Russ Cox <rsc@golang.org> | 2009-03-23 18:50:35 -0700 |
commit | aa109c5dc9919d5a7d9399eda1f3a9c09f38d4f9 (patch) | |
tree | 419e2f2fb716f5389fefb545bfb8be86a8c23522 /test | |
parent | 3d6fc36664d36dc29e6c2930c8c4fe96b2d7a911 (diff) | |
download | go-aa109c5dc9919d5a7d9399eda1f3a9c09f38d4f9.tar.gz |
add test for close/closed, fix a few implementation bugs.
R=ken
OCL=26664
CL=26664
Diffstat (limited to 'test')
-rw-r--r-- | test/closedchan.go | 197 |
1 files changed, 197 insertions, 0 deletions
diff --git a/test/closedchan.go b/test/closedchan.go new file mode 100644 index 000000000..4ab12c775 --- /dev/null +++ b/test/closedchan.go @@ -0,0 +1,197 @@ +// $G $D/$F.go && $L $F.$A && ./$A.out + +// 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. + +// Test close(c), closed(c). +// +// TODO(rsc): Doesn't check behavior of close(c) when there +// are blocked senders/receivers. + +package main + +type Chan interface { + Send(int); + Nbsend(int) bool; + Recv() int; + Nbrecv() (int, bool); + Close(); + Closed() bool; + Impl() string; +} + +// direct channel operations +type XChan chan int +func (c XChan) Send(x int) { + c <- x +} + +func (c XChan) Nbsend(x int) bool { + return c <- x; +} + +func (c XChan) Recv() int { + return <-c +} + +func (c XChan) Nbrecv() (int, bool) { + x, ok := <-c; + return x, ok; +} + +func (c XChan) Close() { + close(c) +} + +func (c XChan) Closed() bool { + return closed(c) +} + +func (c XChan) Impl() string { + return "(<- operator)" +} + +// indirect operations via select +type SChan chan int +func (c SChan) Send(x int) { + select { + case c <- x: + } +} + +func (c SChan) Nbsend(x int) bool { + select { + case c <- x: + return true; + default: + return false; + } + panic("nbsend"); +} + +func (c SChan) Recv() int { + select { + case x := <-c: + return x; + } + panic("recv"); +} + +func (c SChan) Nbrecv() (int, bool) { + select { + case x := <-c: + return x, true; + default: + return 0, false; + } + panic("nbrecv"); +} + +func (c SChan) Close() { + close(c) +} + +func (c SChan) Closed() bool { + return closed(c) +} + +func (c SChan) Impl() string { + return "(select)"; +} + +func test1(c Chan) { + // not closed until the close signal (a zero value) has been received. + if c.Closed() { + println("test1: Closed before Recv zero:", c.Impl()); + } + + for i := 0; i < 3; i++ { + // recv a close signal (a zero value) + if x := c.Recv(); x != 0 { + println("test1: recv on closed got non-zero:", x, c.Impl()); + } + + // should now be closed. + if !c.Closed() { + println("test1: not closed after recv zero", c.Impl()); + } + + // should work with ,ok: received a value without blocking, so ok == true. + x, ok := c.Nbrecv(); + if !ok { + println("test1: recv on closed got not ok", c.Impl()); + } + if x != 0 { + println("test1: recv ,ok on closed got non-zero:", x, c.Impl()); + } + } + + // send should work with ,ok too: sent a value without blocking, so ok == true. + ok := c.Nbsend(1); + if !ok { + println("test1: send on closed got not ok", c.Impl()); + } + + // but the value should have been discarded. + if x := c.Recv(); x != 0 { + println("test1: recv on closed got non-zero after send on closed:", x, c.Impl()); + } + + // similarly Send. + c.Send(2); + if x := c.Recv(); x != 0 { + println("test1: recv on closed got non-zero after send on closed:", x, c.Impl()); + } +} + +func testasync1(c Chan) { + // not closed until the close signal (a zero value) has been received. + if c.Closed() { + println("testasync1: Closed before Recv zero:", c.Impl()); + } + + // should be able to get the last value via Recv + if x := c.Recv(); x != 1 { + println("testasync1: Recv did not get 1:", x, c.Impl()); + } + + test1(c); +} + +func testasync2(c Chan) { + // not closed until the close signal (a zero value) has been received. + if c.Closed() { + println("testasync2: Closed before Recv zero:", c.Impl()); + } + + // should be able to get the last value via Nbrecv + if x, ok := c.Nbrecv(); !ok || x != 1 { + println("testasync2: Nbrecv did not get 1, true:", x, ok, c.Impl()); + } + + test1(c); +} + +func closedsync() chan int { + c := make(chan int); + close(c); + return c; +} + +func closedasync() chan int { + c := make(chan int, 2); + c <- 1; + close(c); + return c; +} + +func main() { + test1(XChan(closedsync())); + test1(SChan(closedsync())); + + testasync1(XChan(closedasync())); + testasync1(SChan(closedasync())); + testasync2(XChan(closedasync())); + testasync2(SChan(closedasync())); +} |