114 lines
3.8 KiB
Go
114 lines
3.8 KiB
Go
package h2quic
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"golang.org/x/net/http2"
|
|
"golang.org/x/net/http2/hpack"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Request", func() {
|
|
var (
|
|
rw *requestWriter
|
|
headerStream *mockStream
|
|
decoder *hpack.Decoder
|
|
)
|
|
|
|
BeforeEach(func() {
|
|
headerStream = &mockStream{}
|
|
rw = newRequestWriter(headerStream)
|
|
decoder = hpack.NewDecoder(4096, func(hf hpack.HeaderField) {})
|
|
})
|
|
|
|
decode := func(p []byte) (*http2.HeadersFrame, map[string] /* HeaderField.Name */ string /* HeaderField.Value */) {
|
|
framer := http2.NewFramer(nil, bytes.NewReader(p))
|
|
frame, err := framer.ReadFrame()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
headerFrame := frame.(*http2.HeadersFrame)
|
|
fields, err := decoder.DecodeFull(headerFrame.HeaderBlockFragment())
|
|
Expect(err).ToNot(HaveOccurred())
|
|
values := make(map[string]string)
|
|
for _, headerField := range fields {
|
|
values[headerField.Name] = headerField.Value
|
|
}
|
|
return headerFrame, values
|
|
}
|
|
|
|
It("writes a GET request", func() {
|
|
req, err := http.NewRequest("GET", "https://quic.clemente.io/index.html?foo=bar", nil)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
rw.WriteRequest(req, 1337, true, false)
|
|
headerFrame, headerFields := decode(headerStream.dataWritten.Bytes())
|
|
Expect(headerFrame.StreamID).To(Equal(uint32(1337)))
|
|
Expect(headerFrame.HasPriority()).To(BeTrue())
|
|
Expect(headerFields).To(HaveKeyWithValue(":authority", "quic.clemente.io"))
|
|
Expect(headerFields).To(HaveKeyWithValue(":method", "GET"))
|
|
Expect(headerFields).To(HaveKeyWithValue(":path", "/index.html?foo=bar"))
|
|
Expect(headerFields).To(HaveKeyWithValue(":scheme", "https"))
|
|
Expect(headerFields).ToNot(HaveKey("accept-encoding"))
|
|
})
|
|
|
|
It("sets the EndStream header", func() {
|
|
req, err := http.NewRequest("GET", "https://quic.clemente.io/", nil)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
rw.WriteRequest(req, 1337, true, false)
|
|
headerFrame, _ := decode(headerStream.dataWritten.Bytes())
|
|
Expect(headerFrame.StreamEnded()).To(BeTrue())
|
|
})
|
|
|
|
It("doesn't set the EndStream header, if requested", func() {
|
|
req, err := http.NewRequest("GET", "https://quic.clemente.io/", nil)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
rw.WriteRequest(req, 1337, false, false)
|
|
headerFrame, _ := decode(headerStream.dataWritten.Bytes())
|
|
Expect(headerFrame.StreamEnded()).To(BeFalse())
|
|
})
|
|
|
|
It("requests gzip compression, if requested", func() {
|
|
req, err := http.NewRequest("GET", "https://quic.clemente.io/index.html?foo=bar", nil)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
rw.WriteRequest(req, 1337, true, true)
|
|
_, headerFields := decode(headerStream.dataWritten.Bytes())
|
|
Expect(headerFields).To(HaveKeyWithValue("accept-encoding", "gzip"))
|
|
})
|
|
|
|
It("writes a POST request", func() {
|
|
form := url.Values{}
|
|
form.Add("foo", "bar")
|
|
req, err := http.NewRequest("POST", "https://quic.clemente.io/upload.html", strings.NewReader(form.Encode()))
|
|
Expect(err).ToNot(HaveOccurred())
|
|
rw.WriteRequest(req, 5, true, false)
|
|
_, headerFields := decode(headerStream.dataWritten.Bytes())
|
|
Expect(headerFields).To(HaveKeyWithValue(":method", "POST"))
|
|
Expect(headerFields).To(HaveKey("content-length"))
|
|
contentLength, err := strconv.Atoi(headerFields["content-length"])
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(contentLength).To(BeNumerically(">", 0))
|
|
})
|
|
|
|
It("sends cookies", func() {
|
|
req, err := http.NewRequest("GET", "https://quic.clemente.io/", nil)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
cookie1 := &http.Cookie{
|
|
Name: "Cookie #1",
|
|
Value: "Value #1",
|
|
}
|
|
cookie2 := &http.Cookie{
|
|
Name: "Cookie #2",
|
|
Value: "Value #2",
|
|
}
|
|
req.AddCookie(cookie1)
|
|
req.AddCookie(cookie2)
|
|
rw.WriteRequest(req, 11, true, false)
|
|
_, headerFields := decode(headerStream.dataWritten.Bytes())
|
|
Expect(headerFields).To(HaveKeyWithValue("cookie", "Cookie #1=Value #1; Cookie #2=Value #2"))
|
|
})
|
|
})
|