root
/
kafka
mirror of https://github.com/apache/kafka.git
1
0
Fork 0
Code Issues Actions 2 Packages Projects Releases Wiki Activity

KAFKA-3255: Added unit tests for NetworkClient.connectionDelay(Node node, long now) 

Author: Frank Scholten <frank@frankscholten.nl>

Reviewers: Eno Thereska <eno.thereska@gmail.com>, Ewen Cheslack-Postava <ewen@confluent.io>

Closes #941 from frankscholten/tests/cluster-connection-states
This commit is contained in:
Frank Scholten 2016-02-22 10:13:11 -08:00 committed by Ewen Cheslack-Postava
parent 49fd0ceb09
commit d2527af99a
2 changed files with 29 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clients/src/main/java/org/apache/kafka/clients/ClusterConnectionStates.java
View File

@ -94,15 +94,6 @@ final class ClusterConnectionStates {
return state != null && state.state == ConnectionState.CONNECTED;
}
/**
* Return true iff we are in the process of connecting
* @param id The id of the connection
*/
public boolean isConnecting(String id) {
NodeConnectionState state = nodeState.get(id);
return state != null && state.state == ConnectionState.CONNECTING;
}
/**
* Enter the connected state for the given connection
* @param id The connection identifier

29
clients/src/test/java/org/apache/kafka/clients/NetworkClientTest.java
View File

@ -175,6 +175,35 @@ public class NetworkClientTest {
assertEquals("There should be NO leastloadednode", leastNode, null);
}
@Test
public void testConnectionDelay() {
long now = time.milliseconds();
long delay = client.connectionDelay(node, now);
assertEquals(0, delay);
}
@Test
public void testConnectionDelayConnected() {
awaitReady(client, node);
long now = time.milliseconds();
long delay = client.connectionDelay(node, now);
assertEquals(Long.MAX_VALUE, delay);
}
@Test
public void testConnectionDelayDisconnected() {
awaitReady(client, node);
selector.close(node.idString());
client.poll(requestTimeoutMs, time.milliseconds());
long delay = client.connectionDelay(node, time.milliseconds());
assertEquals(reconnectBackoffMsTest, delay);
}
private static class TestCallbackHandler implements RequestCompletionHandler {
public boolean executed = false;