More ut

neo-project · May 3, 2019 · 70abc9c · 70abc9c
1 parent 5328c33
commit 70abc9c
Showing 1 changed file with 27 additions and 0 deletions.
diff --git 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<Message>();
+
+            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()
         {