Use correct value for WaitTimeSecond

jwhitcraft · jwhitcraft · commit 7bfb70a9cfe1 · 2019-12-17T09:45:42.000-05:00
When the ReceiveMessageInput was create it was using `MaxNubmerOfMessages` as the value for WaitTimeSeconds. Just use the correct value out of the config Struct. (fixes #9) Signed-off-by: Jon Whitcraft <jwhitcraft@mac.com>
diff --git a/worker/worker.go b/worker/worker.go
@@ -82,7 +82,7 @@ func (worker *Worker) Start(ctx context.Context, h Handler) {
 				AttributeNames: []*string{
 					aws.String("All"), // Required
 				},
-				WaitTimeSeconds: aws.Int64(worker.Config.MaxNumberOfMessage),
+				WaitTimeSeconds: aws.Int64(worker.Config.WaitTimeSecond),
 			}
 
 			resp, err := worker.SqsClient.ReceiveMessage(params)
diff --git a/worker/worker_test.go b/worker/worker_test.go
@@ -114,7 +114,7 @@ func setupClientSpies(client *mockedSqsClient) {
 		AttributeNames: []*string{
 			aws.String("All"),
 		},
-		WaitTimeSeconds: aws.Int64(10),
+		WaitTimeSeconds: aws.Int64(20),
 	}
 	client.On("ReceiveMessage", receiveInput).Return()
 

Original file line number	Diff line number	Diff line change
`@@ -82,7 +82,7 @@ func (worker *Worker) Start(ctx context.Context, h Handler) {`
`82`	`82`	`AttributeNames: []*string{`
`83`	`83`	`aws.String("All"), // Required`
`84`	`84`	`},`
`85`		`- WaitTimeSeconds: aws.Int64(worker.Config.MaxNumberOfMessage),`
	`85`	`+ WaitTimeSeconds: aws.Int64(worker.Config.WaitTimeSecond),`
`86`	`86`	`}`
`87`	`87`
`88`	`88`	`resp, err := worker.SqsClient.ReceiveMessage(params)`
Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ func setupClientSpies(client *mockedSqsClient) {`
`114`	`114`	`AttributeNames: []*string{`
`115`	`115`	`aws.String("All"),`
`116`	`116`	`},`
`117`		`- WaitTimeSeconds: aws.Int64(10),`
	`117`	`+ WaitTimeSeconds: aws.Int64(20),`
`118`	`118`	`}`
`119`	`119`	`client.On("ReceiveMessage", receiveInput).Return()`
`120`	`120`