forked from rook/rook
Joshua Hoblitt
57b7eeec80
To allow the caller to pass in their own transport when testing. Signed-off-by: Joshua Hoblitt <josh@hoblitt.com>
102 lines
3.9 KiB
Go
102 lines
3.9 KiB
Go
/*
|
|
Copyright 2021 The Rook Authors. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package object
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewS3Agent(t *testing.T) {
|
|
accessKey := "accessKey"
|
|
secretKey := "secretKey"
|
|
endpoint := "endpoint"
|
|
|
|
t.Run("test without tls/debug", func(t *testing.T) {
|
|
debug := false
|
|
insecure := false
|
|
s3Agent, err := NewS3Agent(accessKey, secretKey, endpoint, debug, nil, insecure, nil)
|
|
assert.NoError(t, err)
|
|
assert.NotEqual(t, aws.LogDebug, s3Agent.Client.Config.LogLevel)
|
|
assert.Equal(t, nil, s3Agent.Client.Config.HTTPClient.Transport)
|
|
assert.True(t, *s3Agent.Client.Config.DisableSSL)
|
|
})
|
|
t.Run("test with debug without tls", func(t *testing.T) {
|
|
debug := true
|
|
logLevel := aws.LogDebug
|
|
insecure := false
|
|
s3Agent, err := NewS3Agent(accessKey, secretKey, endpoint, debug, nil, insecure, nil)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, &logLevel, s3Agent.Client.Config.LogLevel)
|
|
assert.Nil(t, s3Agent.Client.Config.HTTPClient.Transport)
|
|
assert.True(t, *s3Agent.Client.Config.DisableSSL)
|
|
})
|
|
t.Run("test without tls client cert but insecure tls", func(t *testing.T) {
|
|
debug := true
|
|
insecure := true
|
|
s3Agent, err := NewS3Agent(accessKey, secretKey, endpoint, debug, nil, insecure, nil)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, s3Agent.Client.Config.HTTPClient.Transport.(*http.Transport).TLSClientConfig.RootCAs) // still includes sys certs
|
|
assert.True(t, s3Agent.Client.Config.HTTPClient.Transport.(*http.Transport).TLSClientConfig.InsecureSkipVerify)
|
|
assert.False(t, *s3Agent.Client.Config.DisableSSL)
|
|
})
|
|
t.Run("test with secure tls client cert", func(t *testing.T) {
|
|
debug := true
|
|
insecure := false
|
|
tlsCert := []byte("tlsCert")
|
|
s3Agent, err := NewS3Agent(accessKey, secretKey, endpoint, debug, tlsCert, insecure, nil)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, s3Agent.Client.Config.HTTPClient.Transport.(*http.Transport).TLSClientConfig.RootCAs)
|
|
assert.False(t, *s3Agent.Client.Config.DisableSSL)
|
|
})
|
|
t.Run("test with insecure tls client cert", func(t *testing.T) {
|
|
debug := true
|
|
insecure := true
|
|
tlsCert := []byte("tlsCert")
|
|
s3Agent, err := NewS3Agent(accessKey, secretKey, endpoint, debug, tlsCert, insecure, nil)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, s3Agent.Client.Config.HTTPClient.Transport)
|
|
assert.True(t, s3Agent.Client.Config.HTTPClient.Transport.(*http.Transport).TLSClientConfig.InsecureSkipVerify)
|
|
assert.False(t, *s3Agent.Client.Config.DisableSSL)
|
|
})
|
|
t.Run("test with custom http.Client", func(t *testing.T) {
|
|
debug := true
|
|
logLevel := aws.LogDebug
|
|
insecure := false
|
|
httpClient := &http.Client{
|
|
Transport: &http.Transport{
|
|
// set some values to check if they are passed through
|
|
MaxIdleConns: 7,
|
|
MaxIdleConnsPerHost: 13,
|
|
MaxConnsPerHost: 17,
|
|
},
|
|
}
|
|
s3Agent, err := NewS3Agent(accessKey, secretKey, endpoint, debug, nil, insecure, httpClient)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, &logLevel, s3Agent.Client.Config.LogLevel)
|
|
assert.NotNil(t, s3Agent.Client.Config.HTTPClient.Transport)
|
|
assert.True(t, *s3Agent.Client.Config.DisableSSL)
|
|
transport := s3Agent.Client.Config.HTTPClient.Transport
|
|
assert.Equal(t, 7, transport.(*http.Transport).MaxIdleConns)
|
|
assert.Equal(t, 13, transport.(*http.Transport).MaxIdleConnsPerHost)
|
|
assert.Equal(t, 17, transport.(*http.Transport).MaxConnsPerHost)
|
|
})
|
|
}
|