summaryrefslogtreecommitdiff
path: root/libgo/go/net/http/responsewrite_test.go
blob: f8e63acf4f7b1d88b05180c164bc164575b2ad34 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
// Copyright 2010 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 (
	"bytes"
	"io/ioutil"
	"testing"
)

type respWriteTest struct {
	Resp Response
	Raw  string
}

var respWriteTests = []respWriteTest{
	// HTTP/1.0, identity coding; no trailer
	{
		Response{
			StatusCode:    503,
			ProtoMajor:    1,
			ProtoMinor:    0,
			Request:       dummyReq("GET"),
			Header:        Header{},
			Body:          ioutil.NopCloser(bytes.NewBufferString("abcdef")),
			ContentLength: 6,
		},

		"HTTP/1.0 503 Service Unavailable\r\n" +
			"Content-Length: 6\r\n\r\n" +
			"abcdef",
	},
	// Unchunked response without Content-Length.
	{
		Response{
			StatusCode:    200,
			ProtoMajor:    1,
			ProtoMinor:    0,
			Request:       dummyReq("GET"),
			Header:        Header{},
			Body:          ioutil.NopCloser(bytes.NewBufferString("abcdef")),
			ContentLength: -1,
		},
		"HTTP/1.0 200 OK\r\n" +
			"\r\n" +
			"abcdef",
	},
	// HTTP/1.1, chunked coding; empty trailer; close
	{
		Response{
			StatusCode:       200,
			ProtoMajor:       1,
			ProtoMinor:       1,
			Request:          dummyReq("GET"),
			Header:           Header{},
			Body:             ioutil.NopCloser(bytes.NewBufferString("abcdef")),
			ContentLength:    6,
			TransferEncoding: []string{"chunked"},
			Close:            true,
		},

		"HTTP/1.1 200 OK\r\n" +
			"Connection: close\r\n" +
			"Transfer-Encoding: chunked\r\n\r\n" +
			"6\r\nabcdef\r\n0\r\n\r\n",
	},

	// Header value with a newline character (Issue 914).
	// Also tests removal of leading and trailing whitespace.
	{
		Response{
			StatusCode: 204,
			ProtoMajor: 1,
			ProtoMinor: 1,
			Request:    dummyReq("GET"),
			Header: Header{
				"Foo": []string{" Bar\nBaz "},
			},
			Body:             nil,
			ContentLength:    0,
			TransferEncoding: []string{"chunked"},
			Close:            true,
		},

		"HTTP/1.1 204 No Content\r\n" +
			"Connection: close\r\n" +
			"Foo: Bar Baz\r\n" +
			"\r\n",
	},
}

func TestResponseWrite(t *testing.T) {
	for i := range respWriteTests {
		tt := &respWriteTests[i]
		var braw bytes.Buffer
		err := tt.Resp.Write(&braw)
		if err != nil {
			t.Errorf("error writing #%d: %s", i, err)
			continue
		}
		sraw := braw.String()
		if sraw != tt.Raw {
			t.Errorf("Test %d, expecting:\n%q\nGot:\n%q\n", i, tt.Raw, sraw)
			continue
		}
	}
}