Update sqs constructor to allow custom endpoint (#166)

To allow for mocking of SQS queues a configurable endpoint is required. For SNS this has already been done, this is just adding the same for SQS.
nytimes · Nov 19, 2018 · 7084bb2 · 7084bb2
1 parent fe286d8
commit 7084bb2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pubsub/aws/aws.go b/pubsub/aws/aws.go
@@ -219,6 +219,7 @@ func NewSubscriber(cfg SQSConfig) (pubsub.Subscriber, error) {
 	s.sqs = sqs.New(sess, &aws.Config{
 		Credentials: creds,
 		Region:      &cfg.Region,
+		Endpoint:    cfg.EndpointURL,
 	})
 
 	if len(cfg.QueueURL) == 0 {