public async Task Deferred_Messages_Are_Sent_On_Connected(IPAddress ipAddress, int port)
{
var streamMock = new Mock <INetworkStream>();
streamMock.Setup(s => s.ReadAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()))
.Returns(Task.Run(() => 1));
var tcpMock = new Mock <ITcpClient>();
tcpMock.Setup(s => s.Connected).Returns(true);
tcpMock.Setup(s => s.GetStream()).Returns(streamMock.Object);
var msg = new MessageBuilder()
.Code(MessageCode.PeerBrowseRequest)
.Build();
var c = new MessageConnection(MessageConnectionType.Server, ipAddress, port, tcpClient: tcpMock.Object);
await c.WriteMessageAsync(msg);
await c.WriteMessageAsync(msg);
var deferred1 = c.GetProperty <ConcurrentQueue <Message> >("DeferredMessages").Count;
await c.ConnectAsync();
var deferred2 = c.GetProperty <ConcurrentQueue <Message> >("DeferredMessages");
Assert.Equal(2, deferred1);
Assert.Empty(deferred2);
streamMock.Verify(s => s.WriteAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()), Times.Exactly(2));
}
public async Task Codes_Normalized_For_Peer_Connections(string username, IPAddress ipAddress, int port)
{
int code = 0;
var streamMock = new Mock <INetworkStream>();
streamMock.Setup(s => s.ReadAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()))
.Returns(Task.Run(() => 1));
streamMock.Setup(s => s.WriteAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()))
.Callback <byte[], int, int, CancellationToken>((bytes, offset, length, token) => code = BitConverter.ToInt32(bytes, 4))
.Returns(Task.CompletedTask);
var tcpMock = new Mock <ITcpClient>();
tcpMock.Setup(s => s.Connected).Returns(true);
tcpMock.Setup(s => s.GetStream()).Returns(streamMock.Object);
var msg = new MessageBuilder()
.Code(MessageCode.PeerInfoRequest)
.Build();
var c = new MessageConnection(MessageConnectionType.Peer, username, ipAddress, port, tcpClient: tcpMock.Object);
await c.ConnectAsync();
await c.WriteMessageAsync(msg);
Assert.Equal((int)MessageCode.PeerInfoRequest - 20000, code);
streamMock.Verify(s => s.WriteAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()), Times.Once);
}
public async Task WriteMessageAsync_Throws_ConnectionWriteException_When_Stream_WriteAsync_Throws(IPAddress ipAddress, int port)
{
var streamMock = new Mock <INetworkStream>();
streamMock.Setup(s => s.WriteAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()))
.Throws(new IOException());
streamMock.Setup(s => s.ReadAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()))
.Returns(Task.Run(() => 1));
var tcpMock = new Mock <ITcpClient>();
tcpMock.Setup(s => s.Connected).Returns(true);
tcpMock.Setup(s => s.GetStream()).Returns(streamMock.Object);
var msg = new MessageBuilder()
.Code(MessageCode.PeerBrowseRequest)
.Build();
var c = new MessageConnection(MessageConnectionType.Server, ipAddress, port, tcpClient: tcpMock.Object);
await c.ConnectAsync();
var ex = await Record.ExceptionAsync(async() => await c.WriteMessageAsync(msg));
Assert.NotNull(ex);
Assert.IsType <ConnectionWriteException>(ex);
Assert.IsType <IOException>(ex.InnerException);
streamMock.Verify(s => s.WriteAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()), Times.Once);
}
public async Task WriteMessageAsync_Defers_When_Pending(string username, IPAddress ipAddress, int port)
{
var msg = new MessageBuilder()
.Code(MessageCode.PeerBrowseRequest)
.Build();
var c = new MessageConnection(MessageConnectionType.Peer, username, ipAddress, port);
await c.WriteMessageAsync(msg);
var deferred = c.GetProperty <ConcurrentQueue <Message> >("DeferredMessages");
Assert.Single(deferred);
}
public async Task WriteMessageAsync_Throws_InvalidOperationException_When_Disconnecting(string username, IPAddress ipAddress, int port)
{
var msg = new MessageBuilder()
.Code(MessageCode.PeerBrowseRequest)
.Build();
var c = new MessageConnection(MessageConnectionType.Peer, username, ipAddress, port);
c.SetProperty("State", ConnectionState.Disconnecting);
var ex = await Record.ExceptionAsync(async() => await c.WriteMessageAsync(msg));
Assert.NotNull(ex);
Assert.IsType <InvalidOperationException>(ex);
}
public async Task WriteMessageAsync_Writes_When_Connected(string username, IPAddress ipAddress, int port)
{
var streamMock = new Mock <INetworkStream>();
streamMock.Setup(s => s.ReadAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()))
.Returns(Task.Run(() => 1));
var tcpMock = new Mock <ITcpClient>();
tcpMock.Setup(s => s.Connected).Returns(true);
tcpMock.Setup(s => s.GetStream()).Returns(streamMock.Object);
var msg = new MessageBuilder()
.Code(MessageCode.PeerBrowseRequest)
.Build();
var c = new MessageConnection(MessageConnectionType.Peer, username, ipAddress, port, tcpClient: tcpMock.Object);
await c.ConnectAsync();
await c.WriteMessageAsync(msg);
streamMock.Verify(s => s.WriteAsync(It.IsAny <byte[]>(), It.IsAny <int>(), It.IsAny <int>(), It.IsAny <CancellationToken>()), Times.Once);
}
