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
|
package http3
import (
"bytes"
"io"
"net/http"
"net/http/httptest"
"testing"
"github.com/quic-go/qpack"
"github.com/stretchr/testify/require"
)
func decodeRequest(t *testing.T, str io.Reader) map[string]string {
t.Helper()
fp := frameParser{r: str}
frame, err := fp.ParseNext()
require.NoError(t, err)
require.IsType(t, &headersFrame{}, frame)
headersFrame := frame.(*headersFrame)
data := make([]byte, headersFrame.Length)
_, err = io.ReadFull(str, data)
require.NoError(t, err)
decoder := qpack.NewDecoder(nil)
hfs, err := decoder.DecodeFull(data)
require.NoError(t, err)
values := make(map[string]string)
for _, hf := range hfs {
values[hf.Name] = hf.Value
}
return values
}
func TestRequestWriterGetRequestGzip(t *testing.T) {
t.Run("gzip", func(t *testing.T) {
testRequestWriterGzip(t, true)
})
t.Run("no gzip", func(t *testing.T) {
testRequestWriterGzip(t, false)
})
}
func testRequestWriterGzip(t *testing.T, gzip bool) {
req := httptest.NewRequest(http.MethodGet, "https://quic-go.net/index.html?foo=bar", nil)
req.AddCookie(&http.Cookie{Name: "foo", Value: "bar"})
req.AddCookie(&http.Cookie{Name: "baz", Value: "lorem ipsum"})
rw := newRequestWriter()
buf := &bytes.Buffer{}
require.NoError(t, rw.WriteRequestHeader(buf, req, gzip))
headerFields := decodeRequest(t, buf)
require.Equal(t, "quic-go.net", headerFields[":authority"])
require.Equal(t, http.MethodGet, headerFields[":method"])
require.Equal(t, "/index.html?foo=bar", headerFields[":path"])
require.Equal(t, "https", headerFields[":scheme"])
require.Equal(t, `foo=bar; baz="lorem ipsum"`, headerFields["cookie"])
switch gzip {
case true:
require.Equal(t, "gzip", headerFields["accept-encoding"])
case false:
require.NotContains(t, headerFields, "accept-encoding")
}
}
func TestRequestWriterInvalidHostHeader(t *testing.T) {
req := httptest.NewRequest(http.MethodGet, "https://quic-go.net/index.html?foo=bar", nil)
req.Host = "foo@bar" // @ is invalid
rw := newRequestWriter()
require.EqualError(t,
rw.WriteRequestHeader(&bytes.Buffer{}, req, false),
"http3: invalid Host header",
)
}
func TestRequestWriterConnect(t *testing.T) {
// httptest.NewRequest does not properly support the CONNECT method
req, err := http.NewRequest(http.MethodConnect, "https://quic-go.net/", nil)
require.NoError(t, err)
rw := newRequestWriter()
buf := &bytes.Buffer{}
require.NoError(t, rw.WriteRequestHeader(buf, req, false))
headerFields := decodeRequest(t, buf)
require.Equal(t, http.MethodConnect, headerFields[":method"])
require.Equal(t, "quic-go.net", headerFields[":authority"])
require.NotContains(t, headerFields, ":path")
require.NotContains(t, headerFields, ":scheme")
require.NotContains(t, headerFields, ":protocol")
}
func TestRequestWriterExtendedConnect(t *testing.T) {
// httptest.NewRequest does not properly support the CONNECT method
req, err := http.NewRequest(http.MethodConnect, "https://quic-go.net/", nil)
require.NoError(t, err)
req.Proto = "webtransport"
rw := newRequestWriter()
buf := &bytes.Buffer{}
require.NoError(t, rw.WriteRequestHeader(buf, req, false))
headerFields := decodeRequest(t, buf)
require.Equal(t, "quic-go.net", headerFields[":authority"])
require.Equal(t, http.MethodConnect, headerFields[":method"])
require.Equal(t, "/", headerFields[":path"])
require.Equal(t, "https", headerFields[":scheme"])
require.Equal(t, "webtransport", headerFields[":protocol"])
}
|
