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); }