LEFT | RIGHT |
(no file at all) | |
| 1 // Copyright 2011 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 ssh |
| 6 |
| 7 import ( |
| 8 "io" |
| 9 "testing" |
| 10 ) |
| 11 |
| 12 var BYTES = []byte("abcdefghijklmnopqrstuvwxyz") |
| 13 |
| 14 func TestBufferReadwrite(t *testing.T) { |
| 15 b := newBuffer() |
| 16 b.write(BYTES[:10]) |
| 17 r, _ := b.Read(make([]byte, 10)) |
| 18 if r != 10 { |
| 19 t.Fatalf("Expected written == read == 10, written: 10, read %d",
r) |
| 20 } |
| 21 |
| 22 b = newBuffer() |
| 23 b.write(BYTES[:5]) |
| 24 r, _ = b.Read(make([]byte, 10)) |
| 25 if r != 5 { |
| 26 t.Fatalf("Expected written == read == 5, written: 5, read %d", r
) |
| 27 } |
| 28 |
| 29 b = newBuffer() |
| 30 b.write(BYTES[:10]) |
| 31 r, _ = b.Read(make([]byte, 5)) |
| 32 if r != 5 { |
| 33 t.Fatalf("Expected written == 10, read == 5, written: 10, read %
d", r) |
| 34 } |
| 35 |
| 36 b = newBuffer() |
| 37 b.write(BYTES[:5]) |
| 38 b.write(BYTES[5:15]) |
| 39 r, _ = b.Read(make([]byte, 10)) |
| 40 r2, _ := b.Read(make([]byte, 10)) |
| 41 if r != 10 || r2 != 5 || 15 != r+r2 { |
| 42 t.Fatal("Expected written == read == 15") |
| 43 } |
| 44 } |
| 45 |
| 46 func TestBufferClose(t *testing.T) { |
| 47 b := newBuffer() |
| 48 b.write(BYTES[:10]) |
| 49 b.eof() |
| 50 _, err := b.Read(make([]byte, 5)) |
| 51 if err != nil { |
| 52 t.Fatal("expected read of 5 to not return EOF") |
| 53 } |
| 54 b = newBuffer() |
| 55 b.write(BYTES[:10]) |
| 56 b.eof() |
| 57 r, err := b.Read(make([]byte, 5)) |
| 58 r2, err2 := b.Read(make([]byte, 10)) |
| 59 if r != 5 || r2 != 5 || err != nil || err2 != nil { |
| 60 t.Fatal("expected reads of 5 and 5") |
| 61 } |
| 62 |
| 63 b = newBuffer() |
| 64 b.write(BYTES[:10]) |
| 65 b.eof() |
| 66 r, err = b.Read(make([]byte, 5)) |
| 67 r2, err2 = b.Read(make([]byte, 10)) |
| 68 r3, err3 := b.Read(make([]byte, 10)) |
| 69 if r != 5 || r2 != 5 || r3 != 0 || err != nil || err2 != nil || err3 !=
io.EOF { |
| 70 t.Fatal("expected reads of 5 and 5 and 0, with EOF") |
| 71 } |
| 72 |
| 73 b = newBuffer() |
| 74 b.write(make([]byte, 5)) |
| 75 b.write(make([]byte, 10)) |
| 76 b.eof() |
| 77 r, err = b.Read(make([]byte, 9)) |
| 78 r2, err2 = b.Read(make([]byte, 3)) |
| 79 r3, err3 = b.Read(make([]byte, 3)) |
| 80 r4, err4 := b.Read(make([]byte, 10)) |
| 81 if err != nil || err2 != nil || err3 != nil || err4 != io.EOF { |
| 82 t.Fatalf("Expected EOF on forth read only, err=%v, err2=%v, err3
=%v, err4=%v", err, err2, err3, err4) |
| 83 } |
| 84 if r != 9 || r2 != 3 || r3 != 3 || r4 != 0 { |
| 85 t.Fatal("Expected written == read == 15", r, r2, r3, r4) |
| 86 } |
| 87 } |
LEFT | RIGHT |