1 // Copyright 2013 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.
13 // An in-memory packetConn. It is safe to call Close and writePacket
14 // from different goroutines.
15 type memTransport struct {
23 func (t *memTransport) readPacket() ([]byte, error) {
27 if len(t.pending) > 0 {
29 t.pending = t.pending[1:]
39 func (t *memTransport) closeSelf() error {
50 func (t *memTransport) Close() error {
51 err := t.write.closeSelf()
56 func (t *memTransport) writePacket(p []byte) error {
58 defer t.write.Unlock()
62 c := make([]byte, len(p))
64 t.write.pending = append(t.write.pending, c)
69 func memPipe() (a, b packetConn) {
74 t1.Cond = sync.NewCond(&t1.Mutex)
75 t2.Cond = sync.NewCond(&t2.Mutex)
79 func TestMemPipe(t *testing.T) {
81 if err := a.writePacket([]byte{42}); err != nil {
82 t.Fatalf("writePacket: %v", err)
84 if err := a.Close(); err != nil {
85 t.Fatal("Close: ", err)
87 p, err := b.readPacket()
89 t.Fatal("readPacket: ", err)
91 if len(p) != 1 || p[0] != 42 {
92 t.Fatalf("got %v, want {42}", p)
94 p, err = b.readPacket()
96 t.Fatalf("got %v, %v, want EOF", p, err)
100 func TestDoubleClose(t *testing.T) {
104 t.Errorf("Close: %v", err)
108 t.Errorf("expect EOF on double close.")