diff --git a/neo.UnitTests/UT_P2PMessage.cs b/neo.UnitTests/UT_P2PMessage.cs index 407e3692e7..ee27228a2e 100644 --- a/neo.UnitTests/UT_P2PMessage.cs +++ b/neo.UnitTests/UT_P2PMessage.cs @@ -47,6 +47,33 @@ public void Serialize_Deserialize_ByteString() buffer.Count.Should().Be(length); } + [TestMethod] + public void Serialize_Deserialize_WithoutPayload() + { + var msg = Message.Create(MessageCommand.GetAddr); + var buffer = msg.ToArray(); + var copy = buffer.AsSerializable(); + + copy.Command.Should().Be(msg.Command); + copy.Flags.Should().Be(msg.Flags); + copy.Payload.Should().Be(null); + } + + [TestMethod] + public void Serialize_Deserialize_WithoutPayload_ByteString() + { + var msg = Message.Create(MessageCommand.GetAddr); + var buffer = ByteString.CopyFrom(msg.ToArray()); + var length = Message.TryDeserialize(buffer, out var copy); + + copy.Command.Should().Be(msg.Command); + copy.Flags.Should().Be(msg.Flags); + copy.Payload.Should().Be(null); + + buffer.Count.Should().Be(length); + } + + [TestMethod] public void Compression() {