return content encoding

This commit is contained in:
Niccolo Borgioli 2024-05-26 01:03:46 +02:00
parent 4669b2602c
commit a0c57541c2
No known key found for this signature in database
GPG Key ID: 4897ACD13A65977C
2 changed files with 24 additions and 0 deletions

View File

@ -64,6 +64,10 @@ type Routes struct {
}
func Respond(conn net.Conn, req Request, res Response) {
if res.Headers == nil {
res.Headers = make(map[string]string)
}
fmt.Fprintf(conn, "%s %d %s%s", res.Version, res.Code.Code, res.Code.Message, HTTPDelimiter)
bodySize := 0
if res.Body != "" {
@ -78,6 +82,10 @@ func Respond(conn net.Conn, req Request, res Response) {
fmt.Fprintf(conn, "%s: %s%s", header, value, HTTPDelimiter)
}
if req.Headers["Accept-Encoding"] == "gzip" {
res.Headers["Content-Encoding"] = "gzip"
}
fmt.Fprint(conn, HTTPDelimiter)
if bodySize > 0 {
if res.Body != "" {

View File

@ -38,7 +38,9 @@ func checkResponse(t *testing.T, res *http.Response, expected Expected) {
t.Errorf(`Expected body to be "%s" but got "%s"`, expected.body, body)
}
// fmt.Println(expected.headers)
for header, value := range expected.headers {
// fmt.Println(header, res.Header[header])
if actual := res.Header[header][0]; actual != value {
t.Errorf(`Expected "%s" header to be "%s" but got "%s"`, header, value, actual)
}
@ -64,6 +66,20 @@ func TestEcho(t *testing.T) {
}})
}
// func TestEchoGzip(t *testing.T) {
// input := "abc"
// req, _ := http.NewRequest("GET", fmt.Sprintf("http://localhost:4221/echo/%s", input), nil)
// req.Header.Set("Accept-Encoding", "gzip")
// client := &http.Client{}
// res, _ := client.Do(req)
// checkResponse(t, res, Expected{status: 200, body: input, headers: map[string]string{
// // "Content-Length": strconv.Itoa(len(input)),
// "Content-Type": "text/plain",
// "Content-Encoding": "gzip",
// }})
// }
func TestUserAgent(t *testing.T) {
input := "CodeCrafters/1.0"
req, _ := http.NewRequest("GET", "http://localhost:4221/user-agent", nil)