public void SyncNextAnnouncement_WithSyncingAndEmptyAnnoucements_EndsSync()
{
Mock <IMessageReader> reader = new Mock <IMessageReader>();
Mock <IMessageWriter> messageWritter = new Mock <IMessageWriter>();
ECKeyPair kp = new KeyPairGenerator().Generate();
Peer p = new Peer(new TcpClient(), reader.Object, messageWritter.Object, 1234, kp, 0);
var(_, handshake) = NetworkTestHelpers.CreateKeyPairAndHandshake(1235);
p.AuthentifyWith(handshake); // set "other peer as authentified"
var annoucement = new Announce {
Height = 10, Id = ByteString.CopyFromUtf8("FakeHash")
};
p.StashAnnouncement(annoucement);
// trigger sync
p.SyncNextAnnouncement();
// end sync
var movedToNext = p.SyncNextAnnouncement();
// Check that sync ended
Assert.False(movedToNext);
Assert.Null(p.SyncedAnnouncement);
Assert.False(p.AnyStashed);
}
public void Start_AuthentificationNoTimout_ShouldThrowEvent()
{
int localPort = 1234;
int remotePort = 1235;
Mock <IMessageReader> reader = new Mock <IMessageReader>();
Mock <IMessageWriter> messageWritter = new Mock <IMessageWriter>();
ECKeyPair key = new KeyPairGenerator().Generate();
Peer p = new Peer(new TcpClient(), reader.Object, messageWritter.Object, localPort, key, 0);
p.AuthTimeout = 100;
AuthFinishedArgs authFinishedArgs = null;
p.AuthFinished += (sender, args) => {
authFinishedArgs = args as AuthFinishedArgs;
};
p.Start();
var(_, handshake) = NetworkTestHelpers.CreateKeyPairAndHandshake(remotePort);
p.AuthentifyWith(handshake);
Task.Delay(200).Wait();
Assert.Null(authFinishedArgs);
Assert.True(p.IsAuthentified);
}
public void PayloadToPartials_ExactlyTwice_ReturnsTwoPartial()
{
MessageWriter writer = new MessageWriter(null);
writer.MaxOutboundPacketSize = 10;
byte[] bytes = NetworkTestHelpers.GetRandomBytes(20);
var partials = writer.PayloadToPartials(0, bytes, writer.MaxOutboundPacketSize);
Assert.NotNull(partials);
Assert.Equal(partials.Count, 2);
// First should not be end, second should be
Assert.Equal(partials[0].IsEnd, false);
Assert.Equal(partials[1].IsEnd, true);
// Validate lengths
Assert.Equal(partials[0].Data.Length, 10);
Assert.Equal(partials[1].Data.Length, 10);
// validate positions
Assert.Equal(partials[0].Position, 0);
Assert.Equal(partials[1].Position, 1);
}
public void Start_Disposed_ThrowsInvalidOperationException()
{
int port = 1234;
Peer p = new Peer(new TcpClient(), null, null, port, null, 0);
var(_, handshake) = NetworkTestHelpers.CreateKeyPairAndHandshake(port);
p.AuthentifyWith(handshake);
var ex = Assert.Throws <InvalidOperationException>(() => p.Start());
Assert.Equal("Cannot start an already authentified peer.", ex.Message);
}
public void SyncNextAnnouncement_WithAnnouncement_TriggersSync()
{
Mock <IMessageReader> reader = new Mock <IMessageReader>();
Mock <IMessageWriter> messageWritter = new Mock <IMessageWriter>();
ECKeyPair kp = new KeyPairGenerator().Generate();
Peer p = new Peer(new TcpClient(), reader.Object, messageWritter.Object, 1234, kp, 0);
var(_, handshake) = NetworkTestHelpers.CreateKeyPairAndHandshake(1235);
p.AuthentifyWith(handshake); // set "other peer as authentified"
BlockRequest req = null;
messageWritter
.Setup(w => w.EnqueueMessage(It.Is <Message>(m => m.Type == (int)AElfProtocolMsgType.RequestBlock), It.IsAny <Action <Message> >()))
.Callback <Message, Action <Message> >((m, a) =>
{
if (m != null)
{
req = BlockRequest.Parser.ParseFrom(m.Payload);
}
});
var annoucement = new Announce {
Height = 10, Id = ByteString.CopyFromUtf8("FakeHash")
};
p.StashAnnouncement(annoucement);
// trigger sync
var movedToNext = p.SyncNextAnnouncement();
Assert.NotNull(req?.Id);
// Should effectively request the block
Assert.True(movedToNext);
Assert.Equal(req.Height, 10);
Assert.Equal(req.Id.ToStringUtf8(), "FakeHash");
Assert.Equal(p.SyncedAnnouncement, annoucement);
Assert.False(p.AnyStashed);
Assert.Equal(p.SyncTarget, 0);
Assert.Equal(p.CurrentlyRequestedHeight, 0);
}
public void SyncToHeight_WithValidTarget_TriggersSync()
{
// Mutiple things to verify: (1) state is correct, (2) a request is created for startHeight and (3) that
// the corresponding message is sent to the peer.
int startHeight = 2;
int targetHeight = 3;
int peerPort = 1234;
Mock <IMessageReader> reader = new Mock <IMessageReader>();
Mock <IMessageWriter> messageWritter = new Mock <IMessageWriter>();
ECKeyPair kp = new KeyPairGenerator().Generate();
Peer p = new Peer(new TcpClient(), reader.Object, messageWritter.Object, peerPort, kp, 0);
var(_, handshake) = NetworkTestHelpers.CreateKeyPairAndHandshake(1235);
p.AuthentifyWith(handshake); // set "other peer as authentified"
BlockRequest req = null;
messageWritter
.Setup(w => w.EnqueueMessage(It.Is <Message>(m => m.Type == (int)AElfProtocolMsgType.RequestBlock), It.IsAny <Action <Message> >()))
.Callback <Message, Action <Message> >((m, a) =>
{
if (m != null)
{
req = BlockRequest.Parser.ParseFrom(m.Payload);
}
});
p.SyncToHeight(startHeight, targetHeight);
// Check state
Assert.Equal(p.SyncTarget, targetHeight);
Assert.Equal(p.CurrentlyRequestedHeight, startHeight);
// Check block request
Assert.NotEmpty(p.BlockRequests);
Assert.NotEmpty(p.BlockRequests);
// Check request sent
Assert.NotNull(req);
Assert.Equal(req.Height, startHeight);
}
public void PayloadToPartials_OnLimit_ReturnsOnePartial()
{
MessageWriter writer = new MessageWriter(null);
writer.MaxOutboundPacketSize = 10;
byte[] bytes = NetworkTestHelpers.GetRandomBytes(10);
var partials = writer.PayloadToPartials(0, bytes, writer.MaxOutboundPacketSize);
Assert.NotNull(partials);
Assert.Equal(partials.Count, 1);
Assert.Equal(partials[0].IsEnd, true);
Assert.Equal(partials[0].Data.Length, 10);
Assert.Equal(partials[0].Position, 0);
}
public void SyncNextHistory_ReceiveTarget_EndsSync()
{
int startHeight = 2;
int targetHeight = 2;
int peerPort = 1234;
Mock <IMessageReader> reader = new Mock <IMessageReader>();
Mock <IMessageWriter> messageWritter = new Mock <IMessageWriter>();
ECKeyPair kp = new KeyPairGenerator().Generate();
Peer p = new Peer(new TcpClient(), reader.Object, messageWritter.Object, peerPort, kp, 0);
var(_, handshake) = NetworkTestHelpers.CreateKeyPairAndHandshake(1235);
p.AuthentifyWith(handshake); // set "other peer as authentified"
p.SyncToHeight(startHeight, targetHeight);
var movedToNext = p.SyncNextHistory();
Assert.False(movedToNext);
Assert.Equal(p.SyncTarget, 0);
Assert.Equal(p.CurrentlyRequestedHeight, 0);
}
