fixing ut

linkedin · Aug 28, 2024 · c3d21a4 · c3d21a4
1 parent c2d281a
commit c3d21a4
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/clients/src/test/java/org/apache/kafka/clients/NetworkClientTest.java b/clients/src/test/java/org/apache/kafka/clients/NetworkClientTest.java
@@ -633,8 +633,13 @@ public void testResolveBootstrapInLeastLoadedNode() {
 
     @Test
     public void noLeastLoadedNode() {
-        clusterClient.ready(null, time.milliseconds());
-        assertNotEquals(null, clusterClient.leastLoadedNode(time.milliseconds()));
+        NetworkClient nc = new NetworkClient(selector, clusterMetadataUpdater, "mock-cluster-md", Integer.MAX_VALUE,
+            0, 0, 64 * 1024, 64 * 1024,
+            defaultRequestTimeoutMs, connectionSetupTimeoutMsTest, connectionSetupTimeoutMaxMsTest, time, true, new ApiVersions(), new LogContext(),
+            LeastLoadedNodeAlgorithm.VANILLA, new ArrayList<>());
+        nc.ready(node, time.milliseconds());
+        assertFalse(client.isReady(node, time.milliseconds()));
+        assertEquals(null, nc.leastLoadedNode(time.milliseconds()));
     }
 
     @Test