Merged in feature/momocks (pull request #150)

Decrease Mocks * feature/nomocks * nonet * assertsaws * assert
2025-05-23 00:20:01 +00:00
parent 36ff8096f3
commit 33d68b7e04
42 changed files with 617 additions and 650 deletions
@@ -7,16 +7,27 @@ import (
 )

 type ReceiveParams struct {
-	QueueURL   string
-	Attributes []string
+	QueueURL          string
+	Attributes        []string
+	VisibilityTimeout int32
+	WaitTimeSeconds   int32
 }

 func (c *QueueConfig) ReceiveFromQueue(ctx context.Context, params *ReceiveParams) (*sqs.ReceiveMessageOutput, error) {
-	return c.QueueClient.ReceiveMessage(ctx, &sqs.ReceiveMessageInput{
+	p := &sqs.ReceiveMessageInput{
 		QueueUrl:              &params.QueueURL,
 		MaxNumberOfMessages:   1,
 		WaitTimeSeconds:       10,
 		VisibilityTimeout:     10,
 		MessageAttributeNames: params.Attributes,
-	})
+	}
+
+	if params.VisibilityTimeout > 0 {
+		p.VisibilityTimeout = params.VisibilityTimeout
+	}
+	if params.WaitTimeSeconds > 0 {
+		p.WaitTimeSeconds = params.WaitTimeSeconds
+	}
+
+	return c.QueueClient.ReceiveMessage(ctx, p)
 }