public void Can_Validate_PeerId_Format()
{
_peerIdValidator.ValidatePeerIdFormat(_validPeerId);
_output.WriteLine(string.Join(" ", _validPeerId.ToByteArray()));
var fieldsInBytes = new[]
{
_validPeerId.Ip.ToByteArray(), BitConverter.GetBytes(_validPeerId.Port),
_validPeerId.PublicKey.ToByteArray()
};
_output.WriteLine(string.Join(" ", fieldsInBytes.SelectMany(b => b)));
}
protected override void ChannelRead0(IChannelHandlerContext ctx, ProtocolMessage msg)
{
Logger.Verbose("Received {msg}", msg);
if (_peerIdValidator.ValidatePeerIdFormat(msg.PeerId))
{
ctx.FireChannelRead(msg);
}
}
PeerServerChannelFactory_Pipeline_Should_Produce_Response_Object_PeerClientChannelFactory_Can_Process()
{
var recipient = PeerIdHelper.GetPeerId("recipient");
var sender = PeerIdHelper.GetPeerId("sender");
var signature = Substitute.For <ISignature>();
_peerIdValidator.ValidatePeerIdFormat(Arg.Any <PeerId>()).Returns(true);
_serverKeySigner.CryptoContext.SignatureLength.Returns(64);
_serverKeySigner.Sign(Arg.Any <byte[]>(), default).ReturnsForAnyArgs(signature);
var correlationId = CorrelationId.GenerateCorrelationId();
var protocolMessage = new PingResponse().ToProtocolMessage(sender, correlationId);
var dto = new MessageDto(
protocolMessage,
recipient
);
_clientCorrelationManager.TryMatchResponse(Arg.Any <ProtocolMessage>()).Returns(true);
_serverChannel.WriteOutbound(dto);
var sentBytes = _serverChannel.ReadOutbound <DatagramPacket>();
_serverCorrelationManager.DidNotReceiveWithAnyArgs()
.AddPendingRequest(Arg.Any <CorrelatableMessage <ProtocolMessage> >());
_serverKeySigner.ReceivedWithAnyArgs(1).Sign(Arg.Any <byte[]>(), default);
_clientKeySigner.Verify(
Arg.Any <ISignature>(),
Arg.Any <byte[]>(),
default)
.ReturnsForAnyArgs(true);
var observer = new ProtocolMessageObserver(0, Substitute.For <ILogger>());
var messageStream = _clientFactory.InheritedHandlers.OfType <ObservableServiceHandler>().Single()
.MessageStream;
using (messageStream.Subscribe(observer))
{
_clientChannel.WriteInbound(sentBytes);
_clientChannel.ReadInbound <ProtocolMessage>();
_clientCorrelationManager.ReceivedWithAnyArgs(1).TryMatchResponse(Arg.Any <ProtocolMessage>());
_clientKeySigner.ReceivedWithAnyArgs(1).Verify(null, null, null);
_testScheduler.Start();
observer.Received.Count.Should().Be(1);
observer.Received.Single().Payload.CorrelationId.ToCorrelationId().Id.Should().Be(correlationId.Id);
}
await _serverChannel.DisconnectAsync();
await _clientChannel.DisconnectAsync();
}
RpcClientChannelFactory_Pipeline_Should_Produce_Request_Object_RpcServerChannelFactory_Can_Process_Into_Observable()
{
var recipient = PeerIdHelper.GetPeerId("recipient");
var sender = PeerIdHelper.GetPeerId("sender");
var signature = Substitute.For <ISignature>();
signature.SignatureBytes.Returns(ByteUtil.GenerateRandomByteArray(new FfiWrapper().SignatureLength));
_peerIdValidator.ValidatePeerIdFormat(Arg.Any <PeerId>()).Returns(true);
_clientKeySigner.Sign(Arg.Any <byte[]>(), default).ReturnsForAnyArgs(signature);
var correlationId = CorrelationId.GenerateCorrelationId();
var protocolMessage = new GetPeerCountRequest().ToProtocolMessage(sender, correlationId);
var dto = new MessageDto(
protocolMessage,
recipient
);
_clientChannel.WriteOutbound(dto);
var sentBytes = _clientChannel.ReadOutbound <IByteBuffer>();
// obviously
sentBytes.Should().BeAssignableTo <IByteBuffer>();
_clientCorrelationManager.ReceivedWithAnyArgs(1).AddPendingRequest(
Arg.Is <CorrelatableMessage <ProtocolMessage> >(c =>
c.Content.CorrelationId.ToCorrelationId().Equals(correlationId)));
_clientKeySigner.ReceivedWithAnyArgs(1).Sign(Arg.Is(signature.SignatureBytes), default);
_serverKeySigner.Verify(
Arg.Any <ISignature>(),
Arg.Any <byte[]>(),
default
)
.ReturnsForAnyArgs(true);
_authenticationStrategy.Authenticate(Arg.Any <PeerId>()).Returns(true);
var observer = new ProtocolMessageObserver(0, Substitute.For <ILogger>());
var messageStream = _serverFactory.InheritedHandlers.OfType <ObservableServiceHandler>().Single()
.MessageStream;
using (messageStream.Subscribe(observer))
{
_serverChannel.WriteInbound(sentBytes);
_serverCorrelationManager.DidNotReceiveWithAnyArgs().TryMatchResponse(protocolMessage);
_serverKeySigner.ReceivedWithAnyArgs(1).Verify(null, null, null);
_testScheduler.Start();
observer.Received.Count.Should().Be(1);
observer.Received.Single().Payload.CorrelationId.ToCorrelationId().Id.Should().Be(correlationId.Id);
}
await _serverChannel.DisconnectAsync();
await _clientChannel.DisconnectAsync();
}
public void Can_Stop_Next_Pipeline_On_Invalid_Peer()
{
_peerIdValidator.ValidatePeerIdFormat(Arg.Any <PeerId>()).Returns(false);
_peerIdValidationHandler.ChannelRead(_fakeContext, _message);
_fakeContext.DidNotReceiveWithAnyArgs().FireChannelRead(Arg.Any <object>());
}
