| // Copyright 2012 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 http |
| |
| import ( |
| "bufio" |
| "strings" |
| "testing" |
| ) |
| |
| func TestBodyReadBadTrailer(t *testing.T) { |
| b := &body{ |
| Reader: strings.NewReader("foobar"), |
| hdr: true, // force reading the trailer |
| r: bufio.NewReader(strings.NewReader("")), |
| } |
| buf := make([]byte, 7) |
| n, err := b.Read(buf[:3]) |
| got := string(buf[:n]) |
| if got != "foo" || err != nil { |
| t.Fatalf(`first Read = %d (%q), %v; want 3 ("foo")`, n, got, err) |
| } |
| |
| n, err = b.Read(buf[:]) |
| got = string(buf[:n]) |
| if got != "bar" || err != nil { |
| t.Fatalf(`second Read = %d (%q), %v; want 3 ("bar")`, n, got, err) |
| } |
| |
| n, err = b.Read(buf[:]) |
| got = string(buf[:n]) |
| if err == nil { |
| t.Errorf("final Read was successful (%q), expected error from trailer read", got) |
| } |
| } |