net/http/transport_test.go

50 lines
1.0 KiB
Go

package http
import (
"bytes"
"strings"
"testing"
"git.3crabs.ru/VLADIMIR/net/url"
"github.com/stretchr/testify/assert"
)
func Test_WriteRequest(t *testing.T) {
b := &strings.Builder{}
WriteRequest(
b,
&Request{
Method: "GET",
URL: &url.URL{
Path: "/",
},
Protocol: "HTTP/1.0",
},
)
assert.Equal(t, "GET / HTTP/1.0\r\n\r\n", b.String())
}
func Test_ReadResponse(t *testing.T) {
b := bytes.NewBuffer(
[]byte("HTTP/1.1 200 OK\r\nDate: Sat, 03 Feb 2024 16:40:29 GMT\r\nContent-Length: 12\r\nContent-Type: text/plain; charset=utf-8\r\n\r\nHello World!"),
)
r, err := ReadResponse(b)
assert.Nil(t, err)
assert.Equal(t, "200", r.StatusCode)
assert.Equal(t, "Date", r.Headers[0].Name)
assert.Equal(t, "Sat, 03 Feb 2024 16:40:29 GMT", r.Headers[0].Value)
assert.Equal(t, "Content-Length", r.Headers[1].Name)
assert.Equal(t, "12", r.Headers[1].Value)
assert.Equal(t, "Content-Type", r.Headers[2].Name)
assert.Equal(t, "text/plain; charset=utf-8", r.Headers[2].Value)
assert.Equal(t, []byte("Hello World!"), r.Body)
}