@Test public void shouldReportWhenPeerIsConnected() { registry.registerConnection(connection1); assertThat(registry.isAlreadyConnected(PEER1_ID)).isTrue(); } }
@Test public void shouldUnregisterConnections() { registry.registerConnection(connection1); registry.registerConnection(connection2); registry.onDisconnect(connection1, DisconnectReason.TCP_SUBSYSTEM_ERROR, false); assertThat(registry.getPeerConnections()).containsOnly(connection2); assertThat(registry.size()).isEqualTo(1); }
@Test public void shouldRegisterConnections() { registry.registerConnection(connection1); assertThat(registry.getPeerConnections()).containsOnly(connection1); assertThat(registry.size()).isEqualTo(1); registry.registerConnection(connection2); assertThat(registry.getPeerConnections()).containsOnly(connection1, connection2); assertThat(registry.size()).isEqualTo(2); }
@Test public void checkMaintainedConnectionPeersDoesntReconnectConnectedPeers() { final NettyP2PNetwork network = spy(mockNettyP2PNetwork()); final Peer peer = mockPeer(); verify(network, never()).connect(peer); assertThat(network.addMaintainConnectionPeer(peer)).isTrue(); verify(network, times(1)).connect(peer); { final CompletableFuture<PeerConnection> connection; connection = network.pendingConnections.remove(peer); assertThat(connection).isNotNull(); assertThat(connection.cancel(true)).isTrue(); } { final PeerConnection peerConnection = mockPeerConnection(peer.getId()); network.connections.registerConnection(peerConnection); network.checkMaintainedConnectionPeers(); verify(network, times(1)).connect(peer); } }