| // 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. |
| |
| package io_test |
| |
| import ( |
| "bytes" |
| . "io" |
| "os" |
| "testing" |
| ) |
| |
| // An version of bytes.Buffer without ReadFrom and WriteTo |
| type Buffer struct { |
| bytes.Buffer |
| ReaderFrom // conflicts with and hides bytes.Buffer's ReaderFrom. |
| WriterTo // conflicts with and hides bytes.Buffer's WriterTo. |
| } |
| |
| // Simple tests, primarily to verify the ReadFrom and WriteTo callouts inside Copy and Copyn. |
| |
| func TestCopy(t *testing.T) { |
| rb := new(Buffer) |
| wb := new(Buffer) |
| rb.WriteString("hello, world.") |
| Copy(wb, rb) |
| if wb.String() != "hello, world." { |
| t.Errorf("Copy did not work properly") |
| } |
| } |
| |
| func TestCopyReadFrom(t *testing.T) { |
| rb := new(Buffer) |
| wb := new(bytes.Buffer) // implements ReadFrom. |
| rb.WriteString("hello, world.") |
| Copy(wb, rb) |
| if wb.String() != "hello, world." { |
| t.Errorf("Copy did not work properly") |
| } |
| } |
| |
| func TestCopyWriteTo(t *testing.T) { |
| rb := new(bytes.Buffer) // implements WriteTo. |
| wb := new(Buffer) |
| rb.WriteString("hello, world.") |
| Copy(wb, rb) |
| if wb.String() != "hello, world." { |
| t.Errorf("Copy did not work properly") |
| } |
| } |
| |
| func TestCopyn(t *testing.T) { |
| rb := new(Buffer) |
| wb := new(Buffer) |
| rb.WriteString("hello, world.") |
| Copyn(wb, rb, 5) |
| if wb.String() != "hello" { |
| t.Errorf("Copyn did not work properly") |
| } |
| } |
| |
| func TestCopynReadFrom(t *testing.T) { |
| rb := new(Buffer) |
| wb := new(bytes.Buffer) // implements ReadFrom. |
| rb.WriteString("hello") |
| Copyn(wb, rb, 5) |
| if wb.String() != "hello" { |
| t.Errorf("Copyn did not work properly") |
| } |
| } |
| |
| func TestCopynWriteTo(t *testing.T) { |
| rb := new(bytes.Buffer) // implements WriteTo. |
| wb := new(Buffer) |
| rb.WriteString("hello, world.") |
| Copyn(wb, rb, 5) |
| if wb.String() != "hello" { |
| t.Errorf("Copyn did not work properly") |
| } |
| } |
| |
| func TestReadAtLeast(t *testing.T) { |
| var rb bytes.Buffer |
| rb.Write([]byte("0123")) |
| buf := make([]byte, 2) |
| n, err := ReadAtLeast(&rb, buf, 2) |
| if err != nil { |
| t.Error(err) |
| } |
| n, err = ReadAtLeast(&rb, buf, 4) |
| if err != ErrShortBuffer { |
| t.Errorf("expected ErrShortBuffer got %v", err) |
| } |
| if n != 0 { |
| t.Errorf("expected to have read 0 bytes, got %v", n) |
| } |
| n, err = ReadAtLeast(&rb, buf, 1) |
| if err != nil { |
| t.Error(err) |
| } |
| if n != 2 { |
| t.Errorf("expected to have read 2 bytes, got %v", n) |
| } |
| n, err = ReadAtLeast(&rb, buf, 2) |
| if err != os.EOF { |
| t.Errorf("expected EOF, got %v", err) |
| } |
| if n != 0 { |
| t.Errorf("expected to have read 0 bytes, got %v", n) |
| } |
| rb.Write([]byte("4")) |
| n, err = ReadAtLeast(&rb, buf, 2) |
| if err != ErrUnexpectedEOF { |
| t.Errorf("expected ErrUnexpectedEOF, got %v", err) |
| } |
| if n != 1 { |
| t.Errorf("expected to have read 1 bytes, got %v", n) |
| } |
| } |