mirror of https://github.com/hyperledger/besu
Move connect decision into protocol layer (#4759)
Move the decision making for connecting or not connecting to peers into the eth layer. Future changes will take advantage if this to improve peering. Signed-off-by: Stefan <stefan.pingel@consensys.net> --------- Signed-off-by: Stefan <stefan.pingel@consensys.net> Signed-off-by: Sally MacFarlane <macfarla.github@gmail.com> Co-authored-by: Sally MacFarlane <macfarla.github@gmail.com>pull/5344/head
Stefan Pingel
2 years ago
committed by
GitHub
parent
067a263374
commit
534a369574
@ -0,0 +1,32 @@ |
||||
/* |
||||
* Copyright contributors to Hyperledger Besu |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on |
||||
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the |
||||
* specific language governing permissions and limitations under the License. |
||||
* |
||||
* SPDX-License-Identifier: Apache-2.0 |
||||
*/ |
||||
package org.hyperledger.besu.ethereum.eth; |
||||
|
||||
import org.hyperledger.besu.ethereum.p2p.peers.DefaultPeer; |
||||
import org.hyperledger.besu.ethereum.p2p.peers.EnodeURLImpl; |
||||
import org.hyperledger.besu.ethereum.p2p.peers.Peer; |
||||
|
||||
import org.apache.tuweni.bytes.Bytes; |
||||
|
||||
public class EthPeerTestUtil { |
||||
|
||||
public static EnodeURLImpl.Builder enodeBuilder() { |
||||
return EnodeURLImpl.builder().ipAddress("127.0.0.1").useDefaultPorts().nodeId(Peer.randomId()); |
||||
} |
||||
|
||||
public static Peer createPeer(final Bytes nodeId) { |
||||
return DefaultPeer.fromEnodeURL(enodeBuilder().nodeId(nodeId).build()); |
||||
} |
||||
} |
||||
@ -1,254 +0,0 @@ |
||||
/* |
||||
* Copyright ConsenSys AG. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on |
||||
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the |
||||
* specific language governing permissions and limitations under the License. |
||||
* |
||||
* SPDX-License-Identifier: Apache-2.0 |
||||
*/ |
||||
package org.hyperledger.besu.ethereum.p2p.rlpx.connections; |
||||
|
||||
import org.hyperledger.besu.ethereum.p2p.peers.Peer; |
||||
import org.hyperledger.besu.ethereum.p2p.rlpx.wire.messages.DisconnectMessage.DisconnectReason; |
||||
|
||||
import java.util.Objects; |
||||
import java.util.concurrent.CompletableFuture; |
||||
|
||||
import org.apache.tuweni.bytes.Bytes; |
||||
|
||||
public abstract class RlpxConnection { |
||||
|
||||
private final long initiatedAt; |
||||
protected final CompletableFuture<PeerConnection> future; |
||||
|
||||
private RlpxConnection(final CompletableFuture<PeerConnection> future) { |
||||
this.future = future; |
||||
this.initiatedAt = System.currentTimeMillis(); |
||||
} |
||||
|
||||
public static RlpxConnection inboundConnection(final PeerConnection peerConnection) { |
||||
return new RemotelyInitiatedRlpxConnection(peerConnection); |
||||
} |
||||
|
||||
public static RlpxConnection outboundConnection( |
||||
final Peer peer, final CompletableFuture<PeerConnection> future) { |
||||
return new LocallyInitiatedRlpxConnection(peer, future); |
||||
} |
||||
|
||||
public abstract Peer getPeer(); |
||||
|
||||
public abstract void disconnect(DisconnectReason reason); |
||||
|
||||
public Bytes getId() { |
||||
return getPeer().getId(); |
||||
} |
||||
|
||||
public abstract PeerConnection getPeerConnection() throws ConnectionNotEstablishedException; |
||||
|
||||
public CompletableFuture<PeerConnection> getFuture() { |
||||
return future; |
||||
} |
||||
|
||||
public abstract boolean isActive(); |
||||
|
||||
public abstract boolean isPending(); |
||||
|
||||
public abstract boolean isFailedOrDisconnected(); |
||||
|
||||
public abstract boolean initiatedRemotely(); |
||||
|
||||
public void subscribeConnectionEstablished( |
||||
final RlpxConnectCallback successCallback, final RlpxConnectFailedCallback failedCallback) { |
||||
future.whenComplete( |
||||
(conn, err) -> { |
||||
if (err != null) { |
||||
failedCallback.onFailure(this); |
||||
} else { |
||||
successCallback.onConnect(this); |
||||
} |
||||
}); |
||||
} |
||||
|
||||
public boolean initiatedLocally() { |
||||
return !initiatedRemotely(); |
||||
} |
||||
|
||||
public long getInitiatedAt() { |
||||
return initiatedAt; |
||||
} |
||||
|
||||
private static class RemotelyInitiatedRlpxConnection extends RlpxConnection { |
||||
|
||||
private final PeerConnection peerConnection; |
||||
|
||||
private RemotelyInitiatedRlpxConnection(final PeerConnection peerConnection) { |
||||
super(CompletableFuture.completedFuture(peerConnection)); |
||||
this.peerConnection = peerConnection; |
||||
} |
||||
|
||||
@Override |
||||
public Peer getPeer() { |
||||
return peerConnection.getPeer(); |
||||
} |
||||
|
||||
@Override |
||||
public void disconnect(final DisconnectReason reason) { |
||||
peerConnection.disconnect(reason); |
||||
} |
||||
|
||||
@Override |
||||
public PeerConnection getPeerConnection() { |
||||
return peerConnection; |
||||
} |
||||
|
||||
@Override |
||||
public boolean isActive() { |
||||
return !peerConnection.isDisconnected(); |
||||
} |
||||
|
||||
@Override |
||||
public boolean isPending() { |
||||
return false; |
||||
} |
||||
|
||||
@Override |
||||
public boolean isFailedOrDisconnected() { |
||||
return peerConnection.isDisconnected(); |
||||
} |
||||
|
||||
@Override |
||||
public boolean initiatedRemotely() { |
||||
return true; |
||||
} |
||||
|
||||
@Override |
||||
public boolean equals(final Object o) { |
||||
if (o == this) { |
||||
return true; |
||||
} |
||||
if (!(o instanceof RemotelyInitiatedRlpxConnection)) { |
||||
return false; |
||||
} |
||||
final RemotelyInitiatedRlpxConnection that = (RemotelyInitiatedRlpxConnection) o; |
||||
return Objects.equals(peerConnection, that.peerConnection); |
||||
} |
||||
|
||||
@Override |
||||
public int hashCode() { |
||||
return Objects.hash(peerConnection); |
||||
} |
||||
|
||||
@Override |
||||
public String toString() { |
||||
return "RemotelyInitiatedRlpxConnection initiatedAt:" |
||||
+ getInitiatedAt() |
||||
+ " to " |
||||
+ peerConnection.getPeer().getId() |
||||
+ " disconnected? " |
||||
+ isFailedOrDisconnected(); |
||||
} |
||||
} |
||||
|
||||
private static class LocallyInitiatedRlpxConnection extends RlpxConnection { |
||||
|
||||
private final Peer peer; |
||||
|
||||
private LocallyInitiatedRlpxConnection( |
||||
final Peer peer, final CompletableFuture<PeerConnection> future) { |
||||
super(future); |
||||
this.peer = peer; |
||||
} |
||||
|
||||
@Override |
||||
public Peer getPeer() { |
||||
return peer; |
||||
} |
||||
|
||||
@Override |
||||
public void disconnect(final DisconnectReason reason) { |
||||
future.thenAccept((conn) -> conn.disconnect(reason)); |
||||
} |
||||
|
||||
@Override |
||||
public PeerConnection getPeerConnection() throws ConnectionNotEstablishedException { |
||||
if (!future.isDone() || future.isCompletedExceptionally()) { |
||||
throw new ConnectionNotEstablishedException( |
||||
"Cannot access PeerConnection before connection is fully established."); |
||||
} |
||||
return future.getNow(null); |
||||
} |
||||
|
||||
@Override |
||||
public boolean isActive() { |
||||
return future.isDone() |
||||
&& !future.isCompletedExceptionally() |
||||
&& !getPeerConnection().isDisconnected(); |
||||
} |
||||
|
||||
@Override |
||||
public boolean isPending() { |
||||
return !future.isDone(); |
||||
} |
||||
|
||||
@Override |
||||
public boolean isFailedOrDisconnected() { |
||||
return future.isCompletedExceptionally() |
||||
|| (future.isDone() && getPeerConnection().isDisconnected()); |
||||
} |
||||
|
||||
@Override |
||||
public boolean initiatedRemotely() { |
||||
return false; |
||||
} |
||||
|
||||
@Override |
||||
public boolean equals(final Object o) { |
||||
if (o == this) { |
||||
return true; |
||||
} |
||||
if (!(o instanceof LocallyInitiatedRlpxConnection)) { |
||||
return false; |
||||
} |
||||
final LocallyInitiatedRlpxConnection that = (LocallyInitiatedRlpxConnection) o; |
||||
return Objects.equals(peer, that.peer) && Objects.equals(future, that.future); |
||||
} |
||||
|
||||
@Override |
||||
public int hashCode() { |
||||
return Objects.hash(peer, future); |
||||
} |
||||
|
||||
@Override |
||||
public String toString() { |
||||
return "LocallyInitiatedRlpxConnection initiatedAt:" |
||||
+ getInitiatedAt() |
||||
+ " to " |
||||
+ getPeer().getId() |
||||
+ " disconnected? " |
||||
+ isFailedOrDisconnected(); |
||||
} |
||||
} |
||||
|
||||
public static class ConnectionNotEstablishedException extends IllegalStateException { |
||||
|
||||
public ConnectionNotEstablishedException(final String message) { |
||||
super(message); |
||||
} |
||||
} |
||||
|
||||
@FunctionalInterface |
||||
public interface RlpxConnectCallback { |
||||
void onConnect(RlpxConnection connection); |
||||
} |
||||
|
||||
@FunctionalInterface |
||||
public interface RlpxConnectFailedCallback { |
||||
void onFailure(RlpxConnection connection); |
||||
} |
||||
} |
||||
@ -0,0 +1,23 @@ |
||||
/* |
||||
* Copyright contributors to Hyperledger Besu |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on |
||||
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the |
||||
* specific language governing permissions and limitations under the License. |
||||
* |
||||
* SPDX-License-Identifier: Apache-2.0 |
||||
*/ |
||||
package org.hyperledger.besu.ethereum.p2p.rlpx.wire; |
||||
|
||||
import org.hyperledger.besu.ethereum.p2p.peers.Peer; |
||||
|
||||
@FunctionalInterface |
||||
public interface ShouldConnectCallback { |
||||
|
||||
boolean shouldConnect(final Peer peer, final boolean incoming); |
||||
} |
||||
@ -1,66 +0,0 @@ |
||||
/* |
||||
* Copyright ConsenSys AG. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on |
||||
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the |
||||
* specific language governing permissions and limitations under the License. |
||||
* |
||||
* SPDX-License-Identifier: Apache-2.0 |
||||
*/ |
||||
package org.hyperledger.besu.ethereum.p2p.config; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
import org.junit.Test; |
||||
|
||||
public class RlpxConfigurationTest { |
||||
|
||||
@Test |
||||
public void getMaxRemotelyInitiatedConnections_remoteLimitsDisabled() { |
||||
final RlpxConfiguration config = |
||||
RlpxConfiguration.create() |
||||
.setFractionRemoteWireConnectionsAllowed(.5f) |
||||
.setLimitRemoteWireConnectionsEnabled(false) |
||||
.setPeerUpperBound(20); |
||||
|
||||
assertThat(config.getMaxRemotelyInitiatedConnections()).isEqualTo(20); |
||||
} |
||||
|
||||
@Test |
||||
public void getMaxRemotelyInitiatedConnections_remoteLimitsEnabled() { |
||||
final RlpxConfiguration config = |
||||
RlpxConfiguration.create() |
||||
.setFractionRemoteWireConnectionsAllowed(.5f) |
||||
.setLimitRemoteWireConnectionsEnabled(true) |
||||
.setPeerUpperBound(20); |
||||
|
||||
assertThat(config.getMaxRemotelyInitiatedConnections()).isEqualTo(10); |
||||
} |
||||
|
||||
@Test |
||||
public void getMaxRemotelyInitiatedConnections_remoteLimitsEnabledWithNonIntegerRatio() { |
||||
final RlpxConfiguration config = |
||||
RlpxConfiguration.create() |
||||
.setFractionRemoteWireConnectionsAllowed(.5f) |
||||
.setLimitRemoteWireConnectionsEnabled(true) |
||||
.setPeerUpperBound(25); |
||||
|
||||
assertThat(config.getMaxRemotelyInitiatedConnections()).isEqualTo(12); |
||||
} |
||||
|
||||
@Test |
||||
public void getMaxRemotelyInitiatedConnections_remoteLimitsEnabledRoundsToZero() { |
||||
final RlpxConfiguration config = |
||||
RlpxConfiguration.create() |
||||
.setFractionRemoteWireConnectionsAllowed(.5f) |
||||
.setLimitRemoteWireConnectionsEnabled(true) |
||||
.setPeerUpperBound(1); |
||||
|
||||
assertThat(config.getMaxRemotelyInitiatedConnections()).isEqualTo(0); |
||||
} |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
@ -1,269 +0,0 @@ |
||||
/* |
||||
* Copyright ConsenSys AG. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on |
||||
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the |
||||
* specific language governing permissions and limitations under the License. |
||||
* |
||||
* SPDX-License-Identifier: Apache-2.0 |
||||
*/ |
||||
package org.hyperledger.besu.ethereum.p2p.rlpx.connections; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
||||
import static org.hyperledger.besu.ethereum.p2p.peers.PeerTestHelper.createPeer; |
||||
import static org.mockito.Mockito.spy; |
||||
import static org.mockito.Mockito.verify; |
||||
|
||||
import org.hyperledger.besu.ethereum.p2p.peers.Peer; |
||||
import org.hyperledger.besu.ethereum.p2p.rlpx.connections.RlpxConnection.ConnectionNotEstablishedException; |
||||
import org.hyperledger.besu.ethereum.p2p.rlpx.wire.messages.DisconnectMessage.DisconnectReason; |
||||
|
||||
import java.util.concurrent.CompletableFuture; |
||||
|
||||
import org.junit.Test; |
||||
|
||||
public class RlpxConnectionTest { |
||||
|
||||
@Test |
||||
public void getPeer_pendingOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
|
||||
assertThat(conn.getPeer()).isEqualTo(peer); |
||||
} |
||||
|
||||
@Test |
||||
public void getPeer_establishedOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
future.complete(peerConnection(peer)); |
||||
|
||||
assertThat(conn.getPeer()).isEqualTo(peer); |
||||
} |
||||
|
||||
@Test |
||||
public void getPeer_inboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
final RlpxConnection conn = RlpxConnection.inboundConnection(peerConnection); |
||||
|
||||
assertThat(conn.getPeer()).isEqualTo(peer); |
||||
} |
||||
|
||||
@Test |
||||
public void disconnect_pendingOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
|
||||
final DisconnectReason reason = DisconnectReason.REQUESTED; |
||||
conn.disconnect(reason); |
||||
assertThat(conn.isFailedOrDisconnected()).isFalse(); |
||||
|
||||
// Resolve future
|
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
future.complete(peerConnection); |
||||
|
||||
// Check disconnect was issued
|
||||
verify(peerConnection).disconnect(reason); |
||||
assertThat(conn.isFailedOrDisconnected()).isTrue(); |
||||
} |
||||
|
||||
@Test |
||||
public void disconnect_activeOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
future.complete(peerConnection); |
||||
|
||||
final DisconnectReason reason = DisconnectReason.REQUESTED; |
||||
conn.disconnect(reason); |
||||
|
||||
// Check disconnect was issued
|
||||
assertThat(conn.isFailedOrDisconnected()).isTrue(); |
||||
verify(peerConnection).disconnect(reason); |
||||
} |
||||
|
||||
@Test |
||||
public void disconnect_failedOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
future.completeExceptionally(new IllegalStateException("whoops")); |
||||
|
||||
assertThat(conn.isFailedOrDisconnected()).isTrue(); |
||||
final DisconnectReason reason = DisconnectReason.REQUESTED; |
||||
conn.disconnect(reason); |
||||
assertThat(conn.isFailedOrDisconnected()).isTrue(); |
||||
} |
||||
|
||||
@Test |
||||
public void disconnect_inboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
final RlpxConnection conn = RlpxConnection.inboundConnection(peerConnection); |
||||
|
||||
assertThat(conn.isFailedOrDisconnected()).isFalse(); |
||||
final DisconnectReason reason = DisconnectReason.REQUESTED; |
||||
conn.disconnect(reason); |
||||
|
||||
// Check disconnect was issued
|
||||
assertThat(conn.isFailedOrDisconnected()).isTrue(); |
||||
verify(peerConnection).disconnect(reason); |
||||
} |
||||
|
||||
@Test |
||||
public void getPeerConnection_pendingOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
|
||||
assertThatThrownBy(conn::getPeerConnection) |
||||
.isInstanceOf(ConnectionNotEstablishedException.class); |
||||
} |
||||
|
||||
@Test |
||||
public void getPeerConnection_activeOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
future.complete(peerConnection); |
||||
|
||||
assertThat(conn.getPeerConnection()).isEqualTo(peerConnection); |
||||
} |
||||
|
||||
@Test |
||||
public void getPeerConnection_failedOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
future.completeExceptionally(new IllegalStateException("whoops")); |
||||
|
||||
assertThatThrownBy(conn::getPeerConnection) |
||||
.isInstanceOf(ConnectionNotEstablishedException.class); |
||||
} |
||||
|
||||
@Test |
||||
public void getPeerConnection_disconnectedOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
future.complete(peerConnection); |
||||
conn.disconnect(DisconnectReason.REQUESTED); |
||||
|
||||
assertThat(conn.getPeerConnection()).isEqualTo(peerConnection); |
||||
} |
||||
|
||||
@Test |
||||
public void getPeerConnection_activeInboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
final RlpxConnection conn = RlpxConnection.inboundConnection(peerConnection); |
||||
|
||||
assertThat(conn.getPeerConnection()).isEqualTo(peerConnection); |
||||
} |
||||
|
||||
@Test |
||||
public void getPeerConnection_disconnectedInboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
final RlpxConnection conn = RlpxConnection.inboundConnection(peerConnection); |
||||
conn.disconnect(DisconnectReason.REQUESTED); |
||||
|
||||
assertThat(conn.getPeerConnection()).isEqualTo(peerConnection); |
||||
} |
||||
|
||||
@Test |
||||
public void checkState_pendingOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
|
||||
assertThat(conn.initiatedRemotely()).isFalse(); |
||||
assertThat(conn.isActive()).isFalse(); |
||||
assertThat(conn.isPending()).isTrue(); |
||||
assertThat(conn.isFailedOrDisconnected()).isFalse(); |
||||
} |
||||
|
||||
@Test |
||||
public void checkState_activeOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
future.complete(peerConnection); |
||||
|
||||
assertThat(conn.initiatedRemotely()).isFalse(); |
||||
assertThat(conn.isActive()).isTrue(); |
||||
assertThat(conn.isPending()).isFalse(); |
||||
assertThat(conn.isFailedOrDisconnected()).isFalse(); |
||||
} |
||||
|
||||
@Test |
||||
public void checkState_failedOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
future.completeExceptionally(new IllegalStateException("whoops")); |
||||
|
||||
assertThat(conn.initiatedRemotely()).isFalse(); |
||||
assertThat(conn.isActive()).isFalse(); |
||||
assertThat(conn.isPending()).isFalse(); |
||||
assertThat(conn.isFailedOrDisconnected()).isTrue(); |
||||
} |
||||
|
||||
@Test |
||||
public void checkState_disconnectedOutboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final CompletableFuture<PeerConnection> future = new CompletableFuture<>(); |
||||
final RlpxConnection conn = RlpxConnection.outboundConnection(peer, future); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
future.complete(peerConnection); |
||||
conn.disconnect(DisconnectReason.UNKNOWN); |
||||
|
||||
assertThat(conn.initiatedRemotely()).isFalse(); |
||||
assertThat(conn.isActive()).isFalse(); |
||||
assertThat(conn.isPending()).isFalse(); |
||||
assertThat(conn.isFailedOrDisconnected()).isTrue(); |
||||
} |
||||
|
||||
@Test |
||||
public void checkState_activeInboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
final RlpxConnection conn = RlpxConnection.inboundConnection(peerConnection); |
||||
|
||||
assertThat(conn.initiatedRemotely()).isTrue(); |
||||
assertThat(conn.isActive()).isTrue(); |
||||
assertThat(conn.isPending()).isFalse(); |
||||
assertThat(conn.isFailedOrDisconnected()).isFalse(); |
||||
} |
||||
|
||||
@Test |
||||
public void checkState_disconnectedInboundConnection() { |
||||
final Peer peer = createPeer(); |
||||
final PeerConnection peerConnection = peerConnection(peer); |
||||
final RlpxConnection conn = RlpxConnection.inboundConnection(peerConnection); |
||||
conn.disconnect(DisconnectReason.UNKNOWN); |
||||
|
||||
assertThat(conn.initiatedRemotely()).isTrue(); |
||||
assertThat(conn.isActive()).isFalse(); |
||||
assertThat(conn.isPending()).isFalse(); |
||||
assertThat(conn.isFailedOrDisconnected()).isTrue(); |
||||
} |
||||
|
||||
private PeerConnection peerConnection(final Peer peer) { |
||||
return spy(MockPeerConnection.create(peer)); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue