Examples with Recipient com.quorum.tessera.partyinfo.node.Recipient used on opensource projects

Example 16 with Recipient

use of com.quorum.tessera.partyinfo.node.Recipient in project tessera by ConsenSys.

the class SyncPollerTest method localUrlIsExcludedFromPoll.

@Test
public void localUrlIsExcludedFromPoll() {
    final String targetUrl = "localurl.com";
    final String syncableUrl = "syncable.com";
    final com.quorum.tessera.partyinfo.node.Recipient localKey = com.quorum.tessera.partyinfo.node.Recipient.of(mock(PublicKey.class), targetUrl);
    final com.quorum.tessera.partyinfo.node.Recipient anotherKey = com.quorum.tessera.partyinfo.node.Recipient.of(mock(PublicKey.class), syncableUrl);
    final Party syncableParty = new Party(syncableUrl);
    final Set<Recipient> recipients = Set.of(localKey, anotherKey);
    NodeInfo nodeInfo = NodeInfo.Builder.create().withUrl("localurl.com").withRecipients(recipients).build();
    // partyStore.store(URI.create(syncableUrl));
    when(partyInfoService.getCurrent()).thenReturn(nodeInfo);
    doReturn(Optional.empty()).when(resendPartyStore).getNextParty();
    syncPoller.run();
    verify(resendPartyStore, times(1)).addUnseenParties(Set.of(syncableParty));
    verify(resendPartyStore, times(1)).getNextParty();
}

Example 17 with Recipient

use of com.quorum.tessera.partyinfo.node.Recipient in project tessera by ConsenSys.

the class AutoDiscoveryTest method onUpdateIgnoresKeysThatAreNotOwnedBySender.

@Test
public void onUpdateIgnoresKeysThatAreNotOwnedBySender() {
    String uri = "http://mynode.com";
    PublicKey key = mock(PublicKey.class);
    Recipient recipient = Recipient.of(key, uri);
    Recipient other = Recipient.of(mock(PublicKey.class), "http://othernode.com");
    List<Recipient> recipients = List.of(recipient, other);
    NodeInfo nodeInfo = NodeInfo.Builder.create().withUrl(uri).withRecipients(recipients).withSupportedApiVersions(List.of("Two", "Fifty")).build();
    List<ActiveNode> storedNodes = new ArrayList<>();
    doAnswer(invocation -> {
        storedNodes.add(invocation.getArgument(0));
        return null;
    }).when(networkStore).store(any(ActiveNode.class));
    discovery.onUpdate(nodeInfo);
    assertThat(storedNodes).hasSize(1);
    ActiveNode result = storedNodes.iterator().next();
    assertThat(result.getUri()).isEqualTo(NodeUri.create(uri));
    assertThat(result.getKeys()).containsExactly(key);
    assertThat(result.getSupportedVersions()).containsExactlyInAnyOrder("Two", "Fifty");
    verify(networkStore).store(any(ActiveNode.class));
}

Example 18 with Recipient

use of com.quorum.tessera.partyinfo.node.Recipient in project tessera by ConsenSys.

the class DiscoveryHelperTest method buildAllNodeInfosFilteredOutOwn.

@Test
public void buildAllNodeInfosFilteredOutOwn() {
    when(runtimeContext.getP2pServerUri()).thenReturn(URI.create("http://node1.com"));
    final ActiveNode node1 = ActiveNode.Builder.create().withUri(NodeUri.create("http://node1.com")).withKeys(List.of(PublicKey.from("key1".getBytes()))).withSupportedVersions(List.of("v1")).build();
    final ActiveNode node2 = ActiveNode.Builder.create().withUri(NodeUri.create("http://node2.com")).withKeys(List.of(PublicKey.from("key2".getBytes()))).withSupportedVersions(List.of("v2")).build();
    when(networkStore.getActiveNodes()).thenReturn(Stream.of(node1, node2));
    final Set<NodeInfo> nodeInfos = discoveryHelper.buildRemoteNodeInfos();
    assertThat(nodeInfos).hasSize(1);
    Set<ActiveNode> activeNodes = nodeInfos.stream().map(nodeInfo -> ActiveNode.Builder.create().withUri(NodeUri.create(nodeInfo.getUrl())).withKeys(nodeInfo.getRecipients().stream().map(Recipient::getKey).collect(Collectors.toSet())).withSupportedVersions(nodeInfo.supportedApiVersions()).build()).collect(Collectors.toSet());
    assertThat(activeNodes).containsExactlyInAnyOrder(node2);
    verify(networkStore).getActiveNodes();
    verify(runtimeContext).getP2pServerUri();
    mockedRuntimeContext.verify(RuntimeContext::getInstance);
}

Example 19 with Recipient

use of com.quorum.tessera.partyinfo.node.Recipient in project tessera by ConsenSys.

the class EncodedPayloadPublisherTest method noOutstandingPayload.

@Test
public void noOutstandingPayload() {
    BatchWorkflowContext batchWorkflowContext = new BatchWorkflowContext();
    batchWorkflowContext.setBatchSize(100);
    batchWorkflowContext.setExpectedTotal(4L);
    EncodedPayload encodedPayload = mock(EncodedPayload.class);
    batchWorkflowContext.setEncodedPayload(encodedPayload);
    PublicKey recipientKey = mock(PublicKey.class);
    Recipient recipient = mock(Recipient.class);
    when(recipient.getUrl()).thenReturn("http://junit.com");
    batchWorkflowContext.setRecipientKey(recipientKey);
    batchWorkflowContext.setRecipient(recipient);
    encodedPayloadPublisher.checkOutstandingPayloads(batchWorkflowContext);
    assertThat(encodedPayloadPublisher.getPublishedCount()).isEqualTo(0);
}

Example 20 with Recipient

use of com.quorum.tessera.partyinfo.node.Recipient in project tessera by ConsenSys.

the class EncodedPayloadPublisherTest method flattenedPayloadsOverBatchSizeSplitsToSublists.

@Test
public void flattenedPayloadsOverBatchSizeSplitsToSublists() {
    final int numberOfMessagesToSend = 10;
    PublicKey recipientKey = mock(PublicKey.class);
    Recipient recipient = Recipient.of(recipientKey, "http://junit.com");
    EncodedPayload encodedPayloadFirst = mock(EncodedPayload.class);
    EncodedPayload encodedPayloadSecond = mock(EncodedPayload.class);
    BatchWorkflowContext batchWorkflowContext = new BatchWorkflowContext();
    batchWorkflowContext.setBatchSize(3);
    batchWorkflowContext.setExpectedTotal(10L);
    batchWorkflowContext.setPayloadsToPublish(new LinkedHashSet<>(List.of(encodedPayloadFirst, encodedPayloadSecond)));
    batchWorkflowContext.setRecipientKey(recipientKey);
    batchWorkflowContext.setRecipient(recipient);
    boolean result = IntStream.range(0, numberOfMessagesToSend).allMatch(i -> encodedPayloadPublisher.execute(batchWorkflowContext));
    assertThat(result).isTrue();
    assertThat(encodedPayloadPublisher.getPublishedCount()).isEqualTo(10);
    List<EncodedPayload> batchOne = List.of(encodedPayloadFirst, encodedPayloadSecond, encodedPayloadFirst);
    List<EncodedPayload> batchTwo = List.of(encodedPayloadSecond, encodedPayloadFirst, encodedPayloadSecond);
    List<EncodedPayload> leftovers = List.of(encodedPayloadFirst, encodedPayloadSecond);
    verify(resendBatchPublisher, times(3)).publishBatch(batchOne, "http://junit.com");
    verify(resendBatchPublisher, times(3)).publishBatch(batchTwo, "http://junit.com");
    verify(resendBatchPublisher).publishBatch(leftovers, "http://junit.com");
}