refactor: use gomock

Signed-off-by: Stas D <[email protected]>
trustbloc · Mar 22, 2023 · cbcc576 · cbcc576
1 parent f649946
commit cbcc576
Show file tree

Hide file tree

Showing 4 changed files with 480 additions and 176 deletions.
diff --git a/go.mod b/go.mod
@@ -8,6 +8,7 @@ go 1.18
 
 require (
 	github.com/aws/aws-sdk-go v1.43.9
+	github.com/aws/aws-sdk-go-v2 v1.17.3
 	github.com/aws/aws-sdk-go-v2/service/kms v1.20.0
 	github.com/btcsuite/btcd v0.22.1
 	github.com/golang/mock v1.6.0
@@ -29,7 +30,6 @@ require (
 
 require (
 	github.com/VictoriaMetrics/fastcache v1.5.7 // indirect
-	github.com/aws/aws-sdk-go-v2 v1.17.3 // indirect
 	github.com/aws/aws-sdk-go-v2/internal/configsources v1.1.27 // indirect
 	github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.4.21 // indirect
 	github.com/aws/smithy-go v1.13.5 // indirect

diff --git a/pkg/aws/opts.go b/pkg/aws/opts.go
@@ -12,6 +12,7 @@ import (
 
 type opts struct {
 	keyAliasPrefix string
+	awsClient      awsClient
 }
 
 // NewOpts create new opts populated with environment variable.
@@ -34,3 +35,8 @@ type Opts func(opts *opts)
 func WithKeyAliasPrefix(prefix string) Opts {
 	return func(opts *opts) { opts.keyAliasPrefix = prefix }
 }
+
+// WithAWSClient sets custom aws client
+func WithAWSClient(client awsClient) Opts {
+	return func(opts *opts) { opts.awsClient = client }
+}
diff --git a/pkg/aws/service_mocks.go b/pkg/aws/service_mocks.go