|
|
|
@ -1,6 +1,7 @@ |
|
|
|
|
package httpclientprovider |
|
|
|
|
|
|
|
|
|
import ( |
|
|
|
|
"fmt" |
|
|
|
|
"net/http" |
|
|
|
|
"testing" |
|
|
|
|
|
|
|
|
@ -14,12 +15,12 @@ func TestSigV4Middleware(t *testing.T) { |
|
|
|
|
origSigV4Func := newSigV4Func |
|
|
|
|
newSigV4Called := false |
|
|
|
|
middlewareCalled := false |
|
|
|
|
newSigV4Func = func(config *sigv4.Config, next http.RoundTripper) http.RoundTripper { |
|
|
|
|
newSigV4Func = func(config *sigv4.Config, next http.RoundTripper, opts ...sigv4.Opts) (http.RoundTripper, error) { |
|
|
|
|
newSigV4Called = true |
|
|
|
|
return httpclient.RoundTripperFunc(func(r *http.Request) (*http.Response, error) { |
|
|
|
|
middlewareCalled = true |
|
|
|
|
return next.RoundTrip(r) |
|
|
|
|
}) |
|
|
|
|
}), nil |
|
|
|
|
} |
|
|
|
|
t.Cleanup(func() { |
|
|
|
|
newSigV4Func = origSigV4Func |
|
|
|
@ -27,7 +28,7 @@ func TestSigV4Middleware(t *testing.T) { |
|
|
|
|
|
|
|
|
|
ctx := &testContext{} |
|
|
|
|
finalRoundTripper := ctx.createRoundTripper("finalrt") |
|
|
|
|
mw := SigV4Middleware() |
|
|
|
|
mw := SigV4Middleware(false) |
|
|
|
|
rt := mw.CreateMiddleware(httpclient.Options{}, finalRoundTripper) |
|
|
|
|
require.NotNil(t, rt) |
|
|
|
|
middlewareName, ok := mw.(httpclient.MiddlewareName) |
|
|
|
@ -52,12 +53,12 @@ func TestSigV4Middleware(t *testing.T) { |
|
|
|
|
origSigV4Func := newSigV4Func |
|
|
|
|
newSigV4Called := false |
|
|
|
|
middlewareCalled := false |
|
|
|
|
newSigV4Func = func(config *sigv4.Config, next http.RoundTripper) http.RoundTripper { |
|
|
|
|
newSigV4Func = func(config *sigv4.Config, next http.RoundTripper, opts ...sigv4.Opts) (http.RoundTripper, error) { |
|
|
|
|
newSigV4Called = true |
|
|
|
|
return httpclient.RoundTripperFunc(func(r *http.Request) (*http.Response, error) { |
|
|
|
|
middlewareCalled = true |
|
|
|
|
return next.RoundTrip(r) |
|
|
|
|
}) |
|
|
|
|
}), nil |
|
|
|
|
} |
|
|
|
|
t.Cleanup(func() { |
|
|
|
|
newSigV4Func = origSigV4Func |
|
|
|
@ -65,7 +66,7 @@ func TestSigV4Middleware(t *testing.T) { |
|
|
|
|
|
|
|
|
|
ctx := &testContext{} |
|
|
|
|
finalRoundTripper := ctx.createRoundTripper("final") |
|
|
|
|
mw := SigV4Middleware() |
|
|
|
|
mw := SigV4Middleware(false) |
|
|
|
|
rt := mw.CreateMiddleware(httpclient.Options{SigV4: &httpclient.SigV4Config{}}, finalRoundTripper) |
|
|
|
|
require.NotNil(t, rt) |
|
|
|
|
middlewareName, ok := mw.(httpclient.MiddlewareName) |
|
|
|
@ -86,4 +87,32 @@ func TestSigV4Middleware(t *testing.T) { |
|
|
|
|
require.True(t, newSigV4Called) |
|
|
|
|
require.True(t, middlewareCalled) |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
t.Run("With sigv4 error returned", func(t *testing.T) { |
|
|
|
|
origSigV4Func := newSigV4Func |
|
|
|
|
newSigV4Func = func(config *sigv4.Config, next http.RoundTripper, opts ...sigv4.Opts) (http.RoundTripper, error) { |
|
|
|
|
return nil, fmt.Errorf("problem") |
|
|
|
|
} |
|
|
|
|
t.Cleanup(func() { |
|
|
|
|
newSigV4Func = origSigV4Func |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
ctx := &testContext{} |
|
|
|
|
finalRoundTripper := ctx.createRoundTripper("final") |
|
|
|
|
mw := SigV4Middleware(false) |
|
|
|
|
rt := mw.CreateMiddleware(httpclient.Options{SigV4: &httpclient.SigV4Config{}}, finalRoundTripper) |
|
|
|
|
require.NotNil(t, rt) |
|
|
|
|
middlewareName, ok := mw.(httpclient.MiddlewareName) |
|
|
|
|
require.True(t, ok) |
|
|
|
|
require.Equal(t, SigV4MiddlewareName, middlewareName.MiddlewareName()) |
|
|
|
|
|
|
|
|
|
req, err := http.NewRequest(http.MethodGet, "http://", nil) |
|
|
|
|
require.NoError(t, err) |
|
|
|
|
// response is nil
|
|
|
|
|
// nolint:bodyclose
|
|
|
|
|
res, err := rt.RoundTrip(req) |
|
|
|
|
require.Error(t, err) |
|
|
|
|
require.Nil(t, res) |
|
|
|
|
require.Empty(t, ctx.callChain) |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|