| // 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 httputil |
| |
| import ( |
| "bytes" |
| "io/ioutil" |
| "testing" |
| ) |
| |
| func TestChunk(t *testing.T) { |
| var b bytes.Buffer |
| |
| w := NewChunkedWriter(&b) |
| const chunk1 = "hello, " |
| const chunk2 = "world! 0123456789abcdef" |
| w.Write([]byte(chunk1)) |
| w.Write([]byte(chunk2)) |
| w.Close() |
| |
| if g, e := b.String(), "7\r\nhello, \r\n17\r\nworld! 0123456789abcdef\r\n0\r\n"; g != e { |
| t.Fatalf("chunk writer wrote %q; want %q", g, e) |
| } |
| |
| r := NewChunkedReader(&b) |
| data, err := ioutil.ReadAll(r) |
| if err != nil { |
| t.Fatalf("ReadAll from NewChunkedReader: %v", err) |
| } |
| if g, e := string(data), chunk1+chunk2; g != e { |
| t.Errorf("chunk reader read %q; want %q", g, e) |
| } |
| } |
