File tree Expand file tree Collapse file tree 2 files changed +13
-3
lines changed
main/java/software/amazon/lambda/powertools/sqs/internal
test/java/software/amazon/lambda/powertools/sqs Expand file tree Collapse file tree 2 files changed +13
-3
lines changed Original file line number Diff line number Diff line change 8
8
import software .amazon .awssdk .services .sqs .SqsClient ;
9
9
import software .amazon .awssdk .services .sqs .model .DeleteMessageBatchRequest ;
10
10
import software .amazon .awssdk .services .sqs .model .DeleteMessageBatchRequestEntry ;
11
+ import software .amazon .awssdk .services .sqs .model .DeleteMessageBatchResponse ;
11
12
import software .amazon .awssdk .services .sqs .model .GetQueueUrlRequest ;
12
13
import software .amazon .lambda .powertools .sqs .SQSBatchProcessingException ;
13
14
@@ -68,15 +69,16 @@ private void deleteSuccessMessage() {
68
69
.build ()).collect (toList ()))
69
70
.build ();
70
71
71
- client .deleteMessageBatch (request );
72
+ DeleteMessageBatchResponse deleteMessageBatchResponse = client .deleteMessageBatch (request );
73
+ LOG .debug (format ("Response from delete request %s" , deleteMessageBatchResponse ));
72
74
}
73
75
}
74
76
75
77
private String url () {
76
78
String [] arnArray = success .get (0 ).getEventSourceArn ().split (":" );
77
79
return client .getQueueUrl (GetQueueUrlRequest .builder ()
78
- .queueOwnerAWSAccountId (arnArray [1 ])
79
- .queueName (arnArray [2 ])
80
+ .queueOwnerAWSAccountId (arnArray [4 ])
81
+ .queueName (arnArray [5 ])
80
82
.build ())
81
83
.queueUrl ();
82
84
}
Original file line number Diff line number Diff line change 9
9
import org .junit .jupiter .api .Test ;
10
10
import org .junit .jupiter .params .ParameterizedTest ;
11
11
import org .junit .jupiter .params .provider .ValueSource ;
12
+ import org .mockito .ArgumentCaptor ;
12
13
import software .amazon .awssdk .services .sqs .SqsClient ;
13
14
import software .amazon .awssdk .services .sqs .model .DeleteMessageBatchRequest ;
14
15
import software .amazon .awssdk .services .sqs .model .GetQueueUrlRequest ;
@@ -107,6 +108,13 @@ void shouldBatchProcessAndDeleteSuccessMessageOnPartialFailures() {
107
108
108
109
verify (interactionClient ).listQueues ();
109
110
verify (sqsClient ).deleteMessageBatch (any (DeleteMessageBatchRequest .class ));
111
+
112
+ ArgumentCaptor <GetQueueUrlRequest > captor = ArgumentCaptor .forClass (GetQueueUrlRequest .class );
113
+ verify (sqsClient ).getQueueUrl (captor .capture ());
114
+
115
+ assertThat (captor .getValue ())
116
+ .hasFieldOrPropertyWithValue ("queueName" , "my-queue" )
117
+ .hasFieldOrPropertyWithValue ("queueOwnerAWSAccountId" , "123456789012" );
110
118
}
111
119
112
120
@ Test
You can’t perform that action at this time.
0 commit comments