Skip to content

Commit 10645f2

Browse files
committed
Added extra unit tests
Extra unit tests to verify high sequence logic Fixes #342
1 parent 29eef40 commit 10645f2

File tree

2 files changed

+30
-0
lines changed

2 files changed

+30
-0
lines changed

src/test/java/org/logstash/beats/V1BatchTest.java

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,20 @@ public void testCompleteBatchWithSequenceNumbersNotStartingAtOne(){
6161
assertTrue(batch.isComplete());
6262
}
6363

64+
65+
@Test
66+
public void testHighSequence(){
67+
int numberOfEvent = 2;
68+
int startSequenceNumber = new SecureRandom().nextInt(10000);
69+
batch.setBatchSize(numberOfEvent);
70+
71+
for(int i = 1; i <= numberOfEvent; i++) {
72+
batch.addMessage(new Message(startSequenceNumber + i, new HashMap()));
73+
}
74+
75+
assertEquals(startSequenceNumber + numberOfEvent, batch.getHighestSequence());
76+
}
77+
6478
@Test
6579
public void TestCompleteReturnWhenTheNumberOfEventDoesntMatchBatchSize() {
6680
int numberOfEvent = 2;

src/test/java/org/logstash/beats/V2BatchTest.java

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import io.netty.buffer.Unpooled;
88
import org.junit.Test;
99

10+
import java.security.SecureRandom;
1011
import java.util.HashMap;
1112
import java.util.Map;
1213

@@ -75,6 +76,21 @@ public void testBigBatch() {
7576
}
7677
}
7778

79+
@Test
80+
public void testHighSequence(){
81+
V2Batch batch = new V2Batch();
82+
int numberOfEvent = 2;
83+
int startSequenceNumber = new SecureRandom().nextInt(10000);
84+
batch.setBatchSize(numberOfEvent);
85+
ByteBuf content = messageContents();
86+
87+
for(int i = 1; i <= numberOfEvent; i++) {
88+
batch.addMessage(startSequenceNumber + i, content, content.readableBytes());
89+
}
90+
91+
assertEquals(startSequenceNumber + numberOfEvent, batch.getHighestSequence());
92+
}
93+
7894

7995
@Test
8096
public void TestCompleteReturnWhenTheNumberOfEventDoesntMatchBatchSize() {

0 commit comments

Comments
 (0)