tester.go: Add coverage on ChangeMessageVisibility

Author: sundbry

cmd/smoothmq/tester/tester.go

@@ -39,7 +39,7 @@ func Run(c smoothCfg.TesterCommand) {
 
 	var wg sync.WaitGroup
 
-	var ch chan int
+	ch := make(chan int)
 
 	queueUrl := createQueue(sqsClient)
 
@@ -71,6 +71,10 @@ func Run(c smoothCfg.TesterCommand) {
 				pct = float64(receivedMessages) / float64(sentMessages)
 			}
 			log.Info().Msg(fmt.Sprintf("sent: %d, received: %d, pct: %f", sentMessages, receivedMessages, pct))
+			if (sentMessages > 0 && sentMessages == receivedMessages) {
+				close(ch)
+				return
+			}
 			time.Sleep(1 * time.Second)
 		}
 	}()
@@ -79,12 +83,15 @@ func Run(c smoothCfg.TesterCommand) {
 
 	if c.Senders > 0 {
 		log.Info().Msg("All messages sent")
-		if c.Receivers == 0 {
-			os.Exit(0)
-		}
 	}
 
 	<-ch
+
+	// Test changing message visibility timeout
+	queueUrl2 := createQueue(sqsClient)
+	log.Info().Msgf("queue2: %s", queueUrl)
+	testChangeMessageVisibility(sqsClient, queueUrl2)
+	os.Exit(0)
 }
 
 func GenerateRandomString(n int) string {
@@ -166,6 +173,47 @@ func sendMessage(client *sqs.Client, queueUrl string, goroutineID, requestID, ba
 	// time.Sleep(100 * time.Millisecond)
 }
 
+func testChangeMessageVisibility(client *sqs.Client, queueUrl string) {
+	log.Info().Msg("Testing ChangeMessageVisibility")
+	// Send a message
+	messageBody := "Test message for visibility timeout"
+	sendMessageInput := &sqs.SendMessageInput{
+		QueueUrl:    &queueUrl,
+		MessageBody: &messageBody,
+	}
+	sendMessageOutput, err := client.SendMessage(context.TODO(), sendMessageInput)
+	if err != nil {
+		log.Fatal().Msgf("failed to send message, %v", err)
+	}
+
+	for {
+		recvMsg := &sqs.ReceiveMessageInput{
+			QueueUrl:            aws.String(queueUrl),
+			MaxNumberOfMessages: 1,
+		}
+		msgs, err := client.ReceiveMessage(context.TODO(), recvMsg)
+		if err != nil {
+			log.Error().Err(err).Send()
+		}
+		if len(msgs.Messages) == 1 {
+			break
+		}
+	}
+
+	// Change the visibility timeout
+	changeVisibilityInput := &sqs.ChangeMessageVisibilityInput{
+		QueueUrl:          &queueUrl,
+		ReceiptHandle:     sendMessageOutput.MessageId,
+		VisibilityTimeout: 60,
+	}
+	_, err = client.ChangeMessageVisibility(context.TODO(), changeVisibilityInput)
+	if err != nil {
+		log.Fatal().Msgf("failed to change message visibility timeout, %v", err)
+	}
+
+	log.Info().Msg("Successfully changed message visibility timeout")
+}
+
 func receiveMessage(client *sqs.Client, queueUrl string, goroutineID int) int {
 	i := &sqs.ReceiveMessageInput{
 		QueueUrl:            aws.String(queueUrl),