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