OSDN Git Service

Thanos did someting
[bytom/vapor.git] / vendor / golang.org / x / crypto / ssh / mempipe_test.go
diff --git a/vendor/golang.org/x/crypto/ssh/mempipe_test.go b/vendor/golang.org/x/crypto/ssh/mempipe_test.go
deleted file mode 100644 (file)
index 8697cd6..0000000
+++ /dev/null
@@ -1,110 +0,0 @@
-// Copyright 2013 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 ssh
-
-import (
-       "io"
-       "sync"
-       "testing"
-)
-
-// An in-memory packetConn. It is safe to call Close and writePacket
-// from different goroutines.
-type memTransport struct {
-       eof     bool
-       pending [][]byte
-       write   *memTransport
-       sync.Mutex
-       *sync.Cond
-}
-
-func (t *memTransport) readPacket() ([]byte, error) {
-       t.Lock()
-       defer t.Unlock()
-       for {
-               if len(t.pending) > 0 {
-                       r := t.pending[0]
-                       t.pending = t.pending[1:]
-                       return r, nil
-               }
-               if t.eof {
-                       return nil, io.EOF
-               }
-               t.Cond.Wait()
-       }
-}
-
-func (t *memTransport) closeSelf() error {
-       t.Lock()
-       defer t.Unlock()
-       if t.eof {
-               return io.EOF
-       }
-       t.eof = true
-       t.Cond.Broadcast()
-       return nil
-}
-
-func (t *memTransport) Close() error {
-       err := t.write.closeSelf()
-       t.closeSelf()
-       return err
-}
-
-func (t *memTransport) writePacket(p []byte) error {
-       t.write.Lock()
-       defer t.write.Unlock()
-       if t.write.eof {
-               return io.EOF
-       }
-       c := make([]byte, len(p))
-       copy(c, p)
-       t.write.pending = append(t.write.pending, c)
-       t.write.Cond.Signal()
-       return nil
-}
-
-func memPipe() (a, b packetConn) {
-       t1 := memTransport{}
-       t2 := memTransport{}
-       t1.write = &t2
-       t2.write = &t1
-       t1.Cond = sync.NewCond(&t1.Mutex)
-       t2.Cond = sync.NewCond(&t2.Mutex)
-       return &t1, &t2
-}
-
-func TestMemPipe(t *testing.T) {
-       a, b := memPipe()
-       if err := a.writePacket([]byte{42}); err != nil {
-               t.Fatalf("writePacket: %v", err)
-       }
-       if err := a.Close(); err != nil {
-               t.Fatal("Close: ", err)
-       }
-       p, err := b.readPacket()
-       if err != nil {
-               t.Fatal("readPacket: ", err)
-       }
-       if len(p) != 1 || p[0] != 42 {
-               t.Fatalf("got %v, want {42}", p)
-       }
-       p, err = b.readPacket()
-       if err != io.EOF {
-               t.Fatalf("got %v, %v, want EOF", p, err)
-       }
-}
-
-func TestDoubleClose(t *testing.T) {
-       a, _ := memPipe()
-       err := a.Close()
-       if err != nil {
-               t.Errorf("Close: %v", err)
-       }
-       err = a.Close()
-       if err != io.EOF {
-               t.Errorf("expect EOF on double close.")
-       }
-}