jsancio commented on code in PR #16079:
URL: https://github.com/apache/kafka/pull/16079#discussion_r1627947377


##########
raft/src/test/java/org/apache/kafka/raft/internals/KafkaRaftMetricsTest.java:
##########
@@ -98,6 +100,7 @@ private VoterSet localStandaloneVoterSet(short kraftVersion) 
{
     @ValueSource(shorts = {0, 1})
     public void shouldRecordVoterQuorumState(short kraftVersion) {
         boolean withDirectoryId = kraftVersion > 0;
+        Set<Integer> voterSet = Utils.mkSet(localId, 1, 2);

Review Comment:
   Let's remove this line and instead use `voters.voterIds()` where needed.



##########
raft/src/test/java/org/apache/kafka/raft/LeaderStateTest.java:
##########
@@ -272,20 +280,109 @@ public void testUpdateHighWatermarkQuorumSizeThree() {
         assertEquals(Optional.empty(), state.highWatermark());
         assertTrue(state.updateReplicaState(node2, 0, new 
LogOffsetMetadata(15L)));
         assertEquals(Optional.of(new LogOffsetMetadata(15L)), 
state.highWatermark());
-        assertFalse(state.updateLocalState(new LogOffsetMetadata(20L)));
+        assertFalse(state.updateLocalState(new LogOffsetMetadata(20L), 
voterSet));
         assertEquals(Optional.of(new LogOffsetMetadata(15L)), 
state.highWatermark());
         assertTrue(state.updateReplicaState(node1, 0, new 
LogOffsetMetadata(20L)));
         assertEquals(Optional.of(new LogOffsetMetadata(20L)), 
state.highWatermark());
         assertFalse(state.updateReplicaState(node2, 0, new 
LogOffsetMetadata(20L)));
         assertEquals(Optional.of(new LogOffsetMetadata(20L)), 
state.highWatermark());
     }
 
+    @Test
+    public void testUpdateHighWatermarkAddingFollowerToVoterStates() {

Review Comment:
   Can we add a test for the decreasing case? For example this change should 
not decrease the HWM.
   ```text
   voter 1 LEO: 15
   voter 2 LEO: 15
   voter 3 LEO: 10
   ```
   
   The HWM is 15.
   Adding voter 4 should not decrease the HWM to 10
   
   ```text
   voter 1 LEO: 15
   voter 2 LEO: 15
   voter 3 LEO: 10
   voter 4 LEO: 5
   ```
   
   The HWM should be 15 even though the naive computation would return 10.



##########
raft/src/test/java/org/apache/kafka/raft/LeaderStateTest.java:
##########
@@ -272,20 +280,109 @@ public void testUpdateHighWatermarkQuorumSizeThree() {
         assertEquals(Optional.empty(), state.highWatermark());
         assertTrue(state.updateReplicaState(node2, 0, new 
LogOffsetMetadata(15L)));
         assertEquals(Optional.of(new LogOffsetMetadata(15L)), 
state.highWatermark());
-        assertFalse(state.updateLocalState(new LogOffsetMetadata(20L)));
+        assertFalse(state.updateLocalState(new LogOffsetMetadata(20L), 
voterSet));
         assertEquals(Optional.of(new LogOffsetMetadata(15L)), 
state.highWatermark());
         assertTrue(state.updateReplicaState(node1, 0, new 
LogOffsetMetadata(20L)));
         assertEquals(Optional.of(new LogOffsetMetadata(20L)), 
state.highWatermark());
         assertFalse(state.updateReplicaState(node2, 0, new 
LogOffsetMetadata(20L)));
         assertEquals(Optional.of(new LogOffsetMetadata(20L)), 
state.highWatermark());
     }
 
+    @Test
+    public void testUpdateHighWatermarkAddingFollowerToVoterStates() {
+        int node1 = 1;
+        int node2 = 2;
+        Set<Integer> originalVoterSet = mkSet(localId, node1);
+        LeaderState<?> state = newLeaderState(originalVoterSet, 10L);
+        assertFalse(state.updateLocalState(new LogOffsetMetadata(15L), 
originalVoterSet));
+        assertFalse(state.updateReplicaState(node1, 0, new 
LogOffsetMetadata(10L)));
+        assertEquals(Optional.empty(), state.highWatermark());

Review Comment:
   The HWM should be 10, no?



##########
raft/src/test/java/org/apache/kafka/raft/LeaderStateTest.java:
##########
@@ -272,20 +280,109 @@ public void testUpdateHighWatermarkQuorumSizeThree() {
         assertEquals(Optional.empty(), state.highWatermark());
         assertTrue(state.updateReplicaState(node2, 0, new 
LogOffsetMetadata(15L)));
         assertEquals(Optional.of(new LogOffsetMetadata(15L)), 
state.highWatermark());
-        assertFalse(state.updateLocalState(new LogOffsetMetadata(20L)));
+        assertFalse(state.updateLocalState(new LogOffsetMetadata(20L), 
voterSet));
         assertEquals(Optional.of(new LogOffsetMetadata(15L)), 
state.highWatermark());
         assertTrue(state.updateReplicaState(node1, 0, new 
LogOffsetMetadata(20L)));
         assertEquals(Optional.of(new LogOffsetMetadata(20L)), 
state.highWatermark());
         assertFalse(state.updateReplicaState(node2, 0, new 
LogOffsetMetadata(20L)));
         assertEquals(Optional.of(new LogOffsetMetadata(20L)), 
state.highWatermark());
     }
 
+    @Test
+    public void testUpdateHighWatermarkAddingFollowerToVoterStates() {
+        int node1 = 1;
+        int node2 = 2;
+        Set<Integer> originalVoterSet = mkSet(localId, node1);
+        LeaderState<?> state = newLeaderState(originalVoterSet, 10L);
+        assertFalse(state.updateLocalState(new LogOffsetMetadata(15L), 
originalVoterSet));
+        assertFalse(state.updateReplicaState(node1, 0, new 
LogOffsetMetadata(10L)));
+        assertEquals(Optional.empty(), state.highWatermark());
+
+        // updating replica state of 2 before it joins voterSet should not 
increase HW to 15L
+        assertFalse(state.updateReplicaState(node2, 0, new 
LogOffsetMetadata(15L)));
+        assertEquals(Optional.empty(), state.highWatermark());

Review Comment:
   The HWM should be 10, no?



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org

Reply via email to