File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
src/test/java/com/uid2/optout/sqs Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -122,14 +122,19 @@ void testReadWindow_messageLimitExceeded() {
122122 for (int i = 0 ; i < 10 ; i ++) {
123123 batch .add (createMessage (windowStartSeconds + i * 10 ));
124124 }
125+ List <Message > batch2 = new ArrayList <>();
126+ for (int i = 0 ; i < 5 ; i ++) {
127+ batch2 .add (createMessage (windowStartSeconds + i * 10 ));
128+ }
125129
126130 when (mockSqsClient .receiveMessage (any (ReceiveMessageRequest .class )))
127- .thenReturn (ReceiveMessageResponse .builder ().messages (batch ).build ());
131+ .thenReturn (ReceiveMessageResponse .builder ().messages (batch ).build ())
132+ .thenReturn (ReceiveMessageResponse .builder ().messages (batch2 ).build ());
128133
129134 SqsWindowReader .WindowReadResult result = smallLimitReader .readWindow ();
130135
131136 assertEquals (StopReason .MESSAGE_LIMIT_EXCEEDED , result .getStopReason ());
132- assertTrue ( result .getMessages ().size () >= 5 );
137+ assertEquals ( 10 , result .getMessages ().size ());
133138 }
134139
135140 @ Test
You can’t perform that action at this time.
0 commit comments