added for other providers

Signed-off-by: Vanshikav123 <vanshikav928@gmail.com>

client/factory.go

@@ -83,7 +83,7 @@ func NewBucket(logger log.Logger, confContentYaml []byte, component string, rt h
 	case string(OCI):
 		bucket, err = oci.NewBucket(logger, config, rt)
 	case string(OBS):
-		bucket, err = obs.NewBucket(logger, config)
+		bucket, err = obs.NewBucket(logger, config, rt)
 	default:
 		return nil, errors.Errorf("bucket with type %s is not supported", bucketConf.Type)
 	}

providers/obs/obs.go

@@ -7,6 +7,7 @@ import (
 	"context"
 	"io"
 	"math"
+	"net/http"
 	"os"
 	"strings"
 	"testing"
@@ -74,13 +75,13 @@ type Bucket struct {
 	name   string
 }
 
-func NewBucket(logger log.Logger, conf []byte) (*Bucket, error) {
+func NewBucket(logger log.Logger, conf []byte, rt http.RoundTripper) (*Bucket, error) {
 	config, err := parseConfig(conf)
 	if err != nil {
 		return nil, errors.Wrap(err, "parsing cos configuration")
 	}
 
-	return NewBucketWithConfig(logger, config)
+	return NewBucketWithConfig(logger, config, rt)
 }
 
 func parseConfig(conf []byte) (Config, error) {
@@ -92,17 +93,27 @@ func parseConfig(conf []byte) (Config, error) {
 	return config, nil
 }
 
-func NewBucketWithConfig(logger log.Logger, config Config) (*Bucket, error) {
+func NewBucketWithConfig(logger log.Logger, config Config, rt http.RoundTripper) (*Bucket, error) {
 	if err := config.validate(); err != nil {
 		return nil, errors.Wrap(err, "validate obs config err")
 	}
+	var tpt *http.Transport
+	var err error
 
-	rt, err := exthttp.DefaultTransport(config.HTTPConfig)
-	if err != nil {
-		return nil, errors.Wrap(err, "get http transport err")
+	if rt != nil {
+		var ok bool
+		tpt, ok = rt.(*http.Transport)
+		if !ok {
+			return nil, errors.New("provided RoundTripper is not an *http.Transport")
+		}
+	} else {
+		tpt, err = exthttp.DefaultTransport(config.HTTPConfig)
+		if err != nil {
+			return nil, errors.Wrap(err, "get http transport err")
+		}
 	}
 
-	client, err := obs.New(config.AccessKey, config.SecretKey, config.Endpoint, obs.WithHttpTransport(rt))
+	client, err := obs.New(config.AccessKey, config.SecretKey, config.Endpoint, obs.WithHttpTransport(tpt))
 	if err != nil {
 		return nil, errors.Wrap(err, "initialize obs client err")
 	}
@@ -369,7 +380,7 @@ func NewTestBucketFromConfig(t testing.TB, c Config, reuseBucket bool, location
 	if err != nil {
 		return nil, nil, err
 	}
-	b, err := NewBucket(log.NewNopLogger(), bc)
+	b, err := NewBucket(log.NewNopLogger(), bc, http.DefaultTransport)
 	if err != nil {
 		return nil, nil, err
 	}