mirror of https://github.com/gogits/gogs.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.2 KiB
87 lines
2.2 KiB
// Copyright 2011 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" |
|
"testing" |
|
) |
|
|
|
var alphabet = []byte("abcdefghijklmnopqrstuvwxyz") |
|
|
|
func TestBufferReadwrite(t *testing.T) { |
|
b := newBuffer() |
|
b.write(alphabet[:10]) |
|
r, _ := b.Read(make([]byte, 10)) |
|
if r != 10 { |
|
t.Fatalf("Expected written == read == 10, written: 10, read %d", r) |
|
} |
|
|
|
b = newBuffer() |
|
b.write(alphabet[:5]) |
|
r, _ = b.Read(make([]byte, 10)) |
|
if r != 5 { |
|
t.Fatalf("Expected written == read == 5, written: 5, read %d", r) |
|
} |
|
|
|
b = newBuffer() |
|
b.write(alphabet[:10]) |
|
r, _ = b.Read(make([]byte, 5)) |
|
if r != 5 { |
|
t.Fatalf("Expected written == 10, read == 5, written: 10, read %d", r) |
|
} |
|
|
|
b = newBuffer() |
|
b.write(alphabet[:5]) |
|
b.write(alphabet[5:15]) |
|
r, _ = b.Read(make([]byte, 10)) |
|
r2, _ := b.Read(make([]byte, 10)) |
|
if r != 10 || r2 != 5 || 15 != r+r2 { |
|
t.Fatal("Expected written == read == 15") |
|
} |
|
} |
|
|
|
func TestBufferClose(t *testing.T) { |
|
b := newBuffer() |
|
b.write(alphabet[:10]) |
|
b.eof() |
|
_, err := b.Read(make([]byte, 5)) |
|
if err != nil { |
|
t.Fatal("expected read of 5 to not return EOF") |
|
} |
|
b = newBuffer() |
|
b.write(alphabet[:10]) |
|
b.eof() |
|
r, err := b.Read(make([]byte, 5)) |
|
r2, err2 := b.Read(make([]byte, 10)) |
|
if r != 5 || r2 != 5 || err != nil || err2 != nil { |
|
t.Fatal("expected reads of 5 and 5") |
|
} |
|
|
|
b = newBuffer() |
|
b.write(alphabet[:10]) |
|
b.eof() |
|
r, err = b.Read(make([]byte, 5)) |
|
r2, err2 = b.Read(make([]byte, 10)) |
|
r3, err3 := b.Read(make([]byte, 10)) |
|
if r != 5 || r2 != 5 || r3 != 0 || err != nil || err2 != nil || err3 != io.EOF { |
|
t.Fatal("expected reads of 5 and 5 and 0, with EOF") |
|
} |
|
|
|
b = newBuffer() |
|
b.write(make([]byte, 5)) |
|
b.write(make([]byte, 10)) |
|
b.eof() |
|
r, err = b.Read(make([]byte, 9)) |
|
r2, err2 = b.Read(make([]byte, 3)) |
|
r3, err3 = b.Read(make([]byte, 3)) |
|
r4, err4 := b.Read(make([]byte, 10)) |
|
if err != nil || err2 != nil || err3 != nil || err4 != io.EOF { |
|
t.Fatalf("Expected EOF on forth read only, err=%v, err2=%v, err3=%v, err4=%v", err, err2, err3, err4) |
|
} |
|
if r != 9 || r2 != 3 || r3 != 3 || r4 != 0 { |
|
t.Fatal("Expected written == read == 15", r, r2, r3, r4) |
|
} |
|
}
|
|
|