github.com/webdestroya/awsmocker@v0.2.6/main.go

package awsmocker

import (
	"os"
	"time"

	"github.com/aws/aws-sdk-go-v2/aws"
)

// Returned when you start the server, provides you some information if needed
type MockerInfo struct {
	// URL of the proxy server
	ProxyURL string

	// Aws configuration to use
	// This is only provided if you gave ReturnAwsConfig in the options
	AwsConfig *aws.Config
}

func Start(t TestingT, options *MockerOptions) *MockerInfo {

	if h, ok := t.(tHelper); ok {
		h.Helper()
	}

	if options == nil {
		options = &MockerOptions{}
	}

	if options.Timeout == 0 {
		options.Timeout = 5 * time.Second
	}

	if !options.SkipDefaultMocks {
		options.Mocks = append(options.Mocks, MockStsGetCallerIdentityValid)
	}

	if options.MockEc2Metadata {
		options.Mocks = append(options.Mocks, Mock_IMDS_Common()...)
	}

	// proxy bypass configuration
	if options.DoNotProxy != "" {
		noProxyStr := os.Getenv("NO_PROXY")
		if noProxyStr == "" {
			noProxyStr = os.Getenv("no_proxy")
		}
		if noProxyStr != "" {
			noProxyStr += ","
		}
		noProxyStr += options.DoNotProxy

		t.Setenv("NO_PROXY", noProxyStr)
		t.Setenv("no_proxy", noProxyStr)
	}

	mocks := make([]*MockedEndpoint, 0, len(options.Mocks))
	for i := range options.Mocks {
		if options.Mocks[i] == nil {
			continue
		}
		mocks = append(mocks, options.Mocks[i])
	}

	server := &mocker{
		t:                  t,
		timeout:            options.Timeout,
		verbose:            options.Verbose,
		debugTraffic:       getDebugMode(), // options.DebugTraffic,
		doNotOverrideCreds: options.DoNotOverrideCreds,
		doNotFailUnhandled: options.DoNotFailUnhandledRequests,
		mocks:              mocks,
		usingAwsConfig:     options.ReturnAwsConfig,
	}
	server.Start()

	info := &MockerInfo{
		ProxyURL: server.httpServer.URL,
	}

	if options.ReturnAwsConfig {
		cfg := server.buildAwsConfig()
		info.AwsConfig = &cfg
	}

	return info
}