Skip to content

Commit d9e2641

Browse files
committed
test fix
1 parent 8acd1a8 commit d9e2641

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/test/java/com/uid2/optout/sqs/SqsWindowReaderTest.java

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)