| // 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) | |
| } | |
| } |