Examples with RadixNodeUri com.radixdlt.network.p2p.RadixNodeUri used on opensource projects

Example 1 with RadixNodeUri

use of com.radixdlt.network.p2p.RadixNodeUri in project radixdlt by radixdlt.

the class MockP2PNetwork method createChannel.

void createChannel(int clientPeerIndex, RadixNodeUri serverPeerUri) {
    final var clientPeer = nodes.get(clientPeerIndex);
    final var serverPeerOpt = nodes.stream().filter(p -> p.uri.getHost().equals(serverPeerUri.getHost()) && p.uri.getPort() == serverPeerUri.getPort()).findAny();
    final var clientSocketChannel = mock(SocketChannel.class);
    final var clientChannel = new PeerChannel(clientPeer.injector.getInstance(P2PConfig.class), Addressing.ofNetwork(Network.LOCALNET), 1, "fork1", clientPeer.injector.getInstance(SystemCounters.class), clientPeer.injector.getInstance(Serialization.class), new SecureRandom(), ECKeyOps.fromKeyPair(clientPeer.keyPair), clientPeer.injector.getInstance(new Key<EventDispatcher<PeerEvent>>() {
    }), Optional.of(serverPeerUri), clientSocketChannel, Optional.empty());
    if (serverPeerOpt.isEmpty()) {
        clientChannel.channelActive(null);
        clientChannel.channelInactive(null);
        return;
    }
    final var serverPeer = serverPeerOpt.get();
    final var serverSocketChannel = mock(SocketChannel.class);
    final var serverChannel = new PeerChannel(serverPeer.injector.getInstance(P2PConfig.class), Addressing.ofNetwork(Network.LOCALNET), 1, "fork1", serverPeer.injector.getInstance(SystemCounters.class), serverPeer.injector.getInstance(Serialization.class), new SecureRandom(), ECKeyOps.fromKeyPair(serverPeer.keyPair), serverPeer.injector.getInstance(new Key<EventDispatcher<PeerEvent>>() {
    }), Optional.empty(), serverSocketChannel, Optional.empty());
    when(clientSocketChannel.writeAndFlush(any())).thenAnswer(inv -> {
        final var rawData = inv.getArgument(0);
        serverChannel.channelRead0(null, (ByteBuf) rawData);
        return null;
    });
    when(serverSocketChannel.writeAndFlush(any())).thenAnswer(inv -> {
        final var rawData = inv.getArgument(0);
        clientChannel.channelRead0(null, (ByteBuf) rawData);
        return null;
    });
    when(clientSocketChannel.close()).thenAnswer(inv -> {
        final var mockChannel = mock(ChannelHandlerContext.class);
        when(mockChannel.channel()).thenReturn(mock(Channel.class));
        clientChannel.channelInactive(mockChannel);
        serverChannel.channelInactive(mockChannel);
        return null;
    });
    when(serverSocketChannel.close()).thenAnswer(inv -> {
        final var mockChannel = mock(ChannelHandlerContext.class);
        when(mockChannel.channel()).thenReturn(mock(Channel.class));
        serverChannel.channelInactive(mockChannel);
        clientChannel.channelInactive(mockChannel);
        return null;
    });
    serverChannel.channelActive(null);
    clientChannel.channelActive(null);
}

Example 2 with RadixNodeUri

use of com.radixdlt.network.p2p.RadixNodeUri in project radixdlt by radixdlt.

the class P2PTestNetworkRunner method createInjector.

private static Injector createInjector(MockP2PNetwork p2pNetwork, DeterministicNetwork network, P2PConfig p2pConfig, ECKeyPair nodeKey, RadixNodeUri selfUri, int selfNodeIndex) throws ParseException {
    final var properties = new RuntimeProperties(new JSONObject(), new String[] {});
    return Guice.createInjector(Modules.override(new P2PModule(properties)).with(new AbstractModule() {

        @Override
        protected void configure() {
            bind(TestCounters.class).toInstance(new TestCounters());
            bind(P2PConfig.class).toInstance(p2pConfig);
            bind(RadixNodeUri.class).annotatedWith(Self.class).toInstance(selfUri);
            bind(SystemCounters.class).to(SystemCountersImpl.class).in(Scopes.SINGLETON);
        }

        @Provides
        public PeerOutboundBootstrap peerOutboundBootstrap(TestCounters testCounters) {
            return uri -> {
                testCounters.outboundChannelsBootstrapped += 1;
                p2pNetwork.createChannel(selfNodeIndex, uri);
            };
        }
    }), new PeerDiscoveryModule(), new PeerLivenessMonitorModule(), new DispatcherModule(), new AbstractModule() {

        @Override
        protected void configure() {
            final var dbDir = new TemporaryFolder();
            try {
                dbDir.create();
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            bindConstant().annotatedWith(NetworkId.class).to(Network.LOCALNET.getId());
            bind(Addressing.class).toInstance(Addressing.ofNetwork(Network.LOCALNET));
            bindConstant().annotatedWith(DatabaseLocation.class).to(dbDir.getRoot().getAbsolutePath());
            bindConstant().annotatedWith(DatabaseCacheSize.class).to(100_000L);
            bind(ECKeyPair.class).annotatedWith(Self.class).toInstance(nodeKey);
            bind(ECPublicKey.class).annotatedWith(Self.class).toInstance(nodeKey.getPublicKey());
            bind(BFTNode.class).annotatedWith(Self.class).toInstance(BFTNode.create(nodeKey.getPublicKey()));
            bind(String.class).annotatedWith(Self.class).toInstance(Addressing.ofNetwork(Network.LOCALNET).forValidators().of(nodeKey.getPublicKey()).substring(0, 10));
            bind(ECKeyOps.class).toInstance(ECKeyOps.fromKeyPair(nodeKey));
            bind(Environment.class).toInstance(network.createSender(BFTNode.create(nodeKey.getPublicKey())));
            bind(RuntimeProperties.class).toInstance(properties);
            bind(Serialization.class).toInstance(DefaultSerialization.getInstance());
            bind(DeterministicProcessor.class);
            Multibinder.newSetBinder(binder(), StartProcessorOnRunner.class);
            bind(ForkConfig.class).annotatedWith(NewestForkConfig.class).toInstance(new FixedEpochForkConfig("genesis", null, 0L));
        }
    });
}