3279a7ef0f
* feat: change x-trace-id to traceparent to follow opentelemetry * feat: add rest/httpc to make http requests governacible * chore: remove blank lines
40 lines
1.0 KiB
Go
40 lines
1.0 KiB
Go
package httpc
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestDo(t *testing.T) {
|
|
svr := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
}))
|
|
req, err := http.NewRequest(http.MethodGet, svr.URL, nil)
|
|
assert.Nil(t, err)
|
|
resp, err := Do("foo", req, func(cli *http.Client) {
|
|
cli.Transport = http.DefaultTransport
|
|
})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
}
|
|
|
|
func TestDoNotFound(t *testing.T) {
|
|
svr := httptest.NewServer(http.NotFoundHandler())
|
|
req, err := http.NewRequest(http.MethodGet, svr.URL, nil)
|
|
assert.Nil(t, err)
|
|
resp, err := Do("foo", req)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, http.StatusNotFound, resp.StatusCode)
|
|
}
|
|
|
|
func TestDoMoved(t *testing.T) {
|
|
svr := httptest.NewServer(http.RedirectHandler("/foo", http.StatusMovedPermanently))
|
|
req, err := http.NewRequest(http.MethodGet, svr.URL, nil)
|
|
assert.Nil(t, err)
|
|
_, err = Do("foo", req)
|
|
// too many redirects
|
|
assert.NotNil(t, err)
|
|
}
|