1 // Copyright 2010 The Go Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
16 type canonicalHeaderKeyTest
struct {
20 var canonicalHeaderKeyTests
= []canonicalHeaderKeyTest
{
23 {"User-Agent", "User-Agent"},
24 {"uSER-aGENT", "User-Agent"},
25 {"user-agent", "User-Agent"},
26 {"USER-AGENT", "User-Agent"},
29 func TestCanonicalHeaderKey(t
*testing
.T
) {
30 for _
, tt
:= range canonicalHeaderKeyTests
{
31 if s
:= CanonicalHeaderKey(tt
.in
); s
!= tt
.out
{
32 t
.Errorf("CanonicalHeaderKey(%q) = %q, want %q", tt
.in
, s
, tt
.out
)
37 func reader(s
string) *Reader
{
38 return NewReader(bufio
.NewReader(strings
.NewReader(s
)))
41 func TestReadLine(t
*testing
.T
) {
42 r
:= reader("line1\nline2\n")
43 s
, err
:= r
.ReadLine()
44 if s
!= "line1" || err
!= nil {
45 t
.Fatalf("Line 1: %s, %v", s
, err
)
48 if s
!= "line2" || err
!= nil {
49 t
.Fatalf("Line 2: %s, %v", s
, err
)
52 if s
!= "" || err
!= os
.EOF
{
53 t
.Fatalf("EOF: %s, %v", s
, err
)
57 func TestReadContinuedLine(t
*testing
.T
) {
58 r
:= reader("line1\nline\n 2\nline3\n")
59 s
, err
:= r
.ReadContinuedLine()
60 if s
!= "line1" || err
!= nil {
61 t
.Fatalf("Line 1: %s, %v", s
, err
)
63 s
, err
= r
.ReadContinuedLine()
64 if s
!= "line 2" || err
!= nil {
65 t
.Fatalf("Line 2: %s, %v", s
, err
)
67 s
, err
= r
.ReadContinuedLine()
68 if s
!= "line3" || err
!= nil {
69 t
.Fatalf("Line 3: %s, %v", s
, err
)
71 s
, err
= r
.ReadContinuedLine()
72 if s
!= "" || err
!= os
.EOF
{
73 t
.Fatalf("EOF: %s, %v", s
, err
)
77 func TestReadCodeLine(t
*testing
.T
) {
78 r
:= reader("123 hi\n234 bye\n345 no way\n")
79 code
, msg
, err
:= r
.ReadCodeLine(0)
80 if code
!= 123 || msg
!= "hi" || err
!= nil {
81 t
.Fatalf("Line 1: %d, %s, %v", code
, msg
, err
)
83 code
, msg
, err
= r
.ReadCodeLine(23)
84 if code
!= 234 || msg
!= "bye" || err
!= nil {
85 t
.Fatalf("Line 2: %d, %s, %v", code
, msg
, err
)
87 code
, msg
, err
= r
.ReadCodeLine(346)
88 if code
!= 345 || msg
!= "no way" || err
== nil {
89 t
.Fatalf("Line 3: %d, %s, %v", code
, msg
, err
)
91 if e
, ok
:= err
.(*Error
); !ok || e
.Code
!= code || e
.Msg
!= msg
{
92 t
.Fatalf("Line 3: wrong error %v\n", err
)
94 code
, msg
, err
= r
.ReadCodeLine(1)
95 if code
!= 0 || msg
!= "" || err
!= os
.EOF
{
96 t
.Fatalf("EOF: %d, %s, %v", code
, msg
, err
)
100 func TestReadDotLines(t
*testing
.T
) {
101 r
:= reader("dotlines\r\n.foo\r\n..bar\n...baz\nquux\r\n\r\n.\r\nanother\n")
102 s
, err
:= r
.ReadDotLines()
103 want
:= []string{"dotlines", "foo", ".bar", "..baz", "quux", ""}
104 if !reflect
.DeepEqual(s
, want
) || err
!= nil {
105 t
.Fatalf("ReadDotLines: %v, %v", s
, err
)
108 s
, err
= r
.ReadDotLines()
109 want
= []string{"another"}
110 if !reflect
.DeepEqual(s
, want
) || err
!= io
.ErrUnexpectedEOF
{
111 t
.Fatalf("ReadDotLines2: %v, %v", s
, err
)
115 func TestReadDotBytes(t
*testing
.T
) {
116 r
:= reader("dotlines\r\n.foo\r\n..bar\n...baz\nquux\r\n\r\n.\r\nanot.her\r\n")
117 b
, err
:= r
.ReadDotBytes()
118 want
:= []byte("dotlines\nfoo\n.bar\n..baz\nquux\n\n")
119 if !reflect
.DeepEqual(b
, want
) || err
!= nil {
120 t
.Fatalf("ReadDotBytes: %q, %v", b
, err
)
123 b
, err
= r
.ReadDotBytes()
124 want
= []byte("anot.her\n")
125 if !reflect
.DeepEqual(b
, want
) || err
!= io
.ErrUnexpectedEOF
{
126 t
.Fatalf("ReadDotBytes2: %q, %v", b
, err
)
130 func TestReadMIMEHeader(t
*testing
.T
) {
131 r
:= reader("my-key: Value 1 \r\nLong-key: Even \n Longer Value\r\nmy-Key: Value 2\r\n\n")
132 m
, err
:= r
.ReadMIMEHeader()
133 want
:= map[string][]string{
134 "My-Key": {"Value 1", "Value 2"},
135 "Long-Key": {"Even Longer Value"},
137 if !reflect
.DeepEqual(m
, want
) || err
!= nil {
138 t
.Fatalf("ReadMIMEHeader: %v, %v; want %v", m
, err
, want
)