mirror of
https://github.com/opencloud-eu/opencloud.git
synced 2026-07-13 09:02:09 -04:00
7e3b722961
* introduce a new toplevel package pkg/http for plain HTTP related functions * move these functions from pkg/jmap to pkg/http, including tests: - PeekResponse - DumpHttpRequest - DumpHttpResponse
93 lines
1.9 KiB
Go
93 lines
1.9 KiB
Go
package http
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPeekResponse(t *testing.T) {
|
|
require := require.New(t)
|
|
n := 6
|
|
text := "hello, world"
|
|
|
|
require.Less(n, len(text))
|
|
body, peek, truncated, err := PeekResponse(io.NopCloser(strings.NewReader(text)), int64(n))
|
|
require.NoError(err)
|
|
require.True(truncated)
|
|
require.Equal(text[:n], string(peek))
|
|
buf := new(strings.Builder)
|
|
_, err = io.Copy(buf, body)
|
|
require.NoError(err)
|
|
require.Equal(text, buf.String())
|
|
}
|
|
|
|
func TestDumpHttpResponse(t *testing.T) {
|
|
require := require.New(t)
|
|
n := 6
|
|
text := "hello, world"
|
|
|
|
h := http.Header{}
|
|
h.Set("testing", "true")
|
|
h.Add("aaa", "1")
|
|
h.Add("aaa", "2")
|
|
|
|
resp := &http.Response{
|
|
Status: "200 OK",
|
|
StatusCode: 200,
|
|
Header: h,
|
|
Body: nil,
|
|
}
|
|
|
|
body, err := DumpHttpResponse(resp, io.NopCloser(strings.NewReader(text)), int64(n), func(method, uri, content string, truncated bool) {
|
|
require.True(truncated)
|
|
require.Equal(fmt.Sprintf(`200 OK
|
|
Testing: true
|
|
Aaa: 1
|
|
Aaa: 2
|
|
%s`, text[:n]), content)
|
|
})
|
|
require.NoError(err)
|
|
buf := new(strings.Builder)
|
|
_, err = io.Copy(buf, body)
|
|
require.NoError(err)
|
|
require.Equal(text, buf.String())
|
|
}
|
|
|
|
func TestDumpHttpRequest(t *testing.T) {
|
|
require := require.New(t)
|
|
n := 6
|
|
text := "hello, world"
|
|
|
|
h := http.Header{}
|
|
h.Set("testing", "true")
|
|
h.Add("aaa", "1")
|
|
h.Add("aaa", "2")
|
|
|
|
req := &http.Request{
|
|
Method: "GET",
|
|
Header: h,
|
|
URL: &url.URL{Scheme: "https", Host: "example.com", Path: "/testing"},
|
|
Body: io.NopCloser(strings.NewReader(text)),
|
|
}
|
|
|
|
err := DumpHttpRequest(req, int64(n), func(method, uri, content string, truncated bool) {
|
|
require.True(truncated)
|
|
require.Equal(fmt.Sprintf(`GET https://example.com/testing
|
|
Testing: true
|
|
Aaa: 1
|
|
Aaa: 2
|
|
%s`, text[:n]), content)
|
|
})
|
|
require.NoError(err)
|
|
buf := new(strings.Builder)
|
|
_, err = io.Copy(buf, req.Body)
|
|
require.NoError(err)
|
|
require.Equal(text, buf.String())
|
|
}
|