public void GetVersionMsgTest() { var ns = new MockNodeStatus(); var cs = new MockClientSettings() { _protoVer = 123, _services = NodeServiceFlags.All, _time = 456, _port = 789, _ua = "foo", _relay = true, _netType = NetworkType.TestNet }; var bc = new MockBlockchain() { _height = 12345 }; var rep = new ReplyManager(ns, bc, cs) { rng = new MockNonceRng(0x0158a8e8ba5f3ed3) }; Message msg = rep.GetVersionMsg(); FastStream actual = new FastStream(); msg.Serialize(actual); byte[] expected = Helper.HexToBytes("0b11090776657273696f6e0000000000590000000ba371327b0000001f04000000000000c8010000000000001f0400000000000000000000000000000000ffff7f00000103151f0400000000000000000000000000000000ffff7f0000010315d33e5fbae8a8580103666f6f3930000001"); Assert.Equal(expected, actual.ToByteArray()); }
public void ConstructorTest() { var cs = new MockClientSettings() { _netType = NetworkType.MainNet, _buffLen = 10 }; MessageManager man = new MessageManager(cs, new MockReplyManager(), new NodeStatus()); Assert.True(man.IsReceiveCompleted); Assert.False(man.HasDataToSend); Assert.Null(man.DataToSend); }
public void GetPingMsgTest() { var cs = new MockClientSettings() { _netType = NetworkType.TestNet, _buffLen = 10 }; var expectedPing = new Message(new PingPayload(1), NetworkType.TestNet); var repMan = new MockReplyManager() { pingMsg = expectedPing }; MessageManager man = new MessageManager(cs, repMan, new NodeStatus()); var actualPing = man.GetPingMsg(); Assert.Same(expectedPing, actualPing); }
public void DataToSendTest() { var cs = new MockClientSettings() { _netType = NetworkType.MainNet, _buffLen = 10 }; MessageManager man = new MessageManager(cs, new MockReplyManager(), new NodeStatus()) { DataToSend = null }; Assert.True(man.IsReceiveCompleted); Assert.False(man.HasDataToSend); man.DataToSend = new byte[0]; Assert.True(man.IsReceiveCompleted); Assert.True(man.HasDataToSend); man.DataToSend = new byte[3] { 1, 2, 3 }; Assert.True(man.IsReceiveCompleted); Assert.True(man.HasDataToSend); }
public static IEnumerable <object[]> GetReplyCases() { var cs = new MockClientSettings() { _netType = NetworkType.MainNet }; var bc = new MockBlockchain(); var mockAddr0 = new NetworkAddressWithTime(NodeServiceFlags.All, IPAddress.Loopback, 1010, 5678); var mockAddr1 = new NetworkAddressWithTime(NodeServiceFlags.All, IPAddress.Parse("200.2.3.4"), 23, 98); var mockAddr2 = new NetworkAddressWithTime(NodeServiceFlags.All, IPAddress.Parse("1.2.3.4"), 8080, 665412); var mockAddr3 = new NetworkAddressWithTime(NodeServiceFlags.All, IPAddress.Parse("99.77.88.66"), 444, 120000); var mockAddrs = new NetworkAddressWithTime[] { mockAddr1, mockAddr2, mockAddr3 }; var mockAddrs1000 = Enumerable.Repeat(mockAddr0, 997).ToList(); mockAddrs1000.AddRange(mockAddrs); // last 3 items are distict var mockAddrs1002 = Enumerable.Repeat(mockAddr0, 999).ToList(); mockAddrs1002.AddRange(mockAddrs); // last 3 items are distict var expAddr1002_1 = Enumerable.Repeat(mockAddr0, 999).ToList(); expAddr1002_1.Add(mockAddr1); var expAddr1002_2 = new NetworkAddressWithTime[2] { mockAddr2, mockAddr3 }; var mockBlock = new MockSerializableBlock(Helper.HexToBytes("01000000161126f0d39ec082e51bbd29a1dfb40b416b445ac8e493f88ce993860000000030e2a3e32abf1663a854efbef1b233c67c8cdcef5656fe3b4f28e52112469e9bae306849ffff001d16d1b42d0101000000010000000000000000000000000000000000000000000000000000000000000000ffffffff0704ffff001d0116ffffffff0100f2052a01000000434104f5efde0c2d30ab28e3dbe804c1a4aaf13066f9b198a4159c76f8f79b3b20caf99f7c979ed6c71481061277a6fc8666977c249da99960c97c8d8714fda9f0e883ac00000000")); yield return(new object[] { // Addr new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, updateTime = true }, new MockClientSettings() { _netType = NetworkType.MainNet, addrsToReceive = mockAddrs }, bc, new Message(new AddrPayload(mockAddrs), NetworkType.MainNet), null }); yield return(new object[] { // Block new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, updateTime = true }, new MockClientSettings() { _netType = NetworkType.MainNet, }, new MockBlockchain() { expProcessBlk = "00000000841cb802ca97cf20fb9470480cae9e5daa5d06b4a18ae2d5dd7f186f", blkProcessSuccess = true }, new Message(new BlockPayload(mockBlock), NetworkType.MainNet), null }); yield return(new object[] { // Block new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, updateTime = true, mediumViolation = true }, new MockClientSettings() { _netType = NetworkType.MainNet, }, new MockBlockchain() { expProcessBlk = "00000000841cb802ca97cf20fb9470480cae9e5daa5d06b4a18ae2d5dd7f186f", blkProcessSuccess = false }, new Message(new BlockPayload(mockBlock), NetworkType.MainNet), null }); yield return(new object[] { // GetAddr with smaller than max items new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, updateTime = true }, new MockClientSettings() { _netType = NetworkType.MainNet, addrsToReturn = mockAddrs }, bc, new Message(new GetAddrPayload(), NetworkType.MainNet), new Message[1] { new Message(new AddrPayload(mockAddrs), NetworkType.MainNet) } }); yield return(new object[] { // GetAddr with max items new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, updateTime = true }, new MockClientSettings() { _netType = NetworkType.MainNet, addrsToReturn = mockAddrs1000.ToArray() }, bc, new Message(new GetAddrPayload(), NetworkType.MainNet), new Message[1] { new Message(new AddrPayload(mockAddrs1000.ToArray()), NetworkType.MainNet) } }); yield return(new object[] { // GetAddr with more than max items (needs 2 messages) new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, updateTime = true }, new MockClientSettings() { _netType = NetworkType.MainNet, addrsToReturn = mockAddrs1002.ToArray() }, bc, new Message(new GetAddrPayload(), NetworkType.MainNet), new Message[2] { new Message(new AddrPayload(expAddr1002_1.ToArray()), NetworkType.MainNet), new Message(new AddrPayload(expAddr1002_2), NetworkType.MainNet) } }); yield return(new object[] { // GetAddr with 0 items (no reply message) new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, updateTime = true }, new MockClientSettings() { _netType = NetworkType.MainNet, addrsToReturn = new NetworkAddressWithTime[0] }, bc, new Message(new GetAddrPayload(), NetworkType.MainNet), null }); yield return(new object[] { // Ping new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, updateTime = true }, cs, bc, new Message(new PingPayload(98765), NetworkType.MainNet), new Message[1] { new Message(new PongPayload(98765), NetworkType.MainNet) } }); yield return(new object[] { // Pong new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, updateTime = true }, cs, bc, new Message(new PongPayload(98765), NetworkType.MainNet), null }); }
public static IEnumerable <object[]> GetVersionCases() { var cs = new MockClientSettings() { _protoVer = 123, _services = NodeServiceFlags.All, _time = 456, _port = 789, _ua = "foo", _relay = true, _netType = NetworkType.MainNet }; var bc = new MockBlockchain() { _height = 12345 }; var verPl = new VersionPayload(); Assert.True(verPl.TryDeserialize(new FastStreamReader(Helper.HexToBytes("721101000100000000000000bc8f5e5400000000010000000000000000000000000000000000ffffc61b6409208d010000000000000000000000000000000000ffffcb0071c0208d128035cbc97953f80f2f5361746f7368693a302e392e332fcf05050001")), out string error), error); var msg = new Message(verPl, NetworkType.MainNet); var rcv = new NetworkAddress(NodeServiceFlags.NodeNetwork, IPAddress.Parse("203.0.113.192"), 8333); var trs = new NetworkAddress(NodeServiceFlags.All, IPAddress.Loopback, 789); var verak = new Message(new VerackPayload(), NetworkType.MainNet); var ver = new Message(new VersionPayload(123, 456, rcv, trs, 0x0158a8e8ba5f3ed3, "foo", 12345, true), NetworkType.MainNet); yield return(new object[] { new MockNodeStatus() { _handShakeToReturn = HandShakeState.None, _handShakeToSet = HandShakeState.ReceivedAndReplied, updateTime = true }, cs, bc, msg, new Message[] { verak, ver } }); yield return(new object[] { new MockNodeStatus() { _handShakeToReturn = HandShakeState.ReceivedAndReplied, mediumViolation = true, updateTime = true }, cs, bc, msg, null }); yield return(new object[] { new MockNodeStatus() { _handShakeToReturn = HandShakeState.Sent, _handShakeToSet = HandShakeState.SentAndReceived, updateTime = true }, cs, bc, msg, new Message[] { verak } }); yield return(new object[] { new MockNodeStatus() { _handShakeToReturn = HandShakeState.SentAndConfirmed, _handShakeToSet = HandShakeState.Finished, updateTime = true }, cs, bc, msg, new Message[] { verak } }); yield return(new object[] { new MockNodeStatus() { _handShakeToReturn = HandShakeState.SentAndReceived, mediumViolation = true, updateTime = true }, cs, bc, msg, null }); yield return(new object[] { new MockNodeStatus() { _handShakeToReturn = HandShakeState.Finished, mediumViolation = true, updateTime = true }, cs, bc, msg, null }); }