Update ReceiveBatchesMockTest to remove delayed response parameter when mock SQS responses are expected ASAP (#5594)

Author: joviegas

services/sqs/src/test/java/software/amazon/awssdk/services/sqs/batchmanager/ReceiveBatchesMockTest.java

@@ -87,13 +87,9 @@ void testTimeoutOccursBeforeSqsResponds() throws Exception {
     void testResponseReceivedBeforeTimeout() throws Exception {
         setupBatchManager();
 
-        // Delays for testing
-        int queueAttributesApiDelay = 5;
-        int receiveMessagesDelay = 5;
-
-        // Set short delays to ensure response before timeout
-        mockQueueAttributes(queueAttributesApiDelay);
-        mockReceiveMessages(receiveMessagesDelay, 2);
+        // No delays set since we mock quick response from SQS.
+        mockQueueAttributes();
+        mockReceiveMessages(2);
 
         CompletableFuture<ReceiveMessageResponse> future = batchManagerReceiveMessage();
         assertThat(future.get(5, TimeUnit.SECONDS).messages()).hasSize(2);
@@ -179,6 +175,22 @@ private void mockReceiveMessages(int delay, int numMessages) {
                                     .withFixedDelay(delay)));
     }
 
+    private void mockReceiveMessages(int numMessages) {
+        stubFor(post(urlEqualTo("/"))
+                    .withHeader("x-amz-target", equalTo("AmazonSQS.ReceiveMessage"))
+                    .willReturn(aResponse()
+                                    .withStatus(200)
+                                    .withBody(generateMessagesJson(numMessages))));
+    }
+
+    private void mockQueueAttributes( ) {
+        stubFor(post(urlEqualTo("/"))
+                    .withHeader("x-amz-target", equalTo("AmazonSQS.GetQueueAttributes"))
+                    .willReturn(aResponse()
+                                    .withStatus(200)
+                                    .withBody(String.format(QUEUE_ATTRIBUTE_RESPONSE, "0", "30"))));
+    }
+
     private CompletableFuture<ReceiveMessageResponse> batchManagerReceiveMessage() {
         return receiveMessageBatchManager.receiveMessage(r -> r.queueUrl("test"));
     }