KAFKA-14909: add tests

apache · Apr 24, 2023 · 019e276 · 019e276
1 parent 4298fc6
commit 019e276
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/clients/src/test/java/org/apache/kafka/clients/ApiVersionsTest.java b/clients/src/test/java/org/apache/kafka/clients/ApiVersionsTest.java
@@ -24,6 +24,8 @@
 import java.util.Collections;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 public class ApiVersionsTest {
 
@@ -55,4 +57,21 @@ public void testMaxUsableProduceMagicWithRaftController() {
                 .setMaxVersion((short) 2))));
         assertEquals(RecordBatch.CURRENT_MAGIC_VALUE, apiVersions.maxUsableProduceMagic());
     }
+
+    @Test
+    public void testZkMigrationReady() {
+        ApiVersions apiVersions = new ApiVersions();
+
+        apiVersions.update("0", NodeApiVersions.create());
+        assertFalse(apiVersions.isAllNodeZkMigrationReady());
+
+        apiVersions.update("0", new NodeApiVersions(Collections.emptyList(), Collections.emptyList(), true));
+        assertTrue(apiVersions.isAllNodeZkMigrationReady());
+
+        apiVersions.update("1", new NodeApiVersions(Collections.emptyList(), Collections.emptyList(), true));
+        assertTrue(apiVersions.isAllNodeZkMigrationReady());
+
+        apiVersions.update("2", NodeApiVersions.create());
+        assertFalse(apiVersions.isAllNodeZkMigrationReady());
+    }
 }