WIP

Author: simone-dimaulo-hs

sqs/sqs.go

@@ -18,6 +18,7 @@ type sqsClient interface {
 	SendMessage(input *sqs.SendMessageInput) (*sqs.SendMessageOutput, error)
 	ReceiveMessage(input *sqs.ReceiveMessageInput) (*sqs.ReceiveMessageOutput, error)
 	ListQueues(input *sqs.ListQueuesInput) (*sqs.ListQueuesOutput, error)
+	GetQueueAttributes(input *sqs.GetQueueAttributesInput) (*sqs.GetQueueAttributesOutput, error)
 }
 
 type Driver struct {
@@ -51,7 +52,7 @@ func New(options ...Option) (*Driver, error) {
 	}
 
 	if driver.testConnectionOnStartup {
-		if err := driver.testConnection(); err != nil {
+		if err := driver.testConnection(driver.url); err != nil {
 			return nil, err
 		}
 	}
@@ -88,7 +89,10 @@ func createClient(queueUrl string, region string, clientCredentials *credentials
 	return sqs.New(session.Must(session.NewSessionWithOptions(options))), nil
 }
 
-func (d *Driver) testConnection() error {
-	_, err := d.sqsClient.ListQueues(&sqs.ListQueuesInput{})
+func (d *Driver) testConnection(queueURL string) error {
+	_, err := d.sqsClient.GetQueueAttributes(&sqs.GetQueueAttributesInput{
+		AttributeNames: []*string{aws.String("All")},
+		QueueUrl:       aws.String(queueURL),
+	})
 	return err
 }